mirror of
https://github.com/django-q2/django-q2.git
synced 2026-09-25 15:38:12 +08:00
Ilan Steemers
f8cd136668
* adds post_execute signal and tests * Updates signal docts * Fixes typo in docs
43 lines
1.1 KiB
Python
43 lines
1.1 KiB
Python
import importlib
|
|
|
|
from django.db.models.signals import post_save
|
|
from django.dispatch import Signal, receiver
|
|
from django.utils.translation import gettext_lazy as _
|
|
|
|
from django_q.conf import logger
|
|
from django_q.models import Task
|
|
|
|
|
|
@receiver(post_save, sender=Task)
|
|
def call_hook(sender, instance, **kwargs):
|
|
if instance.hook:
|
|
f = instance.hook
|
|
if not callable(f):
|
|
try:
|
|
module, func = f.rsplit(".", 1)
|
|
m = importlib.import_module(module)
|
|
f = getattr(m, func)
|
|
except (ValueError, ImportError, AttributeError):
|
|
logger.error(
|
|
_(f"malformed return hook '{instance.hook}' for [{instance.name}]")
|
|
)
|
|
return
|
|
try:
|
|
f(instance)
|
|
except Exception as e:
|
|
logger.error(
|
|
_(
|
|
f"return hook {instance.hook} failed on [{instance.name}] because {str(e)}"
|
|
)
|
|
)
|
|
|
|
|
|
# args: task
|
|
pre_enqueue = Signal()
|
|
|
|
# args: func, task
|
|
pre_execute = Signal()
|
|
|
|
# args: task
|
|
post_execute = Signal()
|