add check attr in request; remove aiohttp_session module;

aio-libs · Oct 26, 2016 · b65ed78 · b65ed78
1 parent 5237eb7
commit b65ed78
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/aiohttp_debugtoolbar/panels/request_vars.py b/aiohttp_debugtoolbar/panels/request_vars.py
@@ -1,5 +1,4 @@
 import asyncio
-from aiohttp_session import get_session
 from pprint import saferepr
 
 from .base import DebugPanel
@@ -37,8 +36,9 @@ def process_response(self, response):
         # TODO: think about aiohttp_security
 
         # session to separate table
-        session = yield from get_session(request)
-        if not session.empty:
-            data.update({
-                'session': [(k, session[k]) for k in session],
-            })
+        if hasattr(request, 'session'):
+            session = request.session
+            if not session.empty:
+                data.update({
+                    'session': [(k, session[k]) for k in session],
+                })