mysql中字段加密

import mysql.connector
from Crypto.Cipher import AES
from Crypto.Util.Padding import pad, unpad
from Crypto.Random import get_random_bytes

# 连接数据库
db = mysql.connector.connect(
    host="localhost",
    user="youruser",
    password="yourpassword",
    database="yourdatabase"
)
cursor = db.cursor()

# 加密函数
def encrypt_data(data, key):
    iv = get_random_bytes(16)
    cipher = AES.new(key, AES.MODE_CBC, iv)
    encrypted_data = cipher.encrypt(pad(data.encode(), AES.block_size))
    return iv + encrypted_data

# 解密函数
def decrypt_data(encrypted_data, key):
    iv = encrypted_data[:16]
    cipher = AES.new(key, AES.MODE_CBC, iv)
    decrypted_data = unpad(cipher.decrypt(encrypted_data[16:]), AES.block_size)
    return decrypted_data.decode()

# 示例：加密并存储数据
key = get_random_bytes(16)  # 生成一个16字节的密钥
data_to_encrypt = "sensitive data"
encrypted_data = encrypt_data(data_to_encrypt, key)

# 插入加密数据到数据库
cursor.execute("INSERT INTO yourtable (encrypted_data) VALUES (%s)", (encrypted_data,))
db.commit()

# 示例：从数据库读取并解密数据
cursor.execute("SELECT encrypted_data FROM yourtable WHERE id = %s", (1,))
encrypted_data = cursor.fetchone()[0]
decrypted_data = decrypt_data(encrypted_data, key)
print(decrypted_data)

# 关闭数据库连接
cursor.close()
db.close()