SelfPrivacy
/
selfprivacy-rest-api
7
2
Fork 1
Code Issues 15 Pull Requests 3 Projects Releases 1 Wiki Activity

fix(tokens): missing timezone import
continuous-integration/drone/push Build is failing Details

pull/68/head
Houkime 2023-10-25 14:53:11 +00:00
parent 829aca14be
commit b7cd703eaa
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
selfprivacy_api/repositories/tokens/redis_tokens_repository.py
View File

@ -4,6 +4,7 @@ Token repository using Redis as backend.
from typing import Any, Optional from typing import Any, Optional
from datetime import datetime from datetime import datetime
from hashlib import md5 from hashlib import md5
from datetime import timezone
from selfprivacy_api.repositories.tokens.abstract_tokens_repository import ( from selfprivacy_api.repositories.tokens.abstract_tokens_repository import (
AbstractTokensRepository, AbstractTokensRepository,
@ -53,6 +54,7 @@ class RedisTokensRepository(AbstractTokensRepository):
token = self._token_from_hash(key) token = self._token_from_hash(key)
if token == input_token: if token == input_token:
return key return key
return None
def delete_token(self, input_token: Token) -> None: def delete_token(self, input_token: Token) -> None:
"""Delete the token""" """Delete the token"""
@ -148,6 +150,7 @@ class RedisTokensRepository(AbstractTokensRepository):
if token is not None: if token is not None:
token.created_at = token.created_at.replace(tzinfo=None) token.created_at = token.created_at.replace(tzinfo=None)
return token return token
return None
def _recovery_key_from_hash(self, redis_key: str) -> Optional[RecoveryKey]: def _recovery_key_from_hash(self, redis_key: str) -> Optional[RecoveryKey]:
return self._hash_as_model(redis_key, RecoveryKey) return self._hash_as_model(redis_key, RecoveryKey)