new: Monitoring script (CLI)

tools/monitoring.py

@@ -0,0 +1,57 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+
+import json
+import logging
+
+from redis import Redis
+
+from lookyloo.helpers import get_config, get_socket_path, splash_status
+
+
+class Monitoring():
+
+    def __init__(self) -> None:
+        self.logger = logging.getLogger(f'{self.__class__.__name__}')
+        self.logger.setLevel(get_config('generic', 'loglevel'))
+
+        self.redis_cache: Redis = Redis(unix_socket_path=get_socket_path('cache'), decode_responses=True)
+        self.redis_indexing: Redis = Redis(unix_socket_path=get_socket_path('indexing'), decode_responses=True)
+
+    @property
+    def queues(self):
+        return self.redis_cache.zrevrangebyscore('queues', 'Inf', '-Inf', withscores=True)
+
+    @property
+    def ongoing_captures(self):
+        captures_uuid = self.redis_cache.zrevrangebyscore('to_capture', 'Inf', '-Inf', withscores=True)
+        if not captures_uuid:
+            return []
+        to_return = []
+        for uuid, rank in captures_uuid:
+            capture_params = self.redis_cache.hgetall(uuid)
+            if capture_params:
+                to_return.append((uuid, rank, capture_params))
+
+        return to_return
+
+
+if __name__ == '__main__':
+
+    status, message = splash_status()
+    if status:
+        print(message)
+    else:
+        print('Splash is down: ', message)
+
+    m = Monitoring()
+    print('Current queues:')
+    for q in m.queues:
+        print(q)
+    # ------------------
+    print('Captures details:')
+    captures = m.ongoing_captures
+    print('Queue length', len(captures))
+    for uuid, rank, d in captures:
+        print(uuid, rank)
+        print(json.dumps(d, indent=2))