# from __future__ import absolute_import, unicode_literals
from celery import Task
from celery import shared_task
import logging
from django.conf import settings

# 实例化myLogger
myLogger = logging.getLogger('log')


class CustomTask(Task):
    """
    自定义的任务回调
    """

    def on_failure(self, exc, task_id, args, kwargs, einfo):
        myLogger.error('{0!r} failed: {1!r}'.format(task_id, exc))
        return super().on_failure(exc, task_id, args, kwargs, einfo)


@shared_task(base=CustomTask)
def send_mail_task(**args):
    from django.core.mail import send_mail
    args['subject'] = args.get('subject', 'ehs_500')
    args['from_email'] = args.get('from_email', settings.EMAIL_HOST_USER)
    send_mail(**args)