Add files via upload

callsmusic/queues/__init__.py

@@ -0,0 +1 @@
+from .queues import put, get, is_empty, task_done, clear

callsmusic/queues/queues.py

@@ -1 +1,42 @@
+from asyncio import Queue, QueueEmpty as Empty
+from typing import Dict, Union
 
+queues: Dict[int, Queue] = {}
+
+
+async def put(chat_id: int, **kwargs) -> int:
+    if chat_id not in queues:
+        queues[chat_id] = Queue()
+    await queues[chat_id].put({**kwargs})
+    return queues[chat_id].qsize()
+
+
+def get(chat_id: int) -> Union[Dict[str, str], None]:
+    if chat_id in queues:
+        try:
+            return queues[chat_id].get_nowait()
+        except Empty:
+            return None
+
+
+def is_empty(chat_id: int) -> bool:
+    if chat_id in queues:
+        return queues[chat_id].empty()
+    return True
+
+
+def task_done(chat_id: int):
+    if chat_id in queues:
+        try:
+            queues[chat_id].task_done()
+        except ValueError:
+            pass
+
+
+def clear(chat_id: int):
+    if chat_id in queues:
+        if queues[chat_id].empty():
+            raise Empty
+        else:
+            queues[chat_id].queue = []
+    raise Empty