From aef24fae3df2893533e1776f88643bb162433023 Mon Sep 17 00:00:00 2001 From: Alex Kuzmenko Date: Tue, 17 Oct 2017 12:32:53 +0300 Subject: [PATCH] `aiohttp-session` is not required (#107) * aiohttp_session is not required * fix code coverage * has_aiohttp_session -> HAS_AIOHTTP_SESSION --- 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..5d9cea0 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: # pragma: no cover + 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: # pragma: no cover + raise ImportError( + 'SessionIdentityPolicy requires `aiohttp_session`') + @asyncio.coroutine def identify(self, request): session = yield from get_session(request)