extract get_current_user, fix tests

This commit is contained in:
dandds
2018-08-06 09:18:53 -04:00
parent ea5c9732ba
commit ad1e1e771b
4 changed files with 26 additions and 15 deletions

View File

@@ -8,11 +8,19 @@ def login_required(f):
@wraps(f)
def decorated_function(*args, **kwargs):
if session.get("user_id"):
g.user = Users.get(session.get("user_id"))
user = get_current_user()
if user:
g.user = user
return f(*args, **kwargs)
else:
return redirect(url_for("atst.root"))
return decorated_function
def get_current_user():
user_id = session.get("user_id")
if user_id:
return Users.get(user_id)
else:
return False