You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
23 lines
781 B
Python
23 lines
781 B
Python
from django.core.management.base import BaseCommand, CommandError
|
|
from redis import StrictRedis
|
|
from core.util import logs
|
|
from core.lib.rules import process_rules
|
|
import msgpack
|
|
|
|
log = logs.get_logger("processing")
|
|
|
|
class Command(BaseCommand):
|
|
def handle(self, *args, **options):
|
|
r = StrictRedis(unix_socket_path="/var/run/socks/redis.sock", db=0)
|
|
p = r.pubsub()
|
|
p.psubscribe("messages")
|
|
for message in p.listen():
|
|
if message:
|
|
if message["channel"] == b"messages":
|
|
data = message["data"]
|
|
try:
|
|
unpacked = msgpack.unpackb(data, raw=False)
|
|
except TypeError:
|
|
continue
|
|
process_rules(unpacked)
|