src/ldt/ldt/security/middleware.py
changeset 278 d16ec14aaf29
parent 265 491d057cbfd2
child 350 c6953232099f
--- a/src/ldt/ldt/security/middleware.py	Mon Dec 12 12:57:38 2011 +0100
+++ b/src/ldt/ldt/security/middleware.py	Mon Dec 12 14:48:57 2011 +0100
@@ -1,17 +1,17 @@
-from ldt.security.utils import protect_models, unprotect_models, _thread_locals
+from ldt.security.utils import protect_models, unprotect_models, _thread_locals, set_current_user, del_current_user
 
 class SecurityMiddleware(object):
     
     def process_request(self, request):
         if not hasattr(_thread_locals, 'user'):
-            _thread_locals.user = request.user
+            set_current_user(request.user)
             protect_models()
     
     def process_response(self, request, response):
         
         if hasattr(_thread_locals, 'user'):            
             unprotect_models()
-            del _thread_locals.user
+            del_current_user()
                 
         return response
     
\ No newline at end of file