Return back to passing response parameter into remember and forget methods

aiohttp_security/abc.py

@@ -15,10 +15,10 @@ class AbstractIdentityPolicy(metaclass=abc.ABCMeta):
 
     @asyncio.coroutine
     @abc.abstractmethod
-    def remember(self, request, identity, **kwargs):
+    def remember(self, request, response, identity, **kwargs):
         """Remember identity.
 
-        Return MultiDict with headers on this request's response.
+        Modify response object by filling it's headers with remembered user.
 
         An individual identity policy and its consumers can decide on
         the composition and meaning of **kw.
@@ -27,8 +27,8 @@ class AbstractIdentityPolicy(metaclass=abc.ABCMeta):
 
     @asyncio.coroutine
     @abc.abstractmethod
-    def forget(self, request):
-        """ Modify request.response which can be used to 'forget' the
+    def forget(self, request, response):
+        """ Modify response which can be used to 'forget' the
         current identity on subsequent requests."""
         pass
 

aiohttp_security/api.py

@@ -7,17 +7,15 @@ AUTZ_KEY = 'aiohttp_security_autz_policy'
 
 
 @asyncio.coroutine
-def remember(request, identity, **kwargs):
+def remember(request, response, identity, **kwargs):
     identity_policy = request.app[IDENTITY_KEY]
-    headers = yield from identity_policy.remember(request, identity, **kwargs)
-    return headers
+    yield from identity_policy.remember(request, response, identity, **kwargs)
 
 
 @asyncio.coroutine
-def forget(request):
+def forget(request, response):
     identity_policy = request.app[IDENTITY_KEY]
-    headers = yield from identity_policy.forget(request)
-    return headers
+    yield from identity_policy.forget(request, response)
 
 
 @asyncio.coroutine

aiohttp_security/cookies_identity.py

@@ -6,6 +6,9 @@ from aiohttp import hdrs, CIMultiDict
 from .abc import AbstractIdentityPolicy
 
 
+sentinel = object()
+
+
 class CookiesIdentityPolicy(AbstractIdentityPolicy):
 
     def __init__(self):
@@ -18,23 +21,12 @@ class CookiesIdentityPolicy(AbstractIdentityPolicy):
         return identity
 
     @asyncio.coroutine
-    def remember(self, request, identity, **kwargs):
-        cookies = http.cookies.SimpleCookie()
-        max_age = kwargs.pop('max_age', self._max_age)
-        cookies[self._cookie_name] = identity
-        cookie = cookies[self._cookie_name]
-        cookie['max-age'] = max_age
-        cookie.update(kwargs)
-
-        value = cookie.output(header='')[1:]
-        result = CIMultiDict({hdrs.SET_COOKIE: value})
-        return result
+    def remember(self, request, response, identity, max_age=sentinel,
+                 **kwargs):
+        if max_age is sentinel:
+            max_age = self._max_age
+        response.set_cookie(self._cookie_name, max_age=max_age, **kwargs)
 
     @asyncio.coroutine
-    def forget(self, request):
-        cookies = http.cookies.SimpleCookie()
-        cookies[self._cookie_name] = ''
-        cookie = cookies[self._cookie_name]
-        value = cookie.output(header='')[1:]
-        result = CIMultiDict({hdrs.SET_COOKIE: value})
-        return result
+    def forget(self, request, response):
+        response.del_cookie(self._cookie_name)