Merge branch 'main' of staging.discours.io:discoursio-api into cudl

This commit is contained in:
tonyrewin 2022-11-15 12:28:43 +03:00
commit c4a1cc1b3f
2 changed files with 5 additions and 5 deletions

View File

@ -41,11 +41,11 @@ class JWTCodec:
r = TokenPayload(**payload)
print('[jwtcodec] debug payload %r' % r)
return r
except jwt.InvalidIssuedAtError:
raise ExpiredToken('check token issued time')
except jwt.ExpiredSignatureError:
raise ExpiredToken('check token lifetime')
except jwt.InvalidTokenError:
raise InvalidToken('token is not valid')
except jwt.InvalidSignatureError:
raise InvalidToken('token is not valid')
except jwt.InvalidIssuedAtError:
raise ExpiredToken('check token issued time')

View File

@ -1,4 +1,4 @@
from datetime import datetime, timedelta
from datetime import datetime, timedelta, timezone
from auth.jwtcodec import JWTCodec
from validations.auth import AuthInput
@ -21,7 +21,7 @@ class TokenStorage:
@staticmethod
async def create_onetime(user: AuthInput) -> str:
life_span = ONETIME_TOKEN_LIFE_SPAN
exp = datetime.utcnow() + timedelta(seconds=life_span)
exp = datetime.now(tz=timezone.utc) + timedelta(seconds=life_span)
one_time_token = JWTCodec.encode(user, exp)
await save(f"{user.id}-{one_time_token}", life_span)
return one_time_token
@ -29,7 +29,7 @@ class TokenStorage:
@staticmethod
async def create_session(user: AuthInput) -> str:
life_span = SESSION_TOKEN_LIFE_SPAN
exp = datetime.utcnow() + timedelta(seconds=life_span)
exp = datetime.now(tz=timezone.utc) + timedelta(seconds=life_span)
session_token = JWTCodec.encode(user, exp)
await save(f"{user.id}-{session_token}", life_span)
return session_token