Explorar o código

Simplified AuthenticationMiddleware a bit.

SimpleLazyObject already caches value in _wrapped.
Sergey Fedoseev %!s(int64=5) %!d(string=hai) anos
pai
achega
2f010795e6
Modificáronse 1 ficheiros con 3 adicións e 7 borrados
  1. 3 7
      django/contrib/auth/middleware.py

+ 3 - 7
django/contrib/auth/middleware.py

@@ -1,3 +1,5 @@
+from functools import partial
+
 from django.contrib import auth
 from django.contrib.auth import load_backend
 from django.contrib.auth.backends import RemoteUserBackend
@@ -6,12 +8,6 @@ from django.utils.deprecation import MiddlewareMixin
 from django.utils.functional import SimpleLazyObject
 
 
-def get_user(request):
-    if not hasattr(request, '_cached_user'):
-        request._cached_user = auth.get_user(request)
-    return request._cached_user
-
-
 class AuthenticationMiddleware(MiddlewareMixin):
     def process_request(self, request):
         assert hasattr(request, 'session'), (
@@ -20,7 +16,7 @@ class AuthenticationMiddleware(MiddlewareMixin):
             "'django.contrib.sessions.middleware.SessionMiddleware' before "
             "'django.contrib.auth.middleware.AuthenticationMiddleware'."
         )
-        request.user = SimpleLazyObject(lambda: get_user(request))
+        request.user = SimpleLazyObject(partial(auth.get_user, request))
 
 
 class RemoteUserMiddleware(MiddlewareMixin):