SQLAlchemy查询MySQL分区表时如何指定需要查询的分区？

你可以通过以下两种常用方案实现指定分区查询：

方案1：使用select_from自定义表源（直观兼容原生SQL写法） #

适合固定分区的场景，逻辑和你写的原生SQL完全一致：

1. 先导入SQLAlchemy的text工具：

from sqlalchemy import text

2. 修改查询语句，通过select_from指定带分区声明的表：

# 注意原代码filter中的单等号是错误的，需要改为双等号做条件判断
res = People.query.select_from(text("people PARTITION (p202108, p202109)"))\
    .filter(People.city == 'new york').all()

注意：如果分区名是动态传入的，需要提前对分区名字段做合法性校验，避免SQL注入风险。

方案2：使用with_hint数据库提示（更安全，支持动态分区） #

这是SQLAlchemy官方推荐的方言专属提示写法，不需要手动拼接SQL字符串，安全性更高：

res = People.query.with_hint(People, "PARTITION (p202108, p202109)", dialect_name='mysql')\
    .filter(People.city == 'new york').all()

该方法会自动将分区声明拼到对应表的后面，动态传入分区名时也可以配合参数化处理，完全避免SQL注入问题。

两种方案生成的最终执行SQL都和你预期的原生分区查询完全一致，可以直接使用。

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