From 9a4d8fb1ad4585513502f9b94be7808d1a4245a7 Mon Sep 17 00:00:00 2001 From: Pedro Crespo-Valero Date: Thu, 6 Sep 2018 03:43:12 +0200 Subject: [PATCH] Got safely request argument --- aiohttp_security/api.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/aiohttp_security/api.py b/aiohttp_security/api.py index d414459..0d6286c 100644 --- a/aiohttp_security/api.py +++ b/aiohttp_security/api.py @@ -94,7 +94,7 @@ def login_required(fn): """ @wraps(fn) async def wrapped(*args, **kwargs): - request = args[-1] + request = kwargs.get('request', args[-1] if args else None) if not isinstance(request, web.BaseRequest): msg = ("Incorrect decorator usage. " "Expecting `def handler(request)` " @@ -125,7 +125,7 @@ def has_permission( def wrapper(fn): @wraps(fn) async def wrapped(*args, **kwargs): - request = args[-1] + request = kwargs.get('request', args[-1] if args else None) if not isinstance(request, web.BaseRequest): msg = ("Incorrect decorator usage. " "Expecting `def handler(request)` "