no js

SprintLib/queue.py

@@ -1,3 +1,7 @@
+import json
+from enum import Enum, auto
+from typing import Union
+
 import pika
 
 from Sprint import settings
@@ -16,3 +20,45 @@ def send_testing(solution):
             routing_key="test",
             body=bytes(str(solution.id), encoding="utf-8"),
         )
+
+
+class Queue(str, Enum):
+    test = auto()
+    notification = auto()
+
+
+class QueueAccessor:
+
+    def publish(self, queue: Union[Queue, str], message: Union[bytes, dict]):
+        if isinstance(message, dict):
+            message = json.dumps(message).encode("UTF-8")
+        if isinstance(queue, str):
+            queue = Queue(queue)
+        with pika.BlockingConnection(
+                pika.ConnectionParameters(host=settings.RABBIT_HOST, port=settings.RABBIT_PORT)
+        ) as connection:
+            channel = connection.channel()
+            channel.queue_declare(queue=queue.name)
+            channel.basic_publish(
+                exchange="",
+                routing_key=queue.name,
+                body=message,
+            )
+
+
+def message_handler(queue: Union[Queue, str]):
+    if isinstance(queue, str):
+        queue = Queue(queue)
+
+    def decorator(func):
+        def new_func(*args, **kwargs):
+            print("Enter listener for queue", queue)
+            with pika.BlockingConnection(
+                pika.ConnectionParameters(host=settings.RABBIT_HOST)
+            ) as connection:
+                channel = connection.channel()
+                channel.queue_declare(queue=queue.name)
+                channel.basic_consume(queue=queue.name, on_message_callback=func, auto_ack=True)
+                channel.start_consuming()
+        return new_func
+    return decorator