p1-taskqueue 0.1.5__tar.gz → 0.1.6__tar.gz

{p1_taskqueue-0.1.5 → p1_taskqueue-0.1.6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: p1-taskqueue
-Version: 0.1.5
+Version: 0.1.6
 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.5 → p1_taskqueue-0.1.6}/pyproject.toml

@@ -5,7 +5,7 @@ build-backend = "setuptools.build_meta"
 [project]
 name = "p1-taskqueue"
 # DO NOT CHANGE THIS VERSION - it gets automatically replaced by CI/CD with the git tag version
-version = "0.1.5"
+version = "0.1.6"
 description = "A Task Queue Wrapper for Dekoruma Backend"
 authors = [
     {name = "Chalvin", email = "engineering@dekoruma.com"}

{p1_taskqueue-0.1.5 → p1_taskqueue-0.1.6}/src/p1_taskqueue.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: p1-taskqueue
-Version: 0.1.5
+Version: 0.1.6
 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.5 → p1_taskqueue-0.1.6}/src/taskqueue/cmanager.py

@@ -160,17 +160,18 @@ class CManager:
             task_name, task_args, task_kwargs = _build_dynamic_task_call(
                 func, *func_args, **func_kwargs)
 
-            self._send_task(task_name, task_args, task_kwargs, queue_kwargs)
+            task_id = self._send_task(task_name, task_args,
+                                      task_kwargs, queue_kwargs)
 
-            logger.info('[_enqueue_op_base %s] Submit Celery Task SUCCESS, task_name: %s args: %s, kwargs: %s' % (
-                enqueue_op_type, task_name, task_args, task_kwargs))
+            logger.info('[_enqueue_op_base %s] Submit Celery Task SUCCESS, task_name: %s args: %s, kwargs: %s, task_id: %s' % (
+                enqueue_op_type, task_name, task_args, task_kwargs, task_id))
 
         except Exception as e:
             logger.exception('[_enqueue_op_base %s] Submit Celery Task FAILED, error: %s, args: %s, kwargs: %s' % (
                 enqueue_op_type, str(e), args, kwargs))
             raise e
 
-    def _send_task(self, task_name: str, task_args: list, task_kwargs: dict, queue_kwargs: Dict[str, Any]) -> None:
+    def _send_task(self, task_name: str, task_args: list, task_kwargs: dict, queue_kwargs: Dict[str, Any]) -> str:
         celery_app = self._get_celery_app()
 
         queue_name = queue_kwargs.pop("channel", None)
@@ -194,8 +195,9 @@ class CManager:
         else:
             task_kwargs_with_retry["retry"] = retry_policy
 
-        celery_app.send_task(task_name, args=task_args,
-                             kwargs=task_kwargs_with_retry, **send_opts)
+        task = celery_app.send_task(task_name, args=task_args,
+                                    kwargs=task_kwargs_with_retry, **send_opts)
+        return str(task.id)
 
 
 cm = CManager()