在MongoDB中使用Python进行数据加密查询

!pip install pymongo

import pymongo
from pymongo import MongoClient

# 连接到MongoDB数据库
client = MongoClient('localhost', 27017)

# 创建一个数据库
db = client['testdb']

# 创建一个集合
collection = db['testcollection']

data = {
    'name': 'pidancode.com',
    'description': '皮蛋编程',
    'age': 25,
    'gender': 'male'
}

# 插入一条数据
collection.insert_one(data)

from Crypto.Cipher import AES
import base64

# 配置加密密钥和向量
key = '1234567890123456'
iv = '1234567890123456'

# 加密函数
def aes_encrypt(text):
    aes = AES.new(key.encode('utf8'), AES.MODE_CBC, iv.encode('utf8'))
    bs = AES.block_size
    # PKCS7的一个实现，填充16个字节的倍数
    pad = lambda s: s + (bs - len(s) % bs) * chr(bs - len(s) % bs)
    # 加密
    encrypt_aes = aes.encrypt(pad(text).encode('utf8'))
    # 使用base64编码
    encrypt_text = str(base64.encodebytes(encrypt_aes), encoding='utf8').replace('\n', '')
    return encrypt_text

encrypted_data = {
    'name': aes_encrypt('pidancode.com'),
    'description': aes_encrypt('皮蛋编程'),
    'age': aes_encrypt(25),
    'gender': aes_encrypt('male')
}

# 插入一条加密后的数据
collection.insert_one(encrypted_data)

# 定义解密函数
def aes_decrypt(text):
    aes = AES.new(key.encode('utf8'), AES.MODE_CBC, iv.encode('utf8'))
    # 先进行base64解码
    decrypt_aes = base64.decodebytes(text.encode('utf8'))
    # 解密
    decrypt_text = aes.decrypt(decrypt_aes).decode('utf8')
    # 删除填充的字符
    decrypt_text = decrypt_text[:-ord(decrypt_text[-1])]
    return decrypt_text

# 查询加密后的数据
query = {'name': aes_encrypt('pidancode.com')}
result = collection.find_one(query)
print(result)
print(aes_decrypt(result['name']))