NeoAnd/netbox
1
0
Fork 0
mirror of https://github.com/netbox-community/netbox.git synced 2025-08-08 16:48:16 -06:00
Code Issues Actions 12 Packages Projects Releases Wiki Activity

Optimize subscriber notifications

Browse Source
This commit is contained in:
Jeremy Stretch 2024-07-04 16:51:21 -04:00
parent 9f6ccec1ea
commit 2bc7b208f1
1 changed files with 9 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
netbox/extras/signals.py
View File

@ -294,24 +294,21 @@ def notify_object_changed(sender, instance, created, raw, **kwargs):
if created or raw:
return
# Skip unsupported object types
ct = ContentType.objects.get_for_model(instance)
if ct.model not in registry['model_features']['notifications'].get(ct.app_label, []):
return
# Find any Subscriptions for supported objects
subscriptions = Subscription.objects.filter(object_type=ct, object_id=instance.pk).values('user')
# Find all subscribed Users
subscribed_users = Subscription.objects.filter(object_type=ct, object_id=instance.pk).values_list('user', flat=True)
if not subscribed_users:
return
# Delete any existing Notifications for the object
subscribed_users = [sub['user'] for sub in subscriptions]
Notification.objects.filter(object_type=ct, object_id=instance.pk, user__in=subscribed_users).delete()
# Create Notifications for Subscribers
notifications = [
Notification(
user_id=sub['user'],
object=instance,
event_name=OBJECT_UPDATED
)
for sub in subscriptions
]
Notification.objects.bulk_create(notifications)
Notification.objects.bulk_create([
Notification(user_id=user, object=instance, event_name=OBJECT_UPDATED)
for user in subscribed_users
])