78 lines
2.5 KiB
Python
78 lines
2.5 KiB
Python
from textwrap import dedent
|
|
|
|
from aiohttp import web
|
|
|
|
from aiohttp_security import (
|
|
remember, forget, authorized_userid,
|
|
has_permission, login_required,
|
|
)
|
|
|
|
from .db_auth import check_credentials
|
|
|
|
|
|
class Web(object):
|
|
index_template = dedent("""
|
|
<!doctype html>
|
|
<head></head>
|
|
<body>
|
|
<p>{message}</p>
|
|
<form action="/login" method="post">
|
|
Login:
|
|
<input type="text" name="login">
|
|
Password:
|
|
<input type="password" name="password">
|
|
<input type="submit" value="Login">
|
|
</form>
|
|
<a href="/logout">Logout</a>
|
|
</body>
|
|
""")
|
|
|
|
async def index(self, request):
|
|
username = await authorized_userid(request)
|
|
if username:
|
|
template = self.index_template.format(
|
|
message='Hello, {username}!'.format(username=username))
|
|
else:
|
|
template = self.index_template.format(message='You need to login')
|
|
response = web.Response(body=template.encode())
|
|
return response
|
|
|
|
async def login(self, request):
|
|
response = web.HTTPFound('/')
|
|
form = await request.post()
|
|
login = form.get('login')
|
|
password = form.get('password')
|
|
db_engine = request.app.db_engine
|
|
if await check_credentials(db_engine, login, password):
|
|
await remember(request, response, login)
|
|
return response
|
|
|
|
return web.HTTPUnauthorized(
|
|
body=b'Invalid username/password combination')
|
|
|
|
@login_required
|
|
async def logout(self, request):
|
|
response = web.Response(body=b'You have been logged out')
|
|
await forget(request, response)
|
|
return response
|
|
|
|
@has_permission('public')
|
|
async def internal_page(self, request):
|
|
response = web.Response(
|
|
body=b'This page is visible for all registered users')
|
|
return response
|
|
|
|
@has_permission('protected')
|
|
async def protected_page(self, request):
|
|
response = web.Response(body=b'You are on protected page')
|
|
return response
|
|
|
|
def configure(self, app):
|
|
router = app.router
|
|
router.add_route('GET', '/', self.index, name='index')
|
|
router.add_route('POST', '/login', self.login, name='login')
|
|
router.add_route('GET', '/logout', self.logout, name='logout')
|
|
router.add_route('GET', '/public', self.internal_page, name='public')
|
|
router.add_route('GET', '/protected', self.protected_page,
|
|
name='protected')
|