Added AsusWRT device import plugin

front/plugins/asuswrt_import/script.py

@@ -0,0 +1,108 @@
+#!/usr/bin/env python
+import os
+import sys
+
+INSTALL_PATH = "/app"
+sys.path.extend([f"{INSTALL_PATH}/front/plugins", f"{INSTALL_PATH}/server"])
+pluginName = "ASUSWRT"
+
+import asyncio
+
+import aiohttp
+import conf
+from asusrouter import AsusData, AsusRouter
+from asusrouter.modules.connection import ConnectionState
+from const import logPath
+from helper import get_setting_value
+from logger import Logger, mylog
+from plugin_helper import (Plugin_Object, Plugin_Objects, decodeBase64,
+                           handleEmpty)
+from pytz import timezone
+
+conf.tz = timezone(get_setting_value("TIMEZONE"))
+
+Logger(get_setting_value("LOG_LEVEL"))
+
+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():
+    mylog("verbose", [f"[{pluginName}] started."])
+
+    device_data = get_device_data()
+
+    mylog(
+        "verbose",
+        [f"[{pluginName}] Found '{len(device_data)}' devices"],
+    )
+
+    filtered_devices = [
+        (key, device)
+        for key, device in device_data.items()
+        if device.state == ConnectionState.CONNECTED
+    ]
+
+    mylog(
+        "verbose",
+        [f"[{pluginName}] Processing '{len(filtered_devices)}' connected devices"],
+    )
+
+    for mac, device in filtered_devices:
+        entry_mac = str(device.description.mac).lower()
+
+        plugin_objects.add_object(
+            primaryId=entry_mac,
+            secondaryId=handleEmpty(device.connection.ip_address),
+            watched1=handleEmpty(device.description.name),
+            watched2=handleEmpty(device.description.vendor),
+            extra=pluginName,
+            foreignKey=entry_mac,
+        )
+
+    plugin_objects.write_result_file()
+
+    mylog("verbose", [f"[{pluginName}] finished"])
+
+    return 0
+
+
+def get_device_data():
+    # Create a new event loop
+    loop = asyncio.new_event_loop()
+
+    # Create aiohttp session
+    session = aiohttp.ClientSession(loop=loop)
+
+    router = AsusRouter(  # required - both IP and URL supported
+        hostname=get_setting_value("ASUSWRT_host"),  # required
+        username=get_setting_value("ASUSWRT_user"),  # required
+        password=get_setting_value("ASUSWRT_password"),  # required
+        use_ssl=get_setting_value("ASUSWRT_ssl"),  # optional
+        session=session,  # optional
+    )
+
+    # Connect to the router
+    loop.run_until_complete(router.async_connect())
+
+    if router.connected:
+        mylog("verbose", [f"[{pluginName}] logged in successfully."])
+    else:
+        mylog("error", [f"[{pluginName}] failed to login."])
+        return []
+
+    # Now you can use the router object to call methods
+    clients = loop.run_until_complete(router.async_get_data(AsusData.CLIENTS))
+
+    # Remember to disconnect and close the session when you're done
+    loop.run_until_complete(router.async_disconnect())
+    loop.run_until_complete(session.close())
+
+    return clients
+
+
+if __name__ == "__main__":
+    main()