sbt-idp/cope2n-api/fwd_api/celery_worker/task_warpper.py

16 lines
646 B
Python
Raw Normal View History

2024-06-26 07:58:24 +00:00
from celery import Task
from celery.utils.log import get_task_logger
logger = get_task_logger(__name__)
class VerboseTask(Task):
abstract = True
def on_failure(self, exc, task_id, args, kwargs, einfo):
# Task failed. What do you want to do?
logger.error('FAILURE: Task raised an exception: {}'.format(exc))
def on_success(self, retval, task_id, args, kwargs):
logger.info(f"SUCCESS: Task: {task_id} succeeded | retval: {retval} | args: {args} | kwargs: {kwargs}")
def before_start(self, task_id, args, kwargs):
logger.info(f"BEFORE_START: Task: {task_id} | args: {args} | kwargs: {kwargs}")