mirror of
https://github.com/jokob-sk/NetAlertX.git
synced 2025-12-07 01:26:11 -08:00
65 lines
2.0 KiB
Python
Executable File
65 lines
2.0 KiB
Python
Executable File
#!/usr/bin/env python
|
|
# test script by running python script.py devices=test,dummy
|
|
|
|
import os
|
|
import pathlib
|
|
import argparse
|
|
import sys
|
|
import hashlib
|
|
|
|
sys.path.append("/home/pi/pialert/front/plugins")
|
|
sys.path.append('/home/pi/pialert/pialert')
|
|
|
|
from plugin_helper import Plugin_Object, Plugin_Objects, decodeBase64
|
|
from logger import mylog, append_line_to_file
|
|
from helper import timeNowTZ
|
|
from const import logPath, pialertPath
|
|
|
|
CUR_PATH = str(pathlib.Path(__file__).parent.resolve())
|
|
LOG_FILE = os.path.join(CUR_PATH, 'script.log')
|
|
RESULT_FILE = os.path.join(CUR_PATH, 'last_result.log')
|
|
|
|
def main():
|
|
|
|
# the script expects a parameter in the format of devices=device1,device2,...
|
|
parser = argparse.ArgumentParser(description='Import devices from settings')
|
|
parser.add_argument('devices', action="store", help="list of device names separated by ','")
|
|
values = parser.parse_args()
|
|
|
|
mylog('verbose', ['[UNDIS] In script'])
|
|
|
|
plugin_objects = Plugin_Objects( RESULT_FILE )
|
|
|
|
if values.devices:
|
|
for fake_dev in values.devices.split('=')[1].split(','):
|
|
|
|
fake_mac = string_to_mac_hash(fake_dev)
|
|
|
|
plugin_objects.add_object(
|
|
primaryId=fake_dev, # MAC (Device Name)
|
|
secondaryId="0.0.0.0", # IP Address (always 0.0.0.0)
|
|
watched1=fake_dev, # Device Name
|
|
watched2="",
|
|
watched3="",
|
|
watched4="",
|
|
extra="",
|
|
foreignKey=fake_mac)
|
|
|
|
plugin_objects.write_result_file()
|
|
|
|
return 0
|
|
|
|
def string_to_mac_hash(input_string):
|
|
# Calculate a hash using SHA-256
|
|
sha256_hash = hashlib.sha256(input_string.encode()).hexdigest()
|
|
|
|
# Take the first 12 characters of the hash and format as a MAC address
|
|
mac_hash = ':'.join(sha256_hash[i:i+2] for i in range(0, 12, 2))
|
|
|
|
return mac_hash
|
|
|
|
#===============================================================================
|
|
# BEGIN
|
|
#===============================================================================
|
|
if __name__ == '__main__':
|
|
main() |