本文介紹了在使用 SQLAlchemy 進(jìn)行多表聯(lián)合查詢時(shí)，如何保持查詢結(jié)果中每個(gè)對象的類型定義，避免類型推斷為 Any。通過使用 .tuples() 方法，可以將查詢結(jié)果轉(zhuǎn)換為元組序列，從而方便地解包并直接使用對象，無需額外定義變量類型。

在使用 SQLAlchemy 進(jìn)行數(shù)據(jù)庫查詢時(shí)，經(jīng)常會(huì)遇到需要聯(lián)合多個(gè)表進(jìn)行查詢的情況。 SQLAlchemy 提供了強(qiáng)大的 select 語句來實(shí)現(xiàn)這一功能。然而，當(dāng)查詢結(jié)果包含多個(gè)對象時(shí)，如何保持每個(gè)對象的類型定義，避免類型推斷為 Any，成為了一個(gè)需要解決的問題。

問題背景

假設(shè)我們有兩個(gè)模型 Item 和 Package，它們之間存在關(guān)聯(lián)關(guān)系。我們想要查詢同時(shí)包含 Item 和 Package 信息的記錄。常見的做法是使用 select 語句進(jìn)行聯(lián)合查詢：

from sqlalchemy import create_engine, Column, Integer, String, ForeignKey
from sqlalchemy.orm import sessionmaker, relationship
from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy import select

# 定義數(shù)據(jù)庫連接
engine = create_engine('sqlite:///:memory:')  # 使用內(nèi)存數(shù)據(jù)庫
Base = declarative_base()

# 定義模型
class Package(Base):
    __tablename__ = 'package'
    Package_id = Column(Integer, primary_key=True)
    name = Column(String)
    items = relationship("Item", back_populates="package")

class Item(Base):
    __tablename__ = 'item'
    Item_id = Column(Integer, primary_key=True)
    Package_id1 = Column(Integer, ForeignKey('package.Package_id'))
    description = Column(String)
    package = relationship("Package", back_populates="items")

Base.metadata.create_all(engine)

# 創(chuàng)建 Session
Session = sessionmaker(bind=engine)
session = Session()

# 插入一些數(shù)據(jù)
package1 = Package(name='Package 1')
package2 = Package(name='Package 2')
item1 = Item(description='Item 1', package=package1)
item2 = Item(description='Item 2', package=package1)
item3 = Item(description='Item 3', package=package2)

session.add_all([package1, package2, item1, item2, item3])
session.commit()

# 查詢數(shù)據(jù)
# DB = DatabaseModel() # 假設(shè) DatabaseModel 已經(jīng)定義好 session
# stmt = select(Item, Package).join(Package, Item.Package_id1 == Package.Package_id)
# exec = DB.session.execute(stmt).all() # Sequence[Row[Tuple[Item, Package]]]

# for row in exec:
#     row #Row[Tuple[Item, Package]]

#     Item_object : Item = row[0]
#     Package_object : Package = row[1]

上述代碼中，DB.session.execute(stmt).all() 返回的結(jié)果類型是 Sequence[Row[Tuple[Item, Package]]]。這意味著我們需要通過索引來訪問 Item 和 Package 對象，并且需要手動(dòng)指定類型，例如 Item_object : Item = row[0]。 這種方式雖然可行，但不夠優(yōu)雅，并且增加了代碼的冗余度。

解決方案：使用 .tuples() 方法

更簡潔、更優(yōu)雅的解決方案是使用 SQLAlchemy 的 .tuples() 方法。該方法可以將查詢結(jié)果轉(zhuǎn)換為元組序列，從而方便我們直接解包并使用對象。

stmt = select(Item, Package).join(Package, Item.Package_id1 == Package.Package_id)
exec = session.execute(stmt).tuples().all() # Sequence[Tuple[Item, Package]]]

for _item, _package in exec:
    print(f"Item Description: {_item.description}, Package Name: {_package.name}")
    # 在這里可以直接使用 _item 和 _package 對象

通過在 execute 語句后添加 .tuples()，返回的結(jié)果類型變?yōu)?Sequence[Tuple[Item, Package]]]。在 for 循環(huán)中，我們可以直接使用 _item 和 _package 來訪問 Item 和 Package 對象，而無需手動(dòng)指定類型。

示例代碼

下面是一個(gè)完整的示例代碼，演示了如何使用 .tuples() 方法進(jìn)行多表聯(lián)合查詢：

from sqlalchemy import create_engine, Column, Integer, String, ForeignKey
from sqlalchemy.orm import sessionmaker, relationship
from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy import select

# 定義數(shù)據(jù)庫連接
engine = create_engine('sqlite:///:memory:')  # 使用內(nèi)存數(shù)據(jù)庫
Base = declarative_base()

# 定義模型
class Package(Base):
    __tablename__ = 'package'
    Package_id = Column(Integer, primary_key=True)
    name = Column(String)
    items = relationship("Item", back_populates="package")

class Item(Base):
    __tablename__ = 'item'
    Item_id = Column(Integer, primary_key=True)
    Package_id1 = Column(Integer, ForeignKey('package.Package_id'))
    description = Column(String)
    package = relationship("Package", back_populates="items")

Base.metadata.create_all(engine)

# 創(chuàng)建 Session
Session = sessionmaker(bind=engine)
session = Session()

# 插入一些數(shù)據(jù)
package1 = Package(name='Package 1')
package2 = Package(name='Package 2')
item1 = Item(description='Item 1', package=package1)
item2 = Item(description='Item 2', package=package1)
item3 = Item(description='Item 3', package=package2)

session.add_all([package1, package2, item1, item2, item3])
session.commit()

# 查詢數(shù)據(jù)
stmt = select(Item, Package).join(Package, Item.Package_id1 == Package.Package_id)
exec = session.execute(stmt).tuples().all()

for _item, _package in exec:
    print(f"Item Description: {_item.description}, Package Name: {_package.name}")

注意事項(xiàng)

• .tuples() 方法只能用于返回多個(gè)對象的查詢。如果查詢只返回一個(gè)對象，則不需要使用該方法。
• 使用 .tuples() 方法后，查詢結(jié)果將不再是 Row 對象，而是元組。因此，不能再使用索引來訪問對象。

總結(jié)

通過使用 SQLAlchemy 的 .tuples() 方法，我們可以更方便地進(jìn)行多表聯(lián)合查詢，并保持查詢結(jié)果中每個(gè)對象的類型定義。這種方法不僅簡化了代碼，還提高了代碼的可讀性。在實(shí)際開發(fā)中，可以根據(jù)具體情況選擇合適的查詢方式，以達(dá)到最佳的效果。

以上が複數(shù)列のクエリ結(jié)果に対する SQLAlchemy オブジェクト定義の永続性の詳細(xì)內(nèi)容です。詳細(xì)については、PHP 中國語 Web サイトの他の関連記事を參照してください。