aiohttp-security/aiohttp_security/cookies_identity.py

31 lines
797 B
Python
Raw Normal View History

2015-07-08 17:30:24 +00:00
import asyncio
from .abc import AbstractIdentityPolicy
sentinel = object()
2015-07-08 17:30:24 +00:00
class CookiesIdentityPolicy(AbstractIdentityPolicy):
def __init__(self):
self._cookie_name = 'AIOHTTP_SECURITY'
self._max_age = 30 * 24 * 3600
@asyncio.coroutine
def identify(self, request):
identity = request.cookies.get(self._cookie_name)
return identity
@asyncio.coroutine
def remember(self, request, response, identity, max_age=sentinel,
**kwargs):
if max_age is sentinel:
max_age = self._max_age
2015-07-29 20:58:55 +00:00
response.set_cookie(self._cookie_name, identity,
max_age=max_age, **kwargs)
2015-07-08 17:30:24 +00:00
@asyncio.coroutine
def forget(self, request, response):
response.del_cookie(self._cookie_name)