Flask-SQLAlchemy过滤与父模型的多对多关系

时间:2016-04-28 13:24:22

标签: python sqlalchemy flask-sqlalchemy

我有一个父模型,不同类型的项目通过外键用作父项。我在父模型上也有很多关系。我试图基于查询多对多模型来获得子模型。

这是父模型

class MediaItem(db.Model):
    __tablename__ = "media_item"
    id = db.Column(db.Integer, primary_key=True)
    title = db.Column(db.String, unique=True)
    tags = db.relationship('Tags', secondary=tags_joiner, backref='media_items')
    videos = db.relationship('Video', backref='Parent', lazy='dynamic')
    audios = db.relationship('Audio', backref='Parent', lazy='dynamic')
    pictures = db.relationship('Picture', backref='Parent', lazy='dynamic')
    codes = db.relationship('Code', backref='Parent', lazy='dynamic')

和多对多的关系

class Tags(db.Model):
    __tablename__ = 'tags'
    id = db.Column(db.Integer, primary_key=True)
    tag = db.Column(db.String, unique=True, nullable=False)


tags_joiner = db.Table('tags_joiner',
                       db.Column('tag_id', db.Integer, db.ForeignKey('tags.id')),
                       db.Column('mediaitem_id', db.Integer, db.ForeignKey('media_item.id')),
                       db.PrimaryKeyConstraint('tag_id', 'mediaitem_id'))

最后有一个儿童模型的例子

class Video(db.Model):
    __tablename__ = 'video'
    id = db.Column(db.Integer, primary_key=True)
    parent_id = db.Column(db.Integer, db.ForeignKey('media_item.id'))
    file_name = db.Column(db.String, unique=True)

MediaItem模型中定义的关系证明了其他几种类型的子模型。

我希望通过标签对子模型进行过滤。也就是说,给定特定标记,返回与该标记关联的所有子模型。

Video.query.join(media_tags).filter_by(MediaItem.tags.any(Tags.tag.in_(tag)))

返回它不知道如何连接这三个表,(找不到要加入的FROM子句。尝试加入,但得到:找不到'media_item'和'tags'之间的任何外键关系。 )

我的方法是什么?

1 个答案:

答案 0 :(得分:7)

版本1:以下查询应返回所需的结果:

tag = 'my_filter_tag'
q = (
    db.session
    .query(Video)
    .filter(Video.Parent.has(MediaItem.tags.any(Tags.tag == tag)))
)

它可能不是最优的,因为它生成带有两个嵌套SQL子句的EXISTS,但它绝对是非常易读的sqlalchemy查询。以下是为sqlite生成的查询:

SELECT  video.id AS video_id, video.parent_id AS video_parent_id, video.file_name AS video_file_name
FROM    video
WHERE EXISTS (
    SELECT  1
    FROM    media_item
    WHERE   media_item.id = video.parent_id
        AND (
        EXISTS (
            SELECT 1
            FROM    tags_joiner, tags
            WHERE   media_item.id = tags_joiner.mediaitem_id
                AND tags.id = tags_joiner.tag_id
                AND tags.tag = :tag_1
            )
        )
    )

版本2:稍微优化的查询是加入media_item,但仍然会对标记执行exists

q = (
    db.session
    .query(Video)
    .join(MediaItem, Video.Parent)
    .filter(MediaItem.tags.any(Tags.tag == tag))
)

将生成SQL,如下所示:

SELECT  video.id AS video_id, video.parent_id AS video_parent_id, video.file_name AS video_file_name
FROM    video
JOIN    media_item
    ON  media_item.id = video.parent_id
WHERE   EXISTS (
    SELECT  1
    FROM    tags_joiner, tags
    WHERE   media_item.id = tags_joiner.mediaitem_id
        AND tags.id = tags_joiner.tag_id
        AND tags.tag = :tag_1
    )

您还可以在tags_joinertags上进一步加入并获得结果。但这会消除一些灵活性:如果您要执行并OR检查多个标记,结果可能会返回多个Video行,而保留查询EXISTS将会处理

请注意,您的代码有media_tags,但目前尚不清楚它是什么。