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]