Source code for tortoise.signals

from enum import Enum
from typing import Callable

Signals = Enum("Signals", ["pre_save", "post_save", "pre_delete", "post_delete"])


[docs]def post_save(*senders) -> Callable: """ Register given models post_save signal. :param senders: Model class """ def decorator(f): for sender in senders: sender.register_listener(Signals.post_save, f) return f return decorator
[docs]def pre_save(*senders) -> Callable: """ Register given models pre_save signal. :param senders: Model class """ def decorator(f): for sender in senders: sender.register_listener(Signals.pre_save, f) return f return decorator
[docs]def pre_delete(*senders) -> Callable: """ Register given models pre_delete signal. :param senders: Model class """ def decorator(f): for sender in senders: sender.register_listener(Signals.pre_delete, f) return f return decorator
[docs]def post_delete(*senders) -> Callable: """ Register given models post_delete signal. :param senders: Model class """ def decorator(f): for sender in senders: sender.register_listener(Signals.post_delete, f) return f return decorator