mirror of
https://github.com/jokob-sk/NetAlertX.git
synced 2026-05-29 11:08:37 -04:00
76612e5d0e
match request timeout to script timeout -1 second. Always use at least 1 sec. timeout for request.
76 lines
2.9 KiB
Python
76 lines
2.9 KiB
Python
#!/usr/bin/env python3
|
|
import os
|
|
import sys
|
|
import requests
|
|
|
|
sys.path.insert(0, os.path.join(os.path.dirname(__file__), '../../server'))
|
|
sys.path.insert(0, os.path.join(os.path.dirname(__file__), '../plugins'))
|
|
|
|
from plugin_helper import Plugin_Objects, mylog, handleEmpty, is_mac
|
|
from helper import get_setting_value
|
|
from const import logPath
|
|
|
|
pluginName = 'KEALSS'
|
|
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')
|
|
|
|
plugin_objects = Plugin_Objects(RESULT_FILE)
|
|
|
|
|
|
def main():
|
|
try:
|
|
url = get_setting_value(f'{pluginName}_URL')
|
|
user = get_setting_value(f'{pluginName}_USER')
|
|
password = get_setting_value(f'{pluginName}_PASS')
|
|
timeout = get_setting_value(f'{pluginName}_RUN_TIMEOUT')
|
|
|
|
mylog('verbose', [f'[{pluginName}] Querying Kea API at {url}'])
|
|
|
|
payload = {'command': 'lease4-get-all', 'service': ['dhcp4']}
|
|
|
|
response = requests.post(url, json=payload, auth=(user, password), timeout=max(1, timeout - 1))
|
|
response.raise_for_status()
|
|
data = response.json()
|
|
|
|
count = 0
|
|
for entry in data:
|
|
text = entry.get('text', '[API provided no text]');
|
|
# Result: 0 (success), 1 (error), or 3 (empty).
|
|
if entry['result'] == 0:
|
|
leases = entry['arguments']['leases']
|
|
for lease in leases:
|
|
mac = lease['hw-address']
|
|
state = lease['state']
|
|
if is_mac(mac):
|
|
plugin_objects.add_object(
|
|
primaryId = mac,
|
|
secondaryId = lease['ip-address'],
|
|
# Active or not, similar to watched1 of DHCPLSS plugin
|
|
watched1 = state == 0,
|
|
watched2 = lease['hostname'],
|
|
watched3 = None,
|
|
# Default (or assigned) (0), declined (1), expired-reclaimed (2), released (3), and registered (4)).
|
|
watched4 = state,
|
|
extra = None,
|
|
foreignKey = mac
|
|
)
|
|
count += 1
|
|
plugin_objects.write_result_file()
|
|
|
|
mylog('verbose', [f'[{pluginName}] Kea API response: {text}'])
|
|
mylog('verbose', [f'[{pluginName}] Successfully imported {count} devices reported by Kea API'])
|
|
elif entry['result'] == 1:
|
|
mylog('none', [f'[{pluginName}] ⚠ ERROR: Kea API indicated error: {text}'])
|
|
elif entry['result'] == 3:
|
|
mylog('verbose', [f'[{pluginName}] Kea API indicates no entries found: {text}'])
|
|
|
|
|
|
except Exception as e:
|
|
mylog('none', [f'[{pluginName}] ⚠ ERROR: {str(e)}'])
|
|
|
|
|
|
|
|
if __name__ == '__main__':
|
|
main()
|