如何使用相关对象的过滤器获取不同项的SQL Alchemy对象

时间:2017-03-30 06:53:47

标签: python sqlalchemy distinct flask-sqlalchemy

我正在尝试在相关对象上过滤下面的不同项目的SQL Alchemy查询,相当于以下查询:

SELECT distinct items.item_id, items.item_name
FROM items
INNER JOIN categories as cat on items.category_id = cat.category_id
INNER JOIN stores on cat.store_id = stores.store_id
WHERE store.store_id = 123    

我创建了如下所示的模型,包含了外键,但是当我运行下面的查询时,它无法正确过滤。

items_query = (db.session.query(Store, Item)
               .filter(Store.store_id == 123)
               ).all()



#SQL Alchemy Models 
class Store(db.Model):
    __tablename__ = 'stores'
    store_id = db.Column(db.Integer, primary_key=True, autoincrement=True)
    store_name = db.Column(db.String(100), unique=True, nullable=False)

    def __repr__(self):
        return '<Store>'+str(self.store_name)

class Category(db.Model):
    __tablename__ = 'categories'
    category_id = db.Column(db.Integer, primary_key=True, autoincrement=True)
    category_name = db.Column(db.String(100), unique=True, nullable=False)
    store_id = db.Column(db.Integer, db.ForeignKey('stores.store_id'))
    store = db.relationship('Store', backref=db.backref('categories', lazy='dynamic'))

    def __repr__(self):
        return '<Category>'+str(self.category_name)

class Item(db.Model):
    __tablename__ = 'items'
    item_id = db.Column(db.Integer, primary_key=True, autoincrement=True)
    item_name = db.Column(db.String(150), unique=True, nullable=False)
    category_id = db.Column(db.Integer, db.ForeignKey('categories.category_id'))
    category = db.relationship('Category', backref=db.backref('items', lazy='dynamic'))

    def __repr__(self):
        return '<Item>'+str(self.item_name)

有人可以帮助我更好地形成查询吗?

1 个答案:

答案 0 :(得分:3)

使用

(db.session.query(Store, Item)
 .filter(Store.store_id == 123)
 ).all()

你会在StoreItem之间得到隐含的cross join,这显然不是你想要的。

首先使用关系显式构建所需的joins

query = db.session.query(Item.item_id, Item.item_name).\
    join(Item.category).\
    join(Category.store)

或简写形式:

query = db.session.query(Item.item_id, Item.item_name).\
    join("category", "store")

然后应用WHERE子句:

query = query.filter(Store.store_id == 123)

然后distinct()

query = query.distinct()

总结一下:

query = db.session.query(Item.item_id, Item.item_name).\
    join("category", "store").\
    filter(Store.store_id == 123).\
    distinct().\
    all()

此外,由于您对Item.item_name有唯一约束,并且由于一对多关系的方向,联接不应每Item生成多行,distinct()应该没必要。