python/aiohttp-security
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Got safely request argument

Browse Source
This commit is contained in:
Pedro Crespo-Valero
2018-09-06 03:43:12 +02:00
parent ca38b38a0e
commit 9a4d8fb1ad
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
aiohttp_security/api.py
View File

@@ -94,7 +94,7 @@ def login_required(fn):
""" """
@wraps(fn) @wraps(fn)
async def wrapped(*args, **kwargs): async def wrapped(*args, **kwargs):
request = args[-1] request = kwargs.get('request', args[-1] if args else None)
if not isinstance(request, web.BaseRequest): if not isinstance(request, web.BaseRequest):
msg = ("Incorrect decorator usage. " msg = ("Incorrect decorator usage. "
"Expecting `def handler(request)` " "Expecting `def handler(request)` "
@@ -125,7 +125,7 @@ def has_permission(
def wrapper(fn): def wrapper(fn):
@wraps(fn) @wraps(fn)
async def wrapped(*args, **kwargs): async def wrapped(*args, **kwargs):
request = args[-1] request = kwargs.get('request', args[-1] if args else None)
if not isinstance(request, web.BaseRequest): if not isinstance(request, web.BaseRequest):
msg = ("Incorrect decorator usage. " msg = ("Incorrect decorator usage. "
"Expecting `def handler(request)` " "Expecting `def handler(request)` "