SQLAlchemy的和關係

Question

從下面的代碼過濾器，我定義的關係是這樣的：

UnitTypeRowModel (self-referencing but only one in the doc) 
    -- PlanPmDefinition (definition in the json doc) 
    -- PlanPmValues (value in the json doc) 

目前JSON作爲結果，如果我只有在數據庫表中的一個plan_ou：

[{ 
    "children": [], 
    "parent_id": 1, 
    "id": 4 
}, { 
    "children": [], 
    "parent_id": 2, 
    "plan_pm_id": 2, 
    "id": 5 
    "definition": { 
     "id": 2, 
     "tag": 0, 
     "value": { 
      "plan_pm_id": 2, 
      "tag": 0, 
      "plan_ou": 1 
     } 
    } 
}] 

代碼：

class PlanPmValues(Base): 
    __tablename__ = 'plan_pm_municipal_values' 

    plan_pm_id = Column(Integer, primary_key=True) 
    tag = Column(Integer, primary_key=True) 
    plan_ou = Column(Integer) # Filter on this 
    ... 


class PlanPmDefinition(Base): 
    __tablename__ = 'plan_pm_municipal_definition' 

    id = Column(Integer, primary_key=True) 
    tag = Column(Integer, primary_key=True) -- always 0 

    value = relationship(PlanPmValues, 
       primaryjoin='and_(PlanPmDefinition.id==PlanPmValues.plan_pm_id, ' + 
       'PlanPmDefinition.tag==PlanPmValues.tag)', 
       foreign_keys='[PlanPmValues.plan_pm_id, PlanPmValues.tag]', 
       lazy='joined', uselist=False) 


class UnitTypeRowModel(Base): 
    __tablename__ = 'unit_type_row_model' 
    __table_args__ = {'schema': base_schema} 

    id = Column(Integer, primary_key=True) 
    client_id = Column(Integer, ForeignKey(base_schema + '.client.id')) 
    parent_id = Column(Integer, ForeignKey(base_schema + '.unit_type_row_model.id'), nullable=True) 
    plan_pm_id = Column(Integer, nullable=True) 

    children = relationship(
      'UnitTypeRowModel', 
      lazy='joined', 
      join_depth=2, 
      order_by="UnitTypeRowModel.sort_order") 

    definition = relationship(
      'PlanPmDefinition', 
      primaryjoin='and_(PlanPmDefinition.id==UnitTypeRowModel.plan_pm_id, ' + 
      'PlanPmDefinition.tag==0)', 
      foreign_keys='[UnitTypeRowModel.plan_pm_id]', 
      lazy='joined', 
      uselist=False) 

    @staticmethod 
    def get_for_unit(client_id, unit_id): 
     db_session = DatabaseEngine.get_session() 
     row_models = db_session.query(UnitTypeRowModel).\ 
      filter(UnitTypeRowModel.client_id == client_id).\ 
      order_by(UnitTypeRowModel.sort_order) 
     json = util.Serialize.serialize_to_json(row_models) 
     db_session.close() 
     return json 

如何在方法UnitTypeRowModel.get_for_unit中從PlanPmValues中的plan_ou過濾？

Answer 1

您應該爲此使用自定義ColumnPropery。

from sqlalchemy import select, and_ 
from sqlalchemy.orm import column_property 

class UnitTypeRowModel(Base): 

    ... 

    _ou_join = and_(plan_pm_id==PlanPmDefinition.id, 
        PlanPmDefinition.tag==PlanPmValues.tag, 
        plan_pm_id==PlanPmValues.plan_pm_id) 

    plan_ou = column_property(select([PlanPmValues.plan_ou], 
            whereclause=_ou_join).as_scalar()) 

我打破_ou_join作爲一個單獨的屬性只爲了可讀性。

您在SQLAlchemy代碼中定義了這種方式，您在UnitTypeRowModel和PlanPmDefinition之間有多對一的關係;並且您在PlanPmDefinition和PlanPmValues之間有多對一的關係。我從uselist=False參數推斷出這兩個relationship()調用，並且如果您沒有通過使用唯一鍵在數據庫中強制執行這些錯誤，則可能會遇到其他一些錯誤。這意味着結果值只能返回一行，我已使用as_scalar()執行該行。

通常，您使用association_proxy()來處理相關對象的查詢。這在這種情況下並不理想，因爲您已經擁有了UnitTypeRowModel和PlanPmValues之間的關聯對象（PlanPmDefinition）這意味着association_proxy將允許您直接從UnitTypeRowModel類而不是plan_ou值處理PlanPmValue對象。

最後，你應該始終確保你需要做到這一點作爲一個SQLAlchemy的Query.filter()凹口 - 是，SQL服務器上執行的查詢和過濾的 - 而不是通過filter()在Python這樣做。前者需要更多時間進行編碼和測試，但幾乎總是會更快，因爲SQL服務器已針對其進行了優化;後者需要更少的時間進行編碼，並且在實踐中只會在執行大量返回大量結果的查詢時導致性能問題。我個人用Python寫東西，然後通過SQL進行慢速查詢。