Image info, endpoint encryption
This commit is contained in:
25
bot/db/encryption.py
Normal file
25
bot/db/encryption.py
Normal file
@@ -0,0 +1,25 @@
|
||||
from cryptography.fernet import Fernet
|
||||
from cryptography.hazmat.primitives import hashes
|
||||
from cryptography.hazmat.primitives.kdf.pbkdf2 import PBKDF2HMAC
|
||||
import base64
|
||||
from bot.config import ENCRYPTION_KEY, BARS_APP_ID
|
||||
|
||||
|
||||
fernet = Fernet(
|
||||
base64.urlsafe_b64encode(
|
||||
PBKDF2HMAC(
|
||||
algorithm=hashes.SHA256(),
|
||||
length=32,
|
||||
iterations=390000,
|
||||
salt=BARS_APP_ID.encode()
|
||||
).derive(ENCRYPTION_KEY)
|
||||
)
|
||||
)
|
||||
|
||||
|
||||
def encrypt(s: str) -> bytes:
|
||||
return fernet.encrypt(s.encode())
|
||||
|
||||
|
||||
def decrypt(s: bytes) -> str:
|
||||
return fernet.decrypt(s).decode()
|
||||
Reference in New Issue
Block a user