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