Updated the whoami-flask example

This commit is contained in:
Joshua Milas
2017-09-11 12:16:17 -04:00
parent 65cad5efad
commit 73564b97ea

View File

@@ -28,8 +28,11 @@ def authenticated(f):
@wraps(f) @wraps(f)
def decorated(*args, **kwargs): def decorated(*args, **kwargs):
cookie = request.cookies.get(auth.cookie_name) cookie = request.cookies.get(auth.cookie_name)
token = request.headers.get(auth.auth_header_name)
if cookie: if cookie:
user = auth.user_for_cookie(cookie) user = auth.user_for_cookie(cookie)
elif token:
user = auth.user_for_token(token)
else: else:
user = None user = None
if user: if user: