26 lines
657 B
Python
26 lines
657 B
Python
from cryptography.fernet import Fernet
|
|
from simple_logger_handler import setup_logger
|
|
|
|
logger = setup_logger(__name__)
|
|
|
|
try:
|
|
with open("/etc/secrets/encryption_key", "rb") as file:
|
|
encryption_key = file.read()
|
|
except FileNotFoundError:
|
|
logger.fatal("[FATAL] Encryption key not found")
|
|
raise
|
|
except Exception as e:
|
|
logger.fatal(f"[FATAL] Failed to read encryption key: {e}")
|
|
raise
|
|
|
|
fernet = Fernet(encryption_key)
|
|
|
|
|
|
def encrypt_token(token: str) -> str:
|
|
"""Encrypt a token string."""
|
|
return fernet.encrypt(token.encode()).decode()
|
|
|
|
|
|
def decrypt_token(token: str) -> str:
|
|
"""Decrypt a token string."""
|
|
return fernet.decrypt(token.encode()).decode() |