aiohttp_session not required

This commit is contained in:
Alex Kuzmenko 2017-01-23 04:10:35 +02:00
parent 131ee39bf6
commit fd863dc229
1 changed files with 9 additions and 1 deletions

View File

@ -6,7 +6,11 @@ to configure aiohttp_session properly.
import asyncio import asyncio
from aiohttp_session import get_session try:
from aiohttp_session import get_session
has_aiohttp_session = True
except ImportError:
has_aiohttp_session = False
from .abc import AbstractIdentityPolicy from .abc import AbstractIdentityPolicy
@ -16,6 +20,10 @@ class SessionIdentityPolicy(AbstractIdentityPolicy):
def __init__(self, session_key='AIOHTTP_SECURITY'): def __init__(self, session_key='AIOHTTP_SECURITY'):
self._session_key = session_key self._session_key = session_key
if not has_aiohttp_session:
raise ImportError(
'SessionIdentityPolicy requires aiohttp_session')
@asyncio.coroutine @asyncio.coroutine
def identify(self, request): def identify(self, request):
session = yield from get_session(request) session = yield from get_session(request)