在多对多关系中按计数排序SQLAlchemy

34 投票
1 回答
16992 浏览
提问于 2025-04-16 17:29

这是我当前模型的一个简化示例 (我正在使用 Flask SQLAlchemy 扩展):

like = db.Table(
    'like',
    db.Column('uid', db.Integer, db.ForeignKey('users.id')),
    db.Column('pid', db.Integer, db.ForeignKey('posts.id'))
)

class User(db.Model):
    __tablename__ = 'users'

    id = db.Column(db.Integer, primary_key = True)
    username = db.Column(db.String(20))

class Post(db.Model):
    __tablename__ = 'posts'

    id = db.Column(db.Integer, primary_key = True)
    title = db.Column(db.String(255))

    likes = db.relationship(
        'User',
        secondary = like,
        backref = db.backref('likes', lazy = 'dynamic'),
        lazy = 'dynamic'
    )

我想根据每个 Post点赞 数量来排序。

我基本上想要执行的查询是:

SELECT p.*, COUNT(l.`pid`) as `likes`
FROM `posts` as p
LEFT JOIN `like` as l
    ON p.`id` = l.`pid`
GROUP BY p.`id`
ORDER BY `likes` DESC

不过在 SQLAlchemy 这边我一直没能搞定。

感谢任何人提供的帮助。

1 个回答

57

我之前没怎么用过SQLAlchemy,所以想试试看。我没有用你们的模型,而是自己写了一些新的(虽然差不多):

likes = db.Table('likes',
    db.Column('user_id', db.Integer, db.ForeignKey('user.id')),
    db.Column('post_id', db.Integer, db.ForeignKey('post.id'))
)

class User(db.Model):
    id = db.Column(db.Integer, primary_key=True)
    username = db.Column(db.String(20))

    def __repr__(self):
        return "<User('%s')>" % self.username

class Post(db.Model):
    id = db.Column(db.Integer, primary_key=True)
    title = db.Column(db.String(255))

    likes = db.relationship('User', secondary = likes,
        backref = db.backref('posts', lazy='dynamic'))

    def __repr__(self):
        return "<Post('%s')>" % self.title

如果你想把likes表连接起来,可以用func.count来计算点赞的数量,然后用group_by来分组Post,最后用order_by来排序:

db.session.query(Post, func.count(likes.c.user_id).label('total'))\
 .join(likes).group_by(Post).order_by('total DESC')

我发现ORM教程和其他SQLAlchemy的文档非常有用。

撰写回答