From e3557ff8fdb00e34edba2a37731ef4a77384e4df Mon Sep 17 00:00:00 2001
From: Siarhei Siniak <siarheisiniak@gmail.com>
Date: Sun, 7 Jul 2024 00:56:49 +0300
Subject: [PATCH] [~] Refactor
---
python/tasks/tiktok/utils.py | 11 +++++++++--
1 file changed, 9 insertions(+), 2 deletions(-)
diff --git a/python/tasks/tiktok/utils.py b/python/tasks/tiktok/utils.py
index 5892a46..01c5367 100644
--- a/python/tasks/tiktok/utils.py
+++ b/python/tasks/tiktok/utils.py
@@ -85,6 +85,7 @@ class Backend(celery.backends.redis.RedisBackend):
*args,
task_id: Optional[str]=None,
state: Optional[str]=None,
+ request: Optional[Any]=None,
**kwargs
):
current_meta = None
@@ -92,7 +93,13 @@ class Backend(celery.backends.redis.RedisBackend):
if not task_id is None:
current_meta = self._get_task_meta_for(task_id)
- res = super()._get_result_meta(*args, state=state, **kwargs)
+ res = super()._get_result_meta(*args, state=state, request=request, **kwargs)
+
+ if not request is None:
+ import pprint
+ print(request)
+ if not '_task_name' in res:
+ res['_task_name'] = request.task
if state == celery.states.STARTED:
if not 'date_started' in res:
@@ -102,7 +109,7 @@ class Backend(celery.backends.redis.RedisBackend):
if not '_date_done' in res:
res['_date_done'] = datetime.datetime.now()
- for k in ['date_started', '_date_done',]:
+ for k in ['date_started', '_date_done', '_task_name']:
if k in current_meta:
res[k] = current_meta[k]