Question

from sqlalchemy.orm import subqueryload, joinedload, eagerload
from sqlalchemy import Column, DateTime, String, Integer, ForeignKey, func,Float, sql
from sqlalchemy.orm import relation
from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy.orm import sessionmaker
from sqlalchemy import create_engine

engine = create_engine('sqlite:///testdb.sqlite')
engine.echo = True
Base = declarative_base()
session = sessionmaker()
session.configure(bind=engine)
Base.metadata.create_all(engine)
s= session()

class Stock(Base):
__tablename__ = 'stock'
stock_id = Column(Integer, primary_key=True)
name = Column(String)
prices = relation("StockPrice")

class StockPrice(Base):
__tablename__ = 'stock_price'
stock_id = Column(Integer, ForeignKey('stock.stock_id'), primary_key=True)
date = Column(String, primary_key=True)
price = Column(Float)
source = Column(String, primary_key=True)
user = Column(String)

Base.metadata.create_all(engine)

stockprice1 = StockPrice(stock_id = 1, date="2014-10-29", price="170.0", source="X Firm", user="U1")
stockprice2 = StockPrice(stock_id = 1, date="2014-10-30", price="175.0", source="X Firm", user="U2")
stock1 = Stock(stock_id = 1, name = "GOOGLE", prices=[stockprice1, stockprice2])

stockprice1 = StockPrice(stock_id = 2, date="2014-10-29", price="150.0", source="X Firm", user="U1")
stockprice2 = StockPrice(stock_id = 2, date="2014-10-30", price="155.0", source="X Firm", user="U2")
stock2 = Stock(stock_id = 2, name = "YAHOO", prices=[stockprice1, stockprice2])

s.add_all([stock1, stock2])
s.commit()

急切加载股票的价格：

stock = s.query(Stock).options(joinedload(Stock.prices)).filter(Stock.stock_id == 1).one()

在给定日期急切加载股票价格的一种方法：

stock = s.query(Stock).options(joinedload(Stock.prices)).filter(Stock.stock_id == 1).filter(StockPrice.date == "2014-10-30").one()

但是这个方法的问题在于你是否有像StockPrice这样的表与Stock有关，如果你想加载给定日期的所有关系，那么在加入所有关系后结果集会变得非常庞大。过滤器在WHERE子句中添加条件，而我需要一种方法来指定连接上的条件以急切加载。

stock = s.query(Stock).options(joinedload(Stock.prices, #condition does not work here)).filter(Stock.stock_id == 1).one()

Answer 1

而不是joinedload(Stock.prices)执行以下操作：

stock = (s.query(Stock)

         # @note: this replaces `joinedload(Stock.prices)`
         .join(StockPrice,
               and_(StockPrice.stock_id == Stock.stock_id,
                    StockPrice.date == "2014-10-30")
               )

         # effectively *trick* SQLAlchemy into thinking that above we loaded all
         # items for the relationship *Stock.prices*
         .options(contains_eager(Stock.prices))

         ).get(1) # will retrieve the instance for Stock.stock_id = 1

加入一个条件，急切地加载sqlalchemy orm

1 个答案: