From 20adb9a5ea190b2e963a7e484302cdc1b183acfc Mon Sep 17 00:00:00 2001
From: pyj256ost <2940908360@qq.com>
Date: Wed, 13 Dec 2023 19:56:06 +0800
Subject: [PATCH] ADD file via upload

---
 encrypt_message.py | 22 ++++++++++++++++++++++
 1 file changed, 22 insertions(+)
 create mode 100644 encrypt_message.py

diff --git a/encrypt_message.py b/encrypt_message.py
new file mode 100644
index 0000000..2521b56
--- /dev/null
+++ b/encrypt_message.py
@@ -0,0 +1,22 @@
+import os
+from cryptography.fernet import Fernet
+
+def generate_and_save_key(key_file_path):
+    # 生成对称密钥
+    key = Fernet.generate_key()
+    
+    # 将密钥保存到文件中
+    with open(key_file_path, 'wb') as key_file:
+        key_file.write(key)
+
+    return key
+
+def encrypt_file(file_path, key):
+    with open(file_path, 'rb') as file:
+        plaintext = file.read()
+
+    cipher_suite = Fernet(key)
+    cipher_text = cipher_suite.encrypt(plaintext)
+
+    with open(file_path + '.encrypted', 'wb') as encrypted_file:
+        encrypted_file.write(cipher_text)