aiohttp-security/aiohttp_security/api.py

77 lines
2.8 KiB
Python
Raw Normal View History

2015-07-08 17:30:24 +00:00
import asyncio
from aiohttp import web
2015-07-08 17:30:24 +00:00
from aiohttp_security.abc import (AbstractIdentityPolicy,
AbstractAuthorizationPolicy)
IDENTITY_KEY = 'aiohttp_security_identity_policy'
AUTZ_KEY = 'aiohttp_security_autz_policy'
@asyncio.coroutine
def remember(request, response, identity, **kwargs):
2015-10-29 08:31:24 +00:00
"""Remember identity into response.
2015-10-29 13:34:38 +00:00
The action is performed by identity_policy.remember()
Usually the idenity is stored in user cookies homehow but may be
pushed into custom header also.
2015-10-29 08:31:24 +00:00
"""
assert isinstance(identity, str), identity
identity_policy = request.app.get(IDENTITY_KEY)
if identity_policy is None:
text = ("Security subsystem is not initialized, "
"call aiohttp_security.setup(...) first")
# in order to see meaningful exception message both: on console
# output and rendered page we add same message to *reason* and
# *text* arguments.
raise web.HTTPInternalServerError(reason=text, text=text)
yield from identity_policy.remember(request, response, identity, **kwargs)
2015-07-08 17:30:24 +00:00
@asyncio.coroutine
def forget(request, response):
"""Forget previously remembered identity.
2015-11-02 20:28:10 +00:00
2015-10-29 13:34:38 +00:00
Usually it clears cookie or server-side storage to forget user
session.
"""
identity_policy = request.app.get(IDENTITY_KEY)
if identity_policy is None:
text = ("Security subsystem is not initialized, "
"call aiohttp_security.setup(...) first")
# in order to see meaningful exception message both: on console
# output and rendered page we add same message to *reason* and
# *text* arguments.
raise web.HTTPInternalServerError(reason=text, text=text)
yield from identity_policy.forget(request, response)
2015-07-08 17:30:24 +00:00
@asyncio.coroutine
def authorized_userid(request):
identity_policy = request.app.get(IDENTITY_KEY)
autz_policy = request.app.get(AUTZ_KEY)
if identity_policy is None or autz_policy is None:
return None
2015-07-08 17:30:24 +00:00
identity = yield from identity_policy.identify(request)
user_id = yield from autz_policy.authorized_userid(identity)
return user_id
@asyncio.coroutine
def permits(request, permission, context=None):
identity_policy = request.app.get(IDENTITY_KEY)
autz_policy = request.app.get(AUTZ_KEY)
if identity_policy is None or autz_policy is None:
return True
2015-07-08 17:30:24 +00:00
identity = yield from identity_policy.identify(request)
access = yield from autz_policy.permits(identity, permission, context)
return access
2015-09-06 05:12:18 +00:00
def setup(app, identity_policy, autz_policy):
2015-07-08 17:30:24 +00:00
assert isinstance(identity_policy, AbstractIdentityPolicy), identity_policy
2015-09-06 05:12:18 +00:00
assert isinstance(autz_policy, AbstractAuthorizationPolicy), autz_policy
2015-07-08 17:30:24 +00:00
app[IDENTITY_KEY] = identity_policy
2015-09-06 05:12:18 +00:00
app[AUTZ_KEY] = autz_policy