SQL拼接易失控因字符串缺乏结构校验,错误仅运行时暴露;QueryBuilder通过累积条件对象、占位符参数、白名单校验字段表名、动态生成IN占位符等方式解决。

SQL拼接为什么容易失控?
因为字符串拼接天然缺乏结构校验,WHERE 条件漏空格、IN 子句少括号、参数类型错位(比如把 None 直接塞进字符串)——这些错误全在运行时才暴露。更麻烦的是,当一个查询要支持“按用户查+按时间范围过滤+可选统计分组”,硬拼 SQL 就会迅速膨胀成 200 行嵌套 if/else,改一处牵五处。
用 QueryBuilder 类封装 WHERE 和 ORDER BY
别从头写 ORM,先聚焦最痛的拼接部分。核心是把条件抽象成可累积的对象,而不是字符串累加:
class QueryBuilder:
def __init__(self):
self.where_clauses = []
self.params = []
<pre class="brush:php;toolbar:false;">def where(self, condition: str, *args):
self.where_clauses.append(condition)
self.params.extend(args)
return self
def order_by(self, field: str, direction="ASC"):
self._order_clause = f"ORDER BY {field} {direction}"
return self
def build(self) -> tuple[str, list]:
sql = "SELECT * FROM users"
if self.where_clauses:
sql += " WHERE " + " AND ".join(self.where_clauses)
if hasattr(self, "_order_clause"):
sql += " " + self._order_clause
return sql, self.params
这样调用就清晰了:qb = QueryBuilder().where("age > ?", 18).where("status = ?", "active").order_by("created_at", "DESC")。关键点:所有参数都走占位符(? 或 %s),避免 SQL 注入;build() 返回 (sql, params) 二元组,直接喂给 cursor.execute(sql, params)。
字段和表名必须白名单校验
用户输入的 order_by 字段、GROUP BY 列名,绝不能直接拼进 SQL。否则 order_by("id; DROP TABLE users--") 就完蛋。
- 维护一个允许排序的字段白名单:
ALLOWED_ORDER_FIELDS = {"id", "name", "created_at"} - 在
order_by方法里加校验:if field not in ALLOWED_ORDER_FIELDS: raise ValueError(f"Invalid order field: {field}") - 表名同理,用枚举或字典映射,比如
TABLE_MAP = {"user": "users", "order": "orders_v2"},外部只传逻辑名
IN 查询怎么安全处理变长参数?
这是最容易翻车的地方。直接 WHERE id IN ({','.join(map(str, ids))}) 不仅注入风险高,还可能超 SQL 长度限制或参数个数上限。
正确做法是动态生成占位符:
def where_in(self, field: str, values: list):
if not values:
self.where("1=0") # 空列表返回空结果
return self
placeholders = ",".join(["?"] * len(values))
self.where(f"{field} IN ({placeholders})", *values)
return self
注意两点:一是空列表要主动拦截,否则 IN () 语法错误;二是占位符数量必须严格匹配 values 长度,不能靠字符串格式化凑。
复杂查询真正难的不是语法,而是边界情况——比如 WHERE 条件全为空时要不要删掉 WHERE 关键字,IN 里混入 None 怎么处理,这些细节不提前想清楚,类一用就崩。
Python免费学习笔记(深入):立即使用
在学习笔记中,你将探索 Python 的核心概念和高级技巧!











