SQLAlchemy使用WHERE IN子查询触发SAWarning警告如何解决

警告触发原因 #

你碰到的警告是SQLAlchemy 1.4+版本的类型校验导致的：in_()方法要求传入的子查询必须是显式的select()构造，你原有代码中调用base_query.subquery()得到的是Subquery类型对象，SQLAlchemy需要对其做强制类型转换才能适配IN子句的参数要求，因此抛出了该提示警告。

修复写法 #

方案1（最简修改） #

直接去掉.subquery()调用即可，base_query本身就是查询用户ID的Select对象，可以直接传入in_()方法使用，同时可以把冗余的select(User).with_only_columns(User.id)简化为select(User.id)，逻辑完全一致：

base_query = select(User.id).join(UserPermission).where(
    UserPermission.permission.in_(["can_access_first_thing", "can_access_second_thing"])
)
results = db.execute(select(User).where(User.id.in_(base_query))).scalars().all()

方案2（显式子查询写法） #

如果你的场景需要显式生成子查询（比如需要对中间子查询做额外处理），可以显式从子查询中选中ID列再传入IN子句：

base_query = select(User.id).join(UserPermission).where(
    UserPermission.permission.in_(["can_access_first_thing", "can_access_second_thing"])
).subquery()
results = db.execute(select(User).where(User.id.in_(select(base_query.c.id)))).scalars().all()

额外优化建议 #

你也可以改用EXISTS半连接实现相同逻辑，不需要额外处理去重，性能通常优于IN子查询，也不会触发任何警告：

from sqlalchemy import exists

results = db.execute(
    select(User).where(
        exists(
            select(1).where(
                UserPermission.user_id == User.id,
                UserPermission.permission.in_(["can_access_first_thing", "can_access_second_thing"])
            )
        )
    )
).scalars().all()

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