SkyForces
21 lines
773 B
Python
21 lines
773 B
Python
from cryptography.fernet import Fernet, InvalidToken
|
|
|
|
|
|
class TokenCipher:
|
|
"""Шифрует OAuth-токены перед хранением в БД."""
|
|
|
|
def __init__(self, key: str) -> None:
|
|
try:
|
|
self._fernet = Fernet(key.encode("ascii"))
|
|
except (ValueError, UnicodeEncodeError) as error:
|
|
raise RuntimeError("TOKEN_ENCRYPTION_KEY is invalid") from error
|
|
|
|
def encrypt(self, value: str) -> bytes:
|
|
return self._fernet.encrypt(value.encode("utf-8"))
|
|
|
|
def decrypt(self, value: bytes) -> str:
|
|
try:
|
|
return self._fernet.decrypt(value).decode("utf-8")
|
|
except InvalidToken as error:
|
|
raise RuntimeError("Не удалось расшифровать OAuth-токен") from error
|