Upload encrypt_it.py

encrypt_it.py

@@ -0,0 +1,46 @@
+import hashlib
+from cryptography.fernet import Fernet
+import base64
+
+# 生成哈希密钥
+def generate_key(username, password):
+    key = username.encode() + password.encode()
+    hash_key = hashlib.sha256(key).hexdigest().encode()[:32]
+    base64_key = base64.urlsafe_b64encode(hash_key)
+    return base64_key
+
+# 加密函数，将明文加密成密文
+def encrypt(message, key):
+    cipher = Fernet(key)
+    ciphertext = cipher.encrypt(message)
+    with open('encrypted.bin', 'wb') as f:
+        f.write(ciphertext)
+    return ciphertext
+
+# 解密函数，将密文解密成明文
+def decrypt(ciphertext, key):
+    cipher = Fernet(key)
+    plaintext = cipher.decrypt(ciphertext)
+    return plaintext
+
+
+if __name__ == "__main__":
+    # 测试加解密函数
+    username = input('请输入用户名：')
+    password = input('请输入密码：')
+
+    # 生成哈希密钥
+    key = generate_key(username, password)
+
+    # # 加密数据并保存到文件中
+    # message = b'Hello, World!'
+    # ciphertext = encrypt(message, key)
+    # with open('encrypted.bin', 'wb') as f:
+    #     f.write(ciphertext)
+
+    # 解密文件并输出明文数据
+    with open('encrypted.bin', 'rb') as f:
+        ciphertext = f.read()
+    key = generate_key(username, password)
+    plaintext = decrypt(ciphertext, key)
+    print(f'{username}的私密信息：{plaintext.decode()}')