From 86962a7c967cdd8cdcbe5b4274e7deefee703363 Mon Sep 17 00:00:00 2001 From: Alex Kuzmenko Date: Mon, 23 Jan 2017 04:10:35 +0200 Subject: [PATCH] aiohttp_session not required --- aiohttp_security/session_identity.py | 10 +++++++++- 1 file changed, 9 insertions(+), 1 deletion(-) diff --git a/aiohttp_security/session_identity.py b/aiohttp_security/session_identity.py index 5fdf55a..69e1563 100644 --- a/aiohttp_security/session_identity.py +++ b/aiohttp_security/session_identity.py @@ -6,7 +6,11 @@ to configure aiohttp_session properly. 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 @@ -16,6 +20,10 @@ class SessionIdentityPolicy(AbstractIdentityPolicy): def __init__(self, session_key='AIOHTTP_SECURITY'): self._session_key = session_key + if not has_aiohttp_session: + raise ImportError( + 'SessionIdentityPolicy requires aiohttp_session') + @asyncio.coroutine def identify(self, request): session = yield from get_session(request)