implement authentication redirect

This commit is contained in:
dandds
2018-08-03 15:51:44 -04:00
parent 9937b77c74
commit ea5c9732ba
6 changed files with 29 additions and 10 deletions

18
atst/domain/auth.py Normal file
View File

@@ -0,0 +1,18 @@
from functools import wraps
from flask import g, request, redirect, url_for, session
from atst.domain.users import Users
def login_required(f):
@wraps(f)
def decorated_function(*args, **kwargs):
if session.get("user_id"):
g.user = Users.get(session.get("user_id"))
return f(*args, **kwargs)
else:
return redirect(url_for("atst.root"))
return decorated_function