-
Notifications
You must be signed in to change notification settings - Fork 2
/
Copy pathmiddlewares.py
36 lines (29 loc) · 1.1 KB
/
middlewares.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
from aiohttp import web
from aiohttp_session import get_session
from config.settings import *
async def db_handler(app, handler):
async def middleware(request):
if request.path.startswith('/static/') or request.path.startswith('/_debugtoolbar'):
response = await handler(request)
return response
request.db = app.db
response = await handler(request)
return response
return middleware
async def authorize(app, handler):
async def middleware(request):
def check_path(path):
result = True
for r in ['/login', '/login-api' '/img', '/static/', '/signout', '/_debugtoolbar/', '/api']:
if path.startswith(r) or path == '/':
result = False
return result
session = await get_session(request)
if session.get("user"):
return await handler(request)
elif check_path(request.path):
url = request.app.router['login'].url()
raise web.HTTPFound(url)
else:
return await handler(request)
return middleware