p1-taskqueue 0.1.7__py3-none-any.whl → 0.1.9__py3-none-any.whl

{p1_taskqueue-0.1.7.dist-info → p1_taskqueue-0.1.9.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: p1-taskqueue
-Version: 0.1.7
+Version: 0.1.9
 Summary: A Task Queue Wrapper for Dekoruma Backend
 Author-email: Chalvin <engineering@dekoruma.com>
 Project-URL: Homepage, https://github.com/Dekoruma/p1-taskqueue

{p1_taskqueue-0.1.7.dist-info → p1_taskqueue-0.1.9.dist-info}/RECORD

@@ -2,8 +2,8 @@ taskqueue/__init__.py,sha256=nNHXIyysNLom9f8of-d__pWJ-YF53Mbrbsb1frPzPPI,298
 taskqueue/celery_app.py,sha256=dUT-7XzsSQbr8vKrLv7f_6iYxTCUEJZHEt9fL-KIQ5U,3302
 taskqueue/cmanager.py,sha256=raQwdccYTpWRiW8UGU5OcUEN0jZr9TTlIEnhZGtamiI,12274
 taskqueue/libs/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-taskqueue/libs/helper_test.py,sha256=_yfPhm_7CzbN30j7E_Ld5KwXRlGHhTNrRTOBZQySEIU,3656
-p1_taskqueue-0.1.7.dist-info/METADATA,sha256=y3FYv9zHOECh2FzdJRBKofYPGvlr1a1A0wUk4B98TyU,1508
-p1_taskqueue-0.1.7.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-p1_taskqueue-0.1.7.dist-info/top_level.txt,sha256=hA3SM1ik2K8iPqtlt_-_nJ4TAePwHPN3vsOc4EiynqU,10
-p1_taskqueue-0.1.7.dist-info/RECORD,,
+taskqueue/libs/helper_test.py,sha256=JCdh2S29PpL8RqUxqkcIqwIvr3M9puqHglBZAmfPkuw,7722
+p1_taskqueue-0.1.9.dist-info/METADATA,sha256=qZ-f8EpVmzgYwvU76H-PQ6CDPuUWIv0cqVs0FgkgRwk,1508
+p1_taskqueue-0.1.9.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+p1_taskqueue-0.1.9.dist-info/top_level.txt,sha256=hA3SM1ik2K8iPqtlt_-_nJ4TAePwHPN3vsOc4EiynqU,10
+p1_taskqueue-0.1.9.dist-info/RECORD,,

taskqueue/libs/helper_test.py

@@ -58,10 +58,18 @@ def celery_worker_burst(include_func_names: List[str], channel: str = "default")
                             decoded_body) > 1 else {}
 
                         full_func_name = ""
-                        if task_name.endswith("dynamic_function_executor") and len(task_args) >= 2:
-                            full_func_name = f"{task_args[0]}.{task_args[1]}"
-                        elif task_name.endswith("dynamic_class_method_executor") and len(task_args) >= 3:
-                            full_func_name = f"{task_args[0]}.{task_args[1]}.{task_args[2]}"
+                        if task_name.endswith("dynamic_function_executor"):
+                            module_path = task_kwargs.get('module_path', '')
+                            function_name = task_kwargs.get(
+                                'function_name', '')
+                            if module_path and function_name:
+                                full_func_name = f"{module_path}.{function_name}"
+                        elif task_name.endswith("dynamic_class_method_executor"):
+                            module_path = task_kwargs.get('module_path', '')
+                            class_name = task_kwargs.get('class_name', '')
+                            method_name = task_kwargs.get('method_name', '')
+                            if module_path and class_name and method_name:
+                                full_func_name = f"{module_path}.{class_name}.{method_name}"
 
                         should_execute = full_func_name in included_set if full_func_name else False
 
@@ -86,3 +94,85 @@ def celery_worker_burst(include_func_names: List[str], channel: str = "default")
     except Exception as e:
         logger.error(
             f"Failed to connect to queue {channel}: {type(e).__name__}: {e}")
+
+
+def get_queued_tasks(channel: str = "default"):
+    app = current_app
+    queued_tasks = []
+
+    try:
+        with app.connection_for_read() as conn:
+            with conn.channel() as chan:
+                queue = app.amqp.queues[channel](chan)
+
+                while True:
+                    message = queue.get(no_ack=True)
+                    if not message:
+                        break
+
+                    task_name = message.headers.get("task")
+
+                    try:
+                        accept = {"application/json",
+                                  "application/x-python-serialize"}
+                        decoded_body = loads(
+                            message.body, message.content_type, message.content_encoding, accept=accept
+                        )
+
+                        task_args = decoded_body[0] if decoded_body else []
+                        task_kwargs = decoded_body[1] if len(
+                            decoded_body) > 1 else {}
+
+                        full_func_name = ""
+                        if task_name and task_name.endswith("dynamic_function_executor"):
+                            module_path = task_kwargs.get('module_path', '')
+                            function_name = task_kwargs.get(
+                                'function_name', '')
+                            if module_path and function_name:
+                                full_func_name = f"{module_path}.{function_name}"
+                        elif task_name and task_name.endswith("dynamic_class_method_executor"):
+                            module_path = task_kwargs.get('module_path', '')
+                            class_name = task_kwargs.get('class_name', '')
+                            method_name = task_kwargs.get('method_name', '')
+                            if module_path and class_name and method_name:
+                                full_func_name = f"{module_path}.{class_name}.{method_name}"
+
+                        queued_tasks.append({
+                            'task_name': task_name,
+                            'full_func_name': full_func_name,
+                            'args': task_args,
+                            'kwargs': task_kwargs,
+                            'headers': message.headers
+                        })
+
+                    except Exception as e:
+                        logger.warning(f"Failed to decode message: {e}")
+                        continue
+
+    except Exception as e:
+        logger.error(
+            f"Failed to get queued tasks from queue {channel}: {type(e).__name__}: {e}")
+
+    return queued_tasks
+
+
+def is_task_in_queue(expected_func_name: str, channel: str = "default"):
+    queued_tasks = get_queued_tasks(channel)
+    for task in queued_tasks:
+        if task['full_func_name'] == expected_func_name:
+            return True
+    logger.info(
+        f"Task {expected_func_name} not found in queue {channel}. Queued tasks: {[task['full_func_name'] for task in queued_tasks]}")
+    return False
+
+
+def assert_task_in_queue(expected_func_name: str, channel: str = "default", msg: str = None):
+    if not is_task_in_queue(expected_func_name, channel):
+        error_msg = msg or f"Task '{expected_func_name}' not found in queue '{channel}'"
+        raise AssertionError(error_msg)
+
+
+def assert_task_not_in_queue(expected_func_name: str, channel: str = "default", msg: str = None):
+    if is_task_in_queue(expected_func_name, channel):
+        error_msg = msg or f"Task '{expected_func_name}' found in queue '{channel}'"
+        raise AssertionError(error_msg)