python - 将查询输出作为输入传递给 SQLAlchemy 中另一个查询的 in 子句

Question

我有 2 个表(比如 Student 和 College)和第三个表(StudentCollege)，其中有 student_id 和 college_id 外键。

我想给出以下查询的输出:

list = (
    db.session.query(StudentCollegeModel.College_id)
    .filter(StudentCollegeModel.student_id== student_id)
    .all()
)

下面的查询:

(
    db.session.query(CollegeModel)
    .filter(CollegeModel.College_id.in_(list))
    .all()
)

但是它给出了编程错误。

Answer 1

您无需执行第一个查询即可在第二个查询中将其用作子查询。这使您不必在进行 in_() 查询之前构建所有 College_id 的内存列表，并且意味着您只需往返一次数据库。

subquery = (
    db.session.query(StudentCollegeModel.College_id)
    .filter(StudentCollegeModel.student_id== student_id)
)

result = (
    db.session.query(CollegeModel)
    .filter(CollegeModel.College_id.in_(subquery))
    .all()
)