sqlalchemy 使用新 API 加入子查询

Question

我无法将 SQL 语句转换为 sqlalchemy orm。 问题是我无法将生成的x子查询与FileTag表连接起来

我想使用新的 API，即不使用query

SQL 语句

    statement = f"""
    SELECT file_tags.id,file_tags.file_id,file_tags.tag,file_tags.category from
    (SELECT  id,file_id, count(DISTINCT tag)
    FROM file_tags
    WHERE {string_condition}
    GROUP BY file_id
    HAVING COUNT(DISTINCT tag) = {num_conditions})  as x
    JOIN file_tags on file_tags.file_id = x.file_id
    """

sqlalchemy ORM（仅最后一部分，加入 x/file_tag 不起作用）

    from sqlalchemy import func, subquery, join, alias

    # x is correct and working
    x = (
        select(FileTag)
        .where((FileTag.tag == "eso") & (FileTag.category == "top_category"))
        .group_by(FileTag.file_id)
        .having(func.count(FileTag.tag) == num_conditions)
        .subquery()
    )

    # I am not able to join x subquery. Throws several different errors
    # Here I tried a lot of different variations.
    query = join(x,FileTag.file_id==x.c.file_id)

    session.execute(query)

我认为主要问题是我不知道如何使用 sqlalchemy，但我尝试了所有可能的方法并且总是得到不同的错误。 有任何想法吗？

Answer 1

将x声明为.subquery() ，然后为您的join提供一些连接：

from sqlalchemy import Column, func, Integer, select, String
from sqlalchemy.orm import declarative_base

Base = declarative_base()


class FileTag(Base):
    __tablename__ = "file_tags"
    id = Column(Integer, primary_key=True)
    tag = Column(String)
    category = Column(String)
    file_id = Column(String)


num_conditions = 3
x = (
    select(FileTag)
    .where((FileTag.tag == "something") & (FileTag.category == "top_category"))
    .group_by(FileTag.file_id)
    .having(func.count(FileTag.tag) == num_conditions)
    .subquery()
)
query = select(FileTag).join(x, x.c.file_id == FileTag.file_id)
print(query)
"""
SELECT file_tags.id,
    file_tags.tag,
    file_tags.category,
    file_tags.file_id
FROM file_tags
JOIN (
    SELECT file_tags.id AS id,
        file_tags.tag AS tag,
        file_tags.category AS category,
        file_tags.file_id AS file_id
    FROM file_tags
    WHERE file_tags.tag = :tag_1
        AND file_tags.category = :category_1
    GROUP BY file_tags.file_id
    HAVING count(file_tags.tag) = :count_1
    ) AS anon_1
    ON anon_1.file_id = file_tags.file_id
"""