Implement more UI elements

core/lib/threshold.py

@@ -0,0 +1,76 @@
+import logging
+from json import dumps
+
+import requests
+from django.conf import settings
+from requests.exceptions import JSONDecodeError
+
+logger = logging.getLogger(__name__)
+
+
+def escape(obj):
+    chars = ["[", "]", "^", "-", "*", "?"]
+    if isinstance(obj, str):
+        obj = obj.replace("\\", "\\\\")
+        for i in chars:
+            obj = obj.replace(i, "\\" + i)
+    elif isinstance(obj, list):
+        for i in obj:
+            i = escape(i)
+    elif isinstance(obj, dict):
+        for key in obj:
+            obj[key] = escape(obj[key])
+    return obj
+
+
+def threshold_request(url, data):
+    headers = {
+        "ApiKey": settings.THRESHOLD_API_KEY,
+        "Token": settings.THRESHOLD_API_TOKEN,
+    }
+    for key in data:
+        data[key] = escape(data[key])
+    r = requests.post(
+        f"{settings.THRESHOLD_ENDPOINT}/{url}/", data=dumps(data), headers=headers
+    )
+    if not r.headers.get("Counter") == settings.THRESHOLD_API_COUNTER:
+        logger.error(
+            (
+                f"Threshold API counter mismatch: "
+                f"{r.headers.get('Counter')} != "
+                f"{settings.THRESHOLD_API_COUNTER}"
+            )
+        )
+        return False
+    try:
+        response = r.json()
+    except JSONDecodeError:
+        logging.error(f"Invalid JSON response: {r.text}")
+    return response
+
+
+def get_chans(net, query):
+    url = "chans"
+    payload = {"net": net, "query": query}
+    channels = threshold_request(url, payload)
+    if not channels:
+        return []
+    return channels["chans"]
+
+
+def get_users(net, query):
+    url = "users"
+    payload = {"net": net, "query": query}
+    users = threshold_request(url, payload)
+    if not users:
+        return []
+    return users["users"]
+
+
+def annotate_online(net, query):
+    url = "online"
+    payload = {"net": net, "query": query}
+    online_info = threshold_request(url, payload)
+    if not online_info:
+        return {}
+    return online_info