遍历SQLAlchemy关系-确定喜欢帖子的用户

时间:2019-07-06 04:03:15

标签: python flask sqlalchemy flask-sqlalchemy

我对flask和sql-alchemy比较陌生。我在处理不同模型之间的关系时遇到麻烦。具体来说,我正在尝试将喜欢特定帖子的用户的用户名打印到前端HTML。

class User(db.Model, UserMixin):
    id = db.Column(db.Integer, primary_key=True) 
    posts = db.relationship('Post', backref='author', lazy=True)
    posts = db.relationship('Post', backref='author', primaryjoin="User.id==Post.author_id")
    ...


    liked = db.relationship(
        'PostLike',
        foreign_keys='PostLike.user_id',
        backref='user', lazy='dynamic')


    def like_post(self, post):
        if not self.has_liked_post(post):
            like = PostLike(user_id=self.id, post_id=post.id)
            db.session.add(like)

    def unlike_post(self, post):
        if self.has_liked_post(post):
            PostLike.query.filter_by(
                user_id=self.id,
                post_id=post.id).delete()

    def has_liked_post(self, post):
        return PostLike.query.filter(
            PostLike.user_id == self.id,
            PostLike.post_id == post.id).count() > 0


class PostLike(db.Model):
    __tablename__ = 'post_like'
    id = db.Column(db.Integer, primary_key=True)
    user_id = db.Column(db.Integer, db.ForeignKey('user.id'))
    post_id = db.Column(db.Integer, db.ForeignKey('post.id'))


class Post(db.Model):
    id = db.Column(db.Integer, primary_key=True)
    author_id = db.Column(db.Integer, db.ForeignKey('user.id'), nullable=False)
    recipient_id = db.Column(db.Integer, db.ForeignKey('user.id'))
    likes = db.relationship('PostLike', backref='post', lazy='dynamic')
    ...

在我的flask应用程序中,当打印出帖子时,我希望能够显示帖子的喜欢者的用户名。我已经可以通过以下方式获得点赞功能,并显示点赞次数:

{{ post.likes.count() }}

通过HTML前端中的jinja2。

1 个答案:

答案 0 :(得分:0)

由于您的post_like表只是不同表中两行的简单配对,因此您可以使用辅助关系直接将其与用户对象联系起来。

class Post(db.Model):
    id = db.Column(db.Integer, primary_key=True)
    author_id = db.Column(db.Integer, db.ForeignKey('user.id'), nullable=False)
    recipient_id = db.Column(db.Integer, db.ForeignKey('user.id'))
    likes = db.relationship('User', secondary='post_like',
                            backref=db.backref('post', lazy='dynamic'))

如果您想向辅助表中添加更多列,则必须使用更为复杂的Association Object模式。

此外,您不应该有两个posts关系,它们需要合并。