customize user regex for switch_user feature

sehmaschine · Jun 21, 2022 · 2689b38 · 2689b38
1 parent a5c6b56
commit 2689b38
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/grappelli/settings.py b/grappelli/settings.py
@@ -24,6 +24,7 @@
 # SWITCH_USER_ORIGINAL: Defines if a user is a valid target.
 # Gets a user object and returns True/False.
 SWITCH_USER_TARGET = getattr(settings, "GRAPPELLI_SWITCH_USER_TARGET", lambda original_user, user: user.is_staff and not user.is_superuser)
+SWITCH_USER_REGEX = getattr(settings, "GRAPPELLI_SWITCH_USER_REGEX", "\d+")
 
 # CLEAN INPUT TYPES
 # Replaces input types: search, email, url, tel, number, range, date

diff --git a/grappelli/urls.py b/grappelli/urls.py
@@ -4,6 +4,8 @@
 
 from .views.related import AutocompleteLookup, M2MLookup, RelatedLookup
 from .views.switch import switch_user
+from .settings import SWITCH_USER_REGEX
+
 
 urlpatterns = [
 
@@ -13,6 +15,5 @@
     re_path(r'^lookup/autocomplete/$', AutocompleteLookup.as_view(), name="grp_autocomplete_lookup"),
 
     # SWITCH USER
-    re_path(r'^switch/user/(?P<object_id>\d+)/$', switch_user, name="grp_switch_user"),
-
+    re_path(r'^switch/user/(?P<object_id>{})/$'.format(SWITCH_USER_REGEX), switch_user, name="grp_switch_user"),
 ]