Revision 47097c8a
b/snf-astakos-app/astakos/im/cookie.py | ||
---|---|---|
45 | 45 |
logger = logging.getLogger(__name__) |
46 | 46 |
|
47 | 47 |
class Cookie(): |
48 |
def __init__(self, request): |
|
48 |
def __init__(self, request, response):
|
|
49 | 49 |
cookies = getattr(request, 'COOKIES', {}) |
50 | 50 |
cookie = unquote(cookies.get(COOKIE_NAME, '')) |
51 | 51 |
self.email, sep, self.auth_token = cookie.partition('|') |
52 | 52 |
self.request = request |
53 |
self.response = response |
|
53 | 54 |
|
54 | 55 |
@property |
55 | 56 |
def is_set(self): |
... | ... | |
80 | 81 |
msg = 'Cookie deleted for %(email)s' % self.__dict__ |
81 | 82 |
logger._log(LOGGING_LEVEL, msg, []) |
82 | 83 |
|
83 |
def fix(self, response):
|
|
84 |
def fix(self): |
|
84 | 85 |
if self.user.is_authenticated(): |
85 | 86 |
if not self.is_set or not self.is_valid: |
86 |
self.__set(response) |
|
87 |
self.__set(self.response)
|
|
87 | 88 |
else: |
88 | 89 |
if self.is_set: |
89 |
self.__delete(response) |
|
90 |
self.__delete(self.response) |
b/snf-astakos-app/astakos/im/middleware.py | ||
---|---|---|
35 | 35 |
|
36 | 36 |
class CookieAuthenticationMiddleware(object): |
37 | 37 |
def process_response(self, request, response): |
38 |
Cookie(request).fix(response)
|
|
38 |
Cookie(request, response).fix()
|
|
39 | 39 |
return response |
Also available in: Unified diff