Ticket #1488: auth_middleware_fix.3.diff

django/contrib/auth/middleware.py

@@ -1 @@
-class UserWrapper(object):
+class LazyUser(object):
     """
-    Proxy to lazily load a user object.
+    Descriptor to lazily load a user object.
     """
-    def __init__(self, login):
-        self._login = login
-        self._cached_user = None
+    def __init__(self, user_id):
+        self.user_id = user_id
 
-    def _get_user(self):
+    def __get__(self, request, obj_type=None):
         from django.contrib.auth.models import User
-        if not self._cached_user:
-            self._cached_user = User.objects.get(pk=self._login)
-        return self._cached_user
+        request.user = User.objects.get(pk=self.user_id)
+        return request.user
 
-    _user = property(_get_user)
-
-    def __getattr__(self, name):
-        return getattr(self._user, name)
-
 class AuthenticationMiddleware:
     def process_request(self, request):
         from django.contrib.auth.models import SESSION_KEY
         assert hasattr(request, 'session'), "The Django authentication middleware requires session middleware to be installed. Edit your MIDDLEWARE_CLASSES setting to insert 'django.contrib.sessions.middleware.SessionMiddleware'."
         try:
             user_id = request.session[SESSION_KEY]
-            request.user = UserWrapper(user_id)
         except KeyError:
             from django.contrib.auth.models import AnonymousUser
             request.user = AnonymousUser()
+        else:
+            class UserRequest(request.__class__):
+                user = LazyUser(user_id)
+            request.__class__ = UserRequest
         return None