equal
deleted
inserted
replaced
1 from ldt.security.utils import protect_models, unprotect_models, _thread_locals, set_current_user, del_current_user |
1 from ldt.security import set_current_user, del_current_user, _thread_locals |
2 |
2 |
3 class SecurityMiddleware(object): |
3 class SecurityMiddleware(object): |
4 |
4 |
5 def process_request(self, request): |
5 def process_request(self, request): |
6 if not hasattr(_thread_locals, 'user'): |
6 if not hasattr(_thread_locals, 'user'): |
7 set_current_user(request.user) |
7 set_current_user(request.user) |
8 protect_models() |
|
9 |
8 |
10 def process_response(self, request, response): |
9 def process_response(self, request, response): |
11 |
10 |
12 if hasattr(_thread_locals, 'user'): |
11 if hasattr(_thread_locals, 'user'): |
13 unprotect_models() |
|
14 del_current_user() |
12 del_current_user() |
15 |
13 |
16 return response |
14 return response |
17 |
15 |