LOGIN page functionality

src/dj_tracker/constants.py

@@ -34,6 +34,7 @@ def _set_dj_tracker_settings():
         "APPS_TO_EXCLUDE": (),
         "IGNORE_MODULES": (),
         "IGNORE_PATHS": (),
+        "LOGIN": False,
     }
     DJ_TRACKER_SETTINGS.update(getattr(settings, "DJ_TRACKER", {}))
 

src/dj_tracker/urlpatterns.py

@@ -0,0 +1,17 @@
+from functools import update_wrapper
+
+
+def decorate_urlpatterns(urlpatterns, decorator):
+    """Decorate all the views in the passed urlpatterns list with the given decorator"""
+    for pattern in urlpatterns:
+        if hasattr(pattern, "url_patterns"):
+            # this is an included RegexURLResolver; recursively decorate the views
+            # contained in it
+            decorate_urlpatterns(pattern.url_patterns, decorator)
+
+        if getattr(pattern, "callback", None):
+            pattern.callback = update_wrapper(
+                decorator(pattern.callback), pattern.callback
+            )
+
+    return urlpatterns

src/dj_tracker/urls.py

@@ -1,6 +1,9 @@
+from django.contrib.admin.views.decorators import staff_member_required
 from django.urls import path, register_converter
 
 from dj_tracker import views
+from dj_tracker.constants import DJ_TRACKER_SETTINGS
+from dj_tracker.urlpatterns import decorate_urlpatterns
 
 
 class CacheKeyConverter:
@@ -36,3 +39,5 @@ def to_url(self, value):
         name="query-group",
     ),
 ]
+if DJ_TRACKER_SETTINGS["LOGIN"]:
+    url_patterns = decorate_urlpatterns(urlpatterns, staff_member_required)