Signals¶
Sometimes you might want to listen to a model event upon the save, meaning, you want to do a specific action when something happens in the models.
Django for instance has this mechanism called Signals
which can be very helpful for these cases
and to perform extra operations once an action happens in your model.
Other ORMs did a similar approach to this and a fantastic one was Ormar which took the Django approach to its own implementation.
Edgy being the way it is designed, got the inspiration from both of these approaches and also
supports the Signal
from blinker. This is in blinker terminology called an anonymous signal.
What are Signals¶
Signals are a mechanism used to trigger specific actions upon a given type of event happens within the Edgy models.
The same way Django approaches signals in terms of registration, Edgy does it in the similar fashion using the blinker library.
Default signals¶
Edgy has default receivers for each model created within the ecosystem. Those can be already used out of the box by you at any time.
There are also custom signals in case you want an "extra" besides the defaults provided.
How to use them¶
The signals are inside the edgy.core.signals
and to import them, simply run:
from edgy.core.signals import (
post_delete,
post_save,
post_update,
pre_delete,
pre_save,
pre_update,
)
pre_save¶
The pre_save
is used when a model is about to be saved and triggered on Model.save()
and
Model.query.create
functions.
pre_save(send: Type["Model"], instance: "Model")
post_save¶
The post_save
is used after the model is already created and stored in the database, meaning,
when an instance already exists after save
. This signal is triggered on Model.save()
and
Model.query.create
functions.
post_save(send: Type["Model"], instance: "Model")
pre_update¶
The pre_update
is used when a model is about to receive the updates and triggered on Model.update()
and Model.query.update
functions.
pre_update(send: Type["Model"], instance: "Model")
post_update¶
The post_update
is used when a model already performed the updates and triggered on Model.update()
and Model.query.update
functions.
post_update(send: Type["Model"], instance: "Model")
pre_delete¶
The pre_delete
is used when a model is about to be deleted and triggered on Model.delete()
and Model.query.delete
functions.
pre_delete(send: Type["Model"], instance: "Model")
post_delete¶
The post_update
is used when a model is already deleted and triggered on Model.delete()
and Model.query.delete
functions.
post_update(send: Type["Model"], instance: "Model")
Receiver¶
The receiver is the function or action that you want to perform upon a signal being triggered, in other words, it is what is listening to a given event.
Let us see an example. Given the following model.
import edgy
database = edgy.Database("sqlite:///db.sqlite")
registry = edgy.Registry(database=database)
class User(edgy.Model):
id: int = edgy.BigIntegerField(primary_key=True)
name: str = edgy.CharField(max_length=255)
email: str = edgy.CharField(max_length=255)
is_verified: bool = edgy.BooleanField(default=False)
class Meta:
registry = registry
You can set a trigger to send an email to the registered user upon the creation of the record by
using the post_save
signal. The reason for the post_save
it it because the notification must
be sent after the creation of the record and not before. If it was before, the pre_save
would
be the one to use.
from edgy.core.signals import post_save
def send_notification(email: str) -> None:
"""
Sends a notification to the user
"""
send_email_confirmation(email)
@post_save.connect_via(User)
async def after_creation(sender, instance, **kwargs):
"""
Sends a notification to the user
"""
send_notification(instance.email)
As you can see, the post_save
decorator is pointing the User
model, meaning, it is "listing"
to events on that same model.
This is called receiver.
You can use any of the default signals available or even create your own custom signal.
Requirements¶
When defining your function or receiver
it must have the following requirements:
- Must be a callable.
- Must have
sender
argument as first parameter which corresponds to the model of the sending object. - Must have **kwargs argument as parameter as each model can change at any given time.
- Must be
async
because Edgy model operations are awaited.
Multiple receivers¶
What if you want to use the same receiver but for multiple models? Let us now add an extra Profile
model.
import edgy
database = edgy.Database("sqlite:///db.sqlite")
registry = edgy.Registry(database=database)
class User(edgy.Model):
id: int = edgy.BigIntegerField(primary_key=True)
name: str = edgy.CharField(max_length=255)
email: str = edgy.CharField(max_length=255)
class Meta:
registry = registry
class Profile(edgy.Model):
id: int = edgy.BigIntegerField(primary_key=True)
profile_type: str = edgy.CharField(max_length=255)
class Meta:
registry = registry
The way you define the receiver for both can simply be achieved like this:
from edgy.core.signals import post_save
def send_notification(email: str) -> None:
"""
Sends a notification to the user
"""
send_email_confirmation(email)
@post_save.connect_via(User)
@post_save.connect_via(Profile)
async def after_creation(sender, instance, **kwargs):
"""
Sends a notification to the user
"""
if isinstance(instance, User):
send_notification(instance.email)
else:
# something else for Profile
...
This way you can match and do any custom logic without the need of replicating yourself too much and keeping your code clean and consistent.
Multiple receivers for the same model¶
What if now you want to have more than one receiver for the same model? Practically you would put all in one place but you might want to do something else entirely and split those in multiple.
You can easily achieve this like this:
from edgy.core.signals import post_save
def push_notification(email: str) -> None:
# Sends a push notification
...
def send_email(email: str) -> None:
# Sends an email
...
@post_save.connect_via(User)
async def after_creation(sender, instance, **kwargs):
"""
Sends a notification to the user
"""
send_email(instance.email)
@post_save.connect_via(User)
async def do_something_else(sender, instance, **kwargs):
"""
Sends a notification to the user
"""
push_notification(instance.email)
This will make sure that every receiver will execute the given defined action.
Disconnecting receivers¶
If you wish to disconnect the receiver and stop it from running for a given model, you can also achieve this in a simple way.
from edgy.core.signals import post_save
def send_notification(email: str) -> None:
"""
Sends a notification to the user
"""
send_email_confirmation(email)
@post_save.connect_via(User)
async def after_creation(sender, instance, **kwargs):
"""
Sends a notification to the user
"""
send_notification(instance.email)
# Disconnect the given function
User.meta.signals.post_save.disconnect(after_creation)
Custom Signals¶
This is where things get interesting. A lot of time you might want to have your own Signal
and
not relying only on the default ones and this perfectly natural and common.
Edgy allows the custom signals to take place per your own design.
Let us continue with the same example of the User
model.
import edgy
database = edgy.Database("sqlite:///db.sqlite")
registry = edgy.Registry(database=database)
class User(edgy.Model):
id: int = edgy.BigIntegerField(primary_key=True)
name: str = edgy.CharField(max_length=255)
email: str = edgy.CharField(max_length=255)
is_verified: bool = edgy.BooleanField(default=False)
class Meta:
registry = registry
Now you want to have a custom signal called on_verify
specifically tailored for your User
needs
and logic.
So define it, you can simply do:
import edgy
from edgy.core import signals
from edgy import Signal
# or:
# from blinker import Signal
database = edgy.Database("sqlite:///db.sqlite")
registry = edgy.Registry(database=database)
class User(edgy.Model):
id: int = edgy.BigIntegerField(primary_key=True)
name: str = edgy.CharField(max_length=255)
email: str = edgy.CharField(max_length=255)
class Meta:
registry = registry
# Create the custom signal
User.meta.signals.on_verify = Signal()
Yes, this simple. You simply need to add a new signal on_verify
to the model signals and the
User
model from now on has a new signal ready to be used.
Danger
Keep in mind signals are class level type, which means it will affect all of the derived instances coming from it. Be mindful when creating a custom signal and its impacts.
Now you want to create a custom functionality to be listened in your new Signal.
import edgy
from edgy import Signal
# or:
# from blinker import Signal
database = edgy.Database("sqlite:///db.sqlite")
registry = edgy.Registry(database=database)
class User(edgy.Model):
id: int = edgy.BigIntegerField(primary_key=True)
name: str = edgy.CharField(max_length=255)
email: str = edgy.CharField(max_length=255)
class Meta:
registry = registry
# Create the custom signal
User.meta.signals.on_verify = Signal()
# Create the receiver
async def trigger_notifications(sender, instance, **kwargs):
"""
Sends email and push notification
"""
send_email(instance.email)
send_push_notification(instance.email)
# Register the receiver into the new Signal.
User.meta.signals.on_verify.connect(trigger_notifications)
Now not only you created the new receiver trigger_notifications
but also connected it to the
the new on_verify
signal.
Rewire signals¶
To not call the default lifecycle signals you can overwrite them per class.
You can either overwrite some or use the set_lifecycle_signals_from
method of the Broadcaster (signals)
This can be used to not call the default lifecycle signals in signals but custom ones or to use namespaces.
Lifecycle methods are the former mentioned signals `
import edgy
from edgy import Signal
# or:
# from blinker import Signal
database = edgy.Database("sqlite:///db.sqlite")
registry = edgy.Registry(database=database)
class User(edgy.Model):
id: int = edgy.BigIntegerField(primary_key=True)
name: str = edgy.CharField(max_length=255)
email: str = edgy.CharField(max_length=255)
class Meta:
registry = registry
# Overwrite a model lifecycle Signal; this way the main signals.pre_delete is not triggered
User.meta.signals.pre_delete = Signal()
# Update all lifecyle signals. Replace pre_delete again with the default
User.meta.signals.set_lifecycle_signals_from(signals)
How to use it¶
Now it is time to use the signal in a custom logic, after all it was created to make sure it is custom enough for the needs of the business logic.
For simplification, the example below will be a very simple logic.
async def create_user(**kwargs):
"""
Creates a user
"""
await User.query.create(**kwargs)
async def is_verified_user(id: int):
"""
Checks if user is verified and sends notification
if true.
"""
user = await User.query.get(pk=id)
if user.is_verified:
# triggers the custom signal
await User.meta.signals.on_verify.send_async(User, instance=user)
As you can see, the on_verify
, it is only triggered if the user is verified and not anywhere else.
Disconnect the signal¶
The process of disconnecting the signal is exactly the same as before.
async def trigger_notifications(sender, instance, **kwargs):
"""
Sends email and push notification
"""
send_email(instance.email)
send_push_notification(instance.email)
# Disconnect the given function
User.meta.signals.on_verify.disconnect(trigger_notifications)