[~] Refactor

This commit is contained in:
Siarhei Siniak 2024-07-07 00:56:49 +03:00
parent 94e3d90df2
commit e3557ff8fd

@ -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]