PyMongo实现MySQL风格多条件模糊查询及stock_id类型兼容问题求助

我刚好之前碰到过类似的类型匹配+多模糊查询的需求，给你整理一个可行的实现方案：

1. 核心问题：字符串匹配数值类型的stock_id #

因为你的stock_id是数值类型，但传入的查询值是字符串，直接匹配会因为类型不匹配返回空结果。我们可以通过尝试类型转换来解决：用try-except块把字符串转成整数，转换成功就用数值去匹配，失败则跳过这个条件（避免无效匹配）。

2. 还原MySQL中的模糊匹配逻辑 #

MongoDB用$regex实现类似SQL的like功能，还要注意转义正则特殊字符（比如.、*）避免意外行为：

• like '{}%'（前缀匹配）→ {"$regex": f"^{re.escape(value)}", "$options": "i"}（i表示忽略大小写，可选）
• like '%{}%'（包含匹配）→ {"$regex": re.escape(value), "$options": "i"}

3. 完整实现代码 #

import re
from pymongo import MongoClient

def stock_search(self, value):
    self.value = value
    # 替换成你实际获取code的方法
    code = self.get_target_code()  
    col = MongoClient()['Shop']['Stock']  # 确保客户端已正确初始化

    # 构建$or查询条件列表
    query_conditions = []

    # 处理stock_id的类型转换与匹配
    try:
        stock_id_num = int(value)
        query_conditions.append({"stock_id": stock_id_num})
    except ValueError:
        # 转整数失败，不添加该条件
        pass

    # 匹配barcode
    query_conditions.append({"barcode": code})

    # 处理main_type的两种模糊匹配
    escaped_value = re.escape(value)
    query_conditions.append({"main_type": {"$regex": f"^{escaped_value}", "$options": "i"}})
    query_conditions.append({"main_type": {"$regex": escaped_value, "$options": "i"}})

    # 处理sub_type的两种模糊匹配
    query_conditions.append({"sub_type": {"$regex": f"^{escaped_value}", "$options": "i"}})
    query_conditions.append({"sub_type": {"$regex": escaped_value, "$options": "i"}})

    # 执行查询并返回结果（可根据需求返回游标或列表）
    data = col.find({"$or": query_conditions})
    return list(data)

额外提示 #

• 若不需要忽略大小写，可去掉"$options": "i"参数
• 针对大量数据的模糊查询，建议给main_type和sub_type创建文本索引，用$text查询优化性能
• 确保MongoDB客户端已正确配置连接地址（远程连接需传入host和port）

内容的提问来源于stack exchange，提问作者ClownBunny