signals.py 1.5 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849
  1. from functools import partial
  2. from django.db.models.utils import make_model_tuple
  3. from django.dispatch import Signal
  4. class_prepared = Signal()
  5. class ModelSignal(Signal):
  6. """
  7. Signal subclass that allows the sender to be lazily specified as a string
  8. of the `app_label.ModelName` form.
  9. """
  10. def _lazy_method(self, method, apps, receiver, sender, **kwargs):
  11. from django.db.models.options import Options
  12. # This partial takes a single optional argument named "sender".
  13. partial_method = partial(method, receiver, **kwargs)
  14. if isinstance(sender, str):
  15. apps = apps or Options.default_apps
  16. apps.lazy_model_operation(partial_method, make_model_tuple(sender))
  17. else:
  18. return partial_method(sender)
  19. def connect(self, receiver, sender=None, weak=True, dispatch_uid=None, apps=None):
  20. self._lazy_method(
  21. super().connect, apps, receiver, sender,
  22. weak=weak, dispatch_uid=dispatch_uid,
  23. )
  24. def disconnect(self, receiver=None, sender=None, dispatch_uid=None, apps=None):
  25. return self._lazy_method(
  26. super().disconnect, apps, receiver, sender, dispatch_uid=dispatch_uid
  27. )
  28. pre_init = ModelSignal(use_caching=True)
  29. post_init = ModelSignal(use_caching=True)
  30. pre_save = ModelSignal(use_caching=True)
  31. post_save = ModelSignal(use_caching=True)
  32. pre_delete = ModelSignal(use_caching=True)
  33. post_delete = ModelSignal(use_caching=True)
  34. m2m_changed = ModelSignal(use_caching=True)
  35. pre_migrate = Signal()
  36. post_migrate = Signal()