27 lines
669 B
Python
27 lines
669 B
Python
import logging
|
|
from os import getenv
|
|
|
|
from cryptography.fernet import Fernet
|
|
|
|
from credentials_loader import get_secret,MissingCredentialError
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
try:
|
|
encryption_key = get_secret(getenv("ENCRYPTION_KEY",""),"/etc/secrets/encryption_key")
|
|
logger.debug("Loaded encryption key")
|
|
except MissingCredentialError:
|
|
logger.fatal("Could not load encryption key")
|
|
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() |