安全存储双十二活动

import sqlite3
from cryptography.fernet import Fernet

# 生成密钥并加密数据
key = Fernet.generate_key()
cipher_suite = Fernet(key)

def encrypt_data(data):
    return cipher_suite.encrypt(data.encode())

def decrypt_data(encrypted_data):
    return cipher_suite.decrypt(encrypted_data).decode()

# 连接到SQLite数据库
conn = sqlite3.connect('sales_data.db')
cursor = conn.cursor()

# 创建表
cursor.execute('''CREATE TABLE IF NOT EXISTS sales (
                    id INTEGER PRIMARY KEY AUTOINCREMENT,
                    product_name TEXT NOT NULL,
                    quantity INTEGER NOT NULL,
                    encrypted_data BLOB NOT NULL)''')

# 插入加密数据
product_name = "Example Product"
quantity = 100
encrypted_data = encrypt_data(f"{product_name},{quantity}")
cursor.execute("INSERT INTO sales (product_name, quantity, encrypted_data) VALUES (?, ?, ?)",
               (product_name, quantity, encrypted_data))

conn.commit()

# 查询并解密数据
cursor.execute("SELECT * FROM sales")
rows = cursor.fetchall()
for row in rows:
    decrypted_data = decrypt_data(row[3])
    print(f"Product: {decrypted_data.split(',')[0]}, Quantity: {decrypted_data.split(',')[1]}")

conn.close()