Renames websocket-filter to ws-sta-monitor to reflect goal of script

py-json/ws-sta-monitor.py

@@ -0,0 +1,162 @@
+#!/usr/bin/python3
+'''
+# - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
+#                                                                             -
+# Example of how to filter messages from the :8081 websocket                  -
+#                                                                             -
+# - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
+You will need websocket-client:
+apt install python3-websocket
+'''
+
+import sys
+if sys.version_info[0] != 3:
+    print("This script requires Python 3")
+    exit()
+import argparse
+import json
+import logging
+import time
+from time import sleep
+import websocket
+import re
+try:
+    import thread
+except ImportError:
+    import _thread as thread
+import pprint
+import LANforge
+from LANforge import LFRequest
+from LANforge import LFUtils
+from LANforge.LFUtils import NA
+
+ignore=[
+    "scan finished",
+    "scan started"
+    "CTRL-EVENT-SCAN-STARTED",
+    "-EVENT-SSID-TEMP-DISABLED",
+    "new station",
+    "del station",
+    "ping",
+    "deleted-alert",
+]
+interesting=[
+    "Trying to authenticate",
+    "auth: timed out",
+    "link DOWN",
+    "link UP",
+    'wifi-event'
+]
+
+rebank = {
+    "ifname" : re.compile("IFNAME=(\S+)")
+}
+# - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
+def main():
+    host = "ct524-debbie.jbr.candelatech.com"
+    base_url = "ws://%s:8081"%host
+    websock = None
+
+    # open websocket
+    start_websocket(base_url, websock)
+    if (websock is not None):
+        print ("Started websocket")
+    else:
+        print ("Failed to start websocket, bye.")
+        exit(1)
+
+# ----- ----- ----- ----- ----- ----- ----- ----- ----- ----- ----- -----
+def sock_filter(wsock, text):
+    global ignore
+    global interesting
+    global rebank
+    debug = 0
+    for test in ignore:
+        if (test in text):
+            if (debug):
+                print ("                ignoring ",text)
+            return;
+
+    try:
+        message = json.loads(text)
+        if (("time" in message) and ("timestamp" in message)):
+            return
+        if ("event_type" in message):
+            if (message["is_alert"]):
+                print ("alert: ", message["details"]);
+                return
+            else:
+                print ("event: ", message["details"]);
+                return
+
+        if ("wifi-event" in message):
+            for test in ignore:
+                if (test in message["wifi-event"]):
+                    if (debug):
+                        print ("                ignoring ",text)
+                    return;
+            #group = rebank["ifname"].match(message["wifi-event"]).group()
+            #if (group):
+            #    print ("IFname: ",group)
+
+            if ("disconnected" in message["wifi-event"]):
+                print ("Station down")
+                return
+            if ("Trying to authenticate" in message["wifi-event"]):
+                print ("station authenticating")
+                return
+            print ("wifi-event: ", message["wifi-event"])
+        else:
+            print ("\nUnhandled: ")
+            LFUtils.debug_printer.pprint(message)
+
+    except json.JSONDecodeError as derr:
+        print ("# ----- Decode err: ----- ----- ----- ----- ----- ----- ----- ----- ----- -----")
+        print (text)
+        print (derr)
+        print ("# ----- ----- ----- ----- ----- ----- ----- ----- ----- ----- ----- -----")
+        return
+    except Exception as ex:
+        print ("# ----- Not JSON: ----- ----- ----- ----- ----- ----- ----- ----- ----- -----")
+        print (text)
+        print (ex)
+        print ("# ----- ----- ----- ----- ----- ----- ----- ----- ----- ----- ----- -----")
+        return
+
+# ----- ----- ----- ----- ----- ----- ----- ----- ----- ----- ----- -----
+def m_error(wsock, err):
+    print ("# ----- ----- ----- ----- ----- ----- ----- ----- ----- ----- ----- -----\n")
+    LFUtils.debug_printer.pprint(err)
+    print ("# ----- ----- ----- ----- ----- ----- ----- ----- ----- ----- ----- -----\n")
+
+# ----- ----- ----- ----- ----- ----- ----- ----- ----- ----- ----- -----
+def m_open(wsock):
+    def run(*args):
+        time.sleep(0.1)
+        #ping = json.loads();
+        wsock.send('{"text":"ping"}')
+    thread.start_new_thread(run, ())
+
+# ----- ----- ----- ----- ----- ----- ----- ----- ----- ----- ----- -----
+def m_close(wsock):
+    LFUtils.debug_printer.pprint(wsock)
+
+# ----- ----- ----- ----- ----- ----- ----- ----- ----- ----- ----- -----
+def start_websocket(uri, *websock):
+    #websocket.enableTrace(True)
+    websock = websocket.WebSocketApp(uri,
+        on_message = sock_filter,
+        on_error = m_error,
+        on_close = m_close)
+    websock.on_open = m_open
+    websock.run_forever()
+
+
+# ----- ----- ----- ----- ----- ----- ----- ----- ----- ----- ----- -----
+if __name__ == '__main__':
+    main()
+
+
+####
+####
+####