refactor: Refactor config

This commit is contained in:
Naruse
2024-11-08 11:11:18 +08:00
parent bfbf1bb2ab
commit 1928801460
6 changed files with 88 additions and 112 deletions

16
utils/aes.py Normal file
View File

@@ -0,0 +1,16 @@
import base64
from Crypto.Cipher import AES
from Crypto.Util.Padding import pad, unpad
def encrypt_ecb(key: str, data: str):
cipher = AES.new(bytes.fromhex(key.replace(" ", "")), AES.MODE_ECB)
encrypted = cipher.encrypt(pad(data.encode(), AES.block_size))
return base64.b64encode(encrypted).decode()
def decrypt_ecb(key: str, data: str):
data = base64.b64decode(data)
cipher = AES.new(bytes.fromhex(key.replace(" ", "")), AES.MODE_ECB)
decrypted = cipher.decrypt(data)
return unpad(decrypted, AES.block_size).decode()