From 94e3d90df233326dd1e19d8d52d2fd2fdebbd2c8 Mon Sep 17 00:00:00 2001 From: Siarhei Siniak Date: Sun, 7 Jul 2024 00:45:50 +0300 Subject: [PATCH] [~] Refactor --- python/tasks/tiktok/utils.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/python/tasks/tiktok/utils.py b/python/tasks/tiktok/utils.py index 079cfcb..5892a46 100644 --- a/python/tasks/tiktok/utils.py +++ b/python/tasks/tiktok/utils.py @@ -98,12 +98,14 @@ class Backend(celery.backends.redis.RedisBackend): if not 'date_started' in res: res['date_started'] = datetime.datetime.now() - for k in ['date_started',]: + if state in celery.states.READY_STATES: + if not '_date_done' in res: + res['_date_done'] = datetime.datetime.now() + + for k in ['date_started', '_date_done',]: if k in current_meta: res[k] = current_meta[k] - print([args, state, kwargs, res]) - return res class Task(celery.app.task.Task):