From 69d77ba09ddad6e2a76d4b10f473b997f2b501fb Mon Sep 17 00:00:00 2001
From: inter <inter@qq.com>
Date: Mon, 8 Sep 2025 16:36:20 +0800
Subject: [PATCH] Add File

---
 backend/apps/datasource/utils/utils.py | 19 +++++++++++++++++++
 1 file changed, 19 insertions(+)
 create mode 100644 backend/apps/datasource/utils/utils.py

diff --git a/backend/apps/datasource/utils/utils.py b/backend/apps/datasource/utils/utils.py
new file mode 100644
index 0000000..993e6dc
--- /dev/null
+++ b/backend/apps/datasource/utils/utils.py
@@ -0,0 +1,19 @@
+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')