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