mirror of https://github.com/CIRCL/lookyloo
58 lines
1.6 KiB
Python
Executable File
58 lines
1.6 KiB
Python
Executable File
#!/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))
|