sbt-idp/cope2n-api/fwd_api/celery_worker/task_warpper.py
2024-06-26 16:40:33 +07:00

16 lines
688 B
Python

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(f'FAILURE: Task: {self.name} - {task_id} | Task raised an exception: {exc}')
def on_success(self, retval, task_id, args, kwargs):
logger.info(f"SUCCESS: Task: {self.name} - {task_id} | retval: {retval} | args: {args} | kwargs: {kwargs}")
def before_start(self, task_id, args, kwargs):
logger.info(f"BEFORE_START: Task: {self.name} - {task_id} | args: {args} | kwargs: {kwargs}")