def dispatch_to_users(notification_name, eta=None, context={}, backends=None, **kwargs):
notification = get_notification(notification_name)
task_backends = notification.get_task_backends(backends)
for backend in task_backends:
users = notification.get_users(backend, context)
for user in users:
if isinstance(user, AnonymousUser):
send_anonymous_notification.apply_async(
(notification_name, backend, context),
eta=eta)
else:
user_backend = UserBackendRegistry.objects.filter(
user=user, backend=backend).first()
if user_backend:
deferred = DeferredNotification.objects.create(
notification=notification_name,
user_backend=user_backend,
context=context)
if celery_is_available():
task_id = send_notification.apply_async(
(deferred.id,),
eta=eta)
deferred.task_id = task_id
deferred.save()
else:
send_notification(deferred.id)
def send_notification(deferred_id):
deferred = ((DeferredNotification.objects
.select_related('user_backend', 'user_backend__user'))
.get(id=deferred_id))
user = deferred.user_backend.user
backend_settings = deferred.user_backend.settings
backend = get_backend(deferred.user_backend.backend)
notification = get_notification(deferred.notification)
context = notification.get_context_data(user, backend.name, **deferred.context)
backend.send_notification(
user, notification, context=context, settings=backend_settings)
deferred.ran_at = timezone.now()
deferred.save()
def send_anonymous_notification(notification_name, backend_name, context):
notification = get_notification(notification_name)
backend = get_backend(backend_name)
backend.send_notification(
AnonymousUser(), notification, context=context)
def test_loads_app_notifications(self):
notification = get_notification('test_app')
self.assertEqual(notification.__class__, TestAppNotification)
def test_get_other_notification(self):
notification = get_notification(OtherNotification.name)
self.assertEqual(notification.__class__, OtherNotification)
def test_get_test_notification(self):
notification = get_notification(TestNotification.name)
self.assertEqual(notification.__class__, TestNotification)
