mirror of
https://github.com/jokob-sk/NetAlertX.git
synced 2025-12-06 17:15:38 -08:00
123 lines
3.6 KiB
Python
Executable File
123 lines
3.6 KiB
Python
Executable File
#!/usr/bin/env python
|
|
|
|
import os
|
|
import pathlib
|
|
import sys
|
|
import json
|
|
import sqlite3
|
|
from pytz import timezone
|
|
from wakeonlan import send_magic_packet
|
|
|
|
# Define the installation path and extend the system path for plugin imports
|
|
INSTALL_PATH = "/app"
|
|
sys.path.extend([f"{INSTALL_PATH}/front/plugins", f"{INSTALL_PATH}/server"])
|
|
|
|
from plugin_helper import Plugin_Object, Plugin_Objects, decodeBase64
|
|
from plugin_utils import get_plugins_configs
|
|
from logger import mylog, Logger
|
|
from const import pluginsPath, fullDbPath, logPath
|
|
from helper import get_setting_value
|
|
from messaging.in_app import write_notification
|
|
from database import DB
|
|
from models.device_instance import DeviceInstance
|
|
import conf
|
|
|
|
# Make sure the TIMEZONE for logging is correct
|
|
conf.tz = timezone(get_setting_value('TIMEZONE'))
|
|
|
|
# Make sure log level is initialized correctly
|
|
Logger(get_setting_value('LOG_LEVEL'))
|
|
|
|
pluginName = 'WOL'
|
|
|
|
# Define the current path and log file paths
|
|
LOG_PATH = logPath + '/plugins'
|
|
LOG_FILE = os.path.join(LOG_PATH, f'script.{pluginName}.log')
|
|
RESULT_FILE = os.path.join(LOG_PATH, f'last_result.{pluginName}.log')
|
|
|
|
# Initialize the Plugin obj output file
|
|
plugin_objects = Plugin_Objects(RESULT_FILE)
|
|
|
|
|
|
|
|
def main():
|
|
mylog('none', [f'[{pluginName}] In script'])
|
|
|
|
# Retrieve configuration settings
|
|
broadcast_ips = get_setting_value('WOL_broadcast_ips')
|
|
devices_to_wake = get_setting_value('WOL_devices_to_wake')
|
|
ports = get_setting_value('WOL_ports')
|
|
|
|
mylog('verbose', [f'[{pluginName}] broadcast_ips value {broadcast_ips}'])
|
|
|
|
# Create a database connection
|
|
db = DB() # instance of class DB
|
|
db.open()
|
|
|
|
# Create a DeviceInstance instance
|
|
device_handler = DeviceInstance(db)
|
|
|
|
# Retrieve devices
|
|
if 'offline' in devices_to_wake:
|
|
|
|
devices_to_wake = device_handler.getOffline()
|
|
|
|
elif 'down' in devices_to_wake:
|
|
|
|
devices_to_wake = device_handler.getDown()
|
|
|
|
else:
|
|
mylog('verbose', [f'[{pluginName}] Invalid setting WOL_devices_to_wake {devices_to_wake}'])
|
|
return
|
|
|
|
# execute script if something to do
|
|
if len(devices_to_wake) > 0:
|
|
for device in devices_to_wake:
|
|
for ip in broadcast_ips:
|
|
for port in ports:
|
|
result = execute(port, ip, device['devMac'], device['devName'])
|
|
|
|
# Process the data into native application tables
|
|
if len(result) > 0:
|
|
|
|
plugin_objects.add_object(
|
|
primaryId = device['devMac'],
|
|
secondaryId = device['devLastIP'],
|
|
watched1 = device['devName'],
|
|
watched2 = result,
|
|
watched3 = '',
|
|
watched4 = '',
|
|
extra = '',
|
|
foreignKey = device['devMac']
|
|
)
|
|
|
|
# log result
|
|
plugin_objects.write_result_file()
|
|
else:
|
|
mylog('none', [f'[{pluginName}] No devices to wake'])
|
|
|
|
mylog('none', [f'[{pluginName}] Script finished'])
|
|
|
|
return 0
|
|
|
|
# wake
|
|
def execute(port, ip, mac, name):
|
|
|
|
result = 'null'
|
|
try:
|
|
# Send the magic packet to wake up the device
|
|
send_magic_packet(mac, ip_address=ip, port=int(port))
|
|
mylog('verbose', [f'[{pluginName}] Magic packet sent to {mac} ({name})'])
|
|
|
|
result = 'success'
|
|
|
|
except Exception as e:
|
|
result = str(e)
|
|
mylog('verbose', [f'[{pluginName}] Failed to send magic packet to {mac} ({name}): {e}'])
|
|
|
|
# Return the data result
|
|
return result
|
|
|
|
if __name__ == '__main__':
|
|
main()
|