Add tags for number of users and channels

core/lib/threshold.py

@@ -50,9 +50,10 @@ def threshold_request(url, data):
         return False
     try:
         response = r.json()
+        sort_data(response)
     except JSONDecodeError:
         logging.error(f"Invalid JSON response: {r.text}")
-    sort_data(response)
+        return False
     return response
 
 
@@ -81,3 +82,21 @@ def annotate_online(net, query):
     if not online_info:
         return {}
     return online_info
+
+
+def annotate_num_users(net, query):
+    url = "num_users"
+    payload = {"net": net, "query": query}
+    user_num_map = threshold_request(url, payload)
+    if not user_num_map:
+        return {}
+    return user_num_map
+
+
+def annotate_num_chans(net, query):
+    url = "num_chans"
+    payload = {"net": net, "query": query}
+    chan_num_map = threshold_request(url, payload)
+    if not chan_num_map:
+        return {}
+    return chan_num_map