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)
def decorated(*args, **kwargs):
cookie = request.cookies.get(auth.cookie_name)
token = request.headers.get(auth.auth_header_name)
if cookie:
user = auth.user_for_cookie(cookie)
elif token:
user = auth.user_for_token(token)
else:
user = None
if user: