allows get_user_by_username to resolve user using a custom model

BawsHuman · Aug 12, 2024 · 81bdec4 · 81bdec4
1 parent c4bb8fb
commit 81bdec4
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/apps/accounts/utils/general.py b/apps/accounts/utils/general.py
@@ -133,12 +133,14 @@ def email_to_username_pretty_unique(email):
     return username
 
 
-def get_user_by_username(username):
+def get_user_by_username(username, UserModel=None):
     """Gets a user by `username`
 
     Returns None if not found
     """
-    UserModel = get_user_model()
+    if UserModel is None:
+        UserModel = get_user_model()
+
     try:
         user = UserModel.objects.get(username=username)
     except UserModel.DoesNotExist: