1
0
Fork 0

Use signal for creating group invite notification

This commit is contained in:
Mouse Reeve 2022-07-04 20:24:29 -07:00
parent 03f5a3f2c1
commit aeefd5a3e9
2 changed files with 16 additions and 13 deletions

View file

@ -2,7 +2,7 @@
from django.db import models, transaction
from django.dispatch import receiver
from .base_model import BookWyrmModel
from . import Boost, Favorite, ImportJob, Report, Status, User
from . import Boost, Favorite, GroupMemberInvitation, ImportJob, Report, Status, User
class Notification(BookWyrmModel):
@ -205,3 +205,16 @@ def notify_admins_on_report(sender, instance, *args, **kwargs):
unread=True,
)
notification.related_reports.add(instance)
@receiver(models.signals.post_save, sender=GroupMemberInvitation)
@transaction.atomic
# pylint: disable=unused-argument
def notify_user_on_group_invite(sender, instance, *args, **kwargs):
"""Cool kids club here we come"""
Notification.notify(
instance.user,
instance.group.user,
related_group=instance.group,
notification_type=Notification.INVITE,
)