在SQLAlchemy中注释“存在”子查询

时间:2019-02-13 11:54:09

标签: python flask sqlalchemy flask-sqlalchemy marshmallow

我有一个简单的新闻应用程序,用户可以在其中查看最新帖子,他们可以喜欢其中一些帖子。

模型:

from sqlalchemy.orm import relationship, backref

from app import db


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

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

class Like(db.Model):
    id = db.Column(db.Integer, primary_key=True)
    user_id = db.Column(db.ForeignKey(User.id))
    post_id = db.Column(db.ForeignKey(Post.id))
    post = relationship(Post, backref=backref('likes'))

我要实现的目的是为每个帖子添加一个额外的字段,该字段指示登录用户是否喜欢该帖子,但以优化的方式而不对每个帖子进行查询。

我尝试使用marshmallow-sqlalchemy来实现架构级别:

from marshmallow import fields

from app import ma
from news.models import Post

class PostSchema(ma.ModelSchema):
    is_liked = fields.Method("get_is_liked")

    def get_is_liked(self, obj):
        user = self.context['user']
        return bool(Like.filter_by(user_id=user.id, post_id=obj.id).first())

    class Meta:
        model = Post

但这将导致每个帖子有N个查询。

我也尝试像Django ORM中的注释一样实现这一点:

class PostListView(Resource):

    # I have the user object here using a decorator
    def get(self, user):
        # I know that this is a bad syntax
        query = Post.query(Post.likes.filter_by(user_id=user.id).exists().label('is_liked')))

        schema = PostSchema(strict=True, many=True, context={"user": user})
        result = schema.dump(query.all())

        return result.data

我也看到了这个answer,它在数据库级别的功能与我相同,但是我想使用SQLAlchemy来实现。

谢谢。

1 个答案:

答案 0 :(得分:1)

这项工作吗?想法是加载帖子,以便menu.removeAllItems()仅包含当前用户是喜欢该帖子的用户的情况。 Post.likes应该将contains_eager('likes')数组加载到likes中。

Post

然后# user_id is the id of the current user q = session.query(Post) \ .outerjoin(Like, and_(Like.post_id == Post.id, Like.user_id == user_id)) \ .options(contains_eager('likes')) 可能类似于is_liked