Ticket #2630: backends_custom_auth.diff

backends.py

@@ +1 @@
+from django.db import models
 from django.contrib.auth.models import User
 
+from django.conf import settings
+AUTH_MODULE_HINT = settings.AUTH_MODULE_HINT
+DEBUG = settings.DEBUG
+
+
+def get_auth_module(module_name):
+    """
+    Return a user-defined module for authentication.
+    """
+    # cls is the last component 
+    tmp = module_name.split('.')
+    cls = tmp.pop()
+    if DEBUG: print "Class: ", cls
+    module_name = ".".join(tmp)
+    if DEBUG: print "module name: %s" % module_name
+    mod = __import__(module_name, globals(), locals(), [''])
+    auth_module = getattr(mod, cls)
+    return auth_module
+    
+
 class ModelBackend:
     """
     Authenticate against django.contrib.auth.models.User
@@ -7 +28 @@
     # TODO: Model, login attribute name and password attribute name should be
     # configurable.
     def authenticate(self, username=None, password=None):
+        # get the custom user model 
+        auth_module = get_auth_module(AUTH_MODULE_HINT)
         try:
-            user = User.objects.get(username=username)
+            # retrieve the user object
+            user = auth_module.objects.get(username=username)
+            # check_password is a method inherited from User
             if user.check_password(password):
                 return user
-        except User.DoesNotExist:
+        except auth_module.DoesNotExist:
+            # fallback on the default method
+            try:
+                user = User.objects.get(username=username)
+                if user.check_password(password):
+                   return user
+            except User.DoesNotExist:
+                return None
+        else:
             return None
-
+            
     def get_user(self, user_id):
         try:
-            return User.objects.get(pk=user_id)
+           return User.objects.get(pk=user_id)
         except User.DoesNotExist:
-            return None
+           return None
+
+