discours.io/core
5
0
Fork 0
Code Pull Requests Actions Projects Activity

token-type-tolerance

Browse Source
This commit is contained in:
Untone 2023-10-13 14:47:31 +03:00
parent e9f68c8fb1
commit 7cd5929df2
2 changed files with 2 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
auth/authenticate.py
View File

@ -29,8 +29,7 @@ class JWTAuthenticate(AuthenticationBackend):
user_id=None, username="" user_id=None, username=""
) )
if token.startswith("Bearer"): token = token.split(" ")[-1]
token = token[len("Bearer "):]
if len(token.split(".")) > 1: if len(token.split(".")) > 1:
payload = await SessionToken.verify(token) payload = await SessionToken.verify(token)

3
resolvers/auth.py
View File

@ -33,8 +33,7 @@ async def get_current_user(_, info):
auth: AuthCredentials = info.context["request"].auth auth: AuthCredentials = info.context["request"].auth
token = info.context["request"].headers.get(SESSION_TOKEN_HEADER) token = info.context["request"].headers.get(SESSION_TOKEN_HEADER)
if token.startswith("Bearer"): token = token.split(" ")[-1]
token = token[len("Bearer "):]
with local_session() as session: with local_session() as session:
user = session.query(User).where(User.id == auth.user_id).one() user = session.query(User).where(User.id == auth.user_id).one()