烧瓶-sqlalchemy;通过外键列表过滤查询

时间:2020-06-12 10:24:09

标签: python flask sqlalchemy flask-sqlalchemy

我的postgres数据库具有类别和产品类。一个类别可以有许多产品。我正在尝试获取属于多个类别列表的产品(full_tree是类别ID的列表)。我尝试的查询似乎不起作用,因为它们总是返回数据库中的所有产品,即过滤部分似乎不起作用。

有人可以帮助我指出我做错了什么吗?还是显示进行此类查询的最佳做法?

我的课程:

class Parent_product(SearchableMixin, db.Model):
    __searchable__ = ['name']
    id = db.Column(db.Integer, primary_key = True)
    name = db.Column(db.String)

    manufacturer_id = db.Column(db.Integer, db.ForeignKey('manufacturer.id'))
    category_id = db.Column(db.Integer, db.ForeignKey('category.id'))

class Category(db.Model, BaseNestedSets):
    id = db.Column(db.Integer, primary_key=True)
    name = db.Column(db.String(400), index=True, unique=True)
    products = db.relationship("Parent_product", backref='category', lazy='dynamic')

我的查询:

def products(parent_category = None):
    # init query
    my_query = Parent_product.query

    if parent_category:

        # Find selected category from db
        category = Category.query.filter_by(name = parent_category).first()

        # Get list of all category id's to be used in query
        full_tree = []
        full_tree.append(category.id)
        for children in category.get_children():
            full_tree.append(children.id)

        # Add filter to query
        my_query.filter_by(category_id = Category.id.in_(full_tree))

    # Execute query, get list of products, paginate
    products = my_query.paginate('PAGE', 'PRODUCTS_PER_PAGE', False)

1 个答案:

答案 0 :(得分:1)

Query.filter_by()用于针对主要实体或作为Query.join()目标的最新实体进行简单的相等性检查。如果您需要更复杂的谓词,请使用Query.filter()。有鉴于此

my_query.filter_by(category_id = Category.id.in_(full_tree))

将产生类似

的SQL
FROM parent_product p, category c
WHERE p.category_id = (c.id IN (...))

是否产生错误取决于所使用的DBMS。 PostgreSQL会抱怨,但是例如SQLite会很乐意执行它。但是,对于您而言,这并不重要,因为您不会在任何地方分配新查询。 SQLAlchemy Query对象是可生成的,对它们的操作将返回新查询。

因此解决方法是将新查询分配给一个名称,并且在查询中不包括Category表:

my_query = my_query.filter(Parent_product.category_id.in_(full_tree))