Add login screen-only notifications

manager/templatetags/notifications.py

@@ -1,10 +1,11 @@
 from django import template
 from manager.models import Notification
 from django.utils import timezone
+from django.db.models import Q
 
 register = template.Library()
 
 @register.inclusion_tag('manager/notifications.html', takes_context=True)
-def getNotifications(context):
-    return { 'notifications': Notification.objects.filter(expiry__gte = timezone.now()), 'context': context } # TODO
+def getNotifications(context, login=False):
+    return { 'notifications': [notification for notification in Notification.objects.filter(Q(expiry__gte = timezone.now()) | Q(expiry = None)) if (notification.on_login or (not login)) ] } 
 

templates/two_factor/core/login.html

@@ -18,7 +18,7 @@
     </div>
 </noscript>
 
-{% getNotifications %}
+{% getNotifications True %}
 
   {% if wizard.steps.current == 'auth' %}
     {% if user.is_authenticated %}