Files
SQLBot/backend/apps/datasource/utils/utils.py
2025-09-08 16:36:20 +08:00

20 lines
579 B
Python

from Crypto.Cipher import AES
from Crypto.Util.Padding import unpad, pad
import base64
key = b'SQLBot1234567890'
def aes_encrypt(data):
data = bytes(data,'utf-8')
cipher = AES.new(key, AES.MODE_ECB)
data = pad(data, AES.block_size)
encrypt = cipher.encrypt(data)
return base64.b64encode(encrypt)
def aes_decrypt(encrypted_data):
encrypted_data = base64.b64decode(encrypted_data)
cipher = AES.new(key, AES.MODE_ECB)
text = cipher.decrypt(encrypted_data)
decrypted_text = unpad(text, AES.block_size)
return decrypted_text.decode('utf-8')