Python的sqlalchemy框架中多表联合查询问题?

Question

在使用 Python的 SqlAlchemy数据库框架时，碰到了如下问题

现有两张表:

A表：

TID  主键
PID  PID
Title  标题

B表:

GID  主键
PID  PID
Name  名称

两张表中PID相关联，A表中的一条数据，对应着B表中的多条数据，
现在我想查出B表中所有和A表中数据关联的数据，并用以下形式展现:

[{
        "TID": "01",
        "PID": "1",
        "Title": "标题1",
        "Name": ["名称1", "名称2", "名称3"...]
    },
    {
        "TID": "02",
        "PID": "2",
        "Title": "标题2",
        "Name": ["名称1", "名称2", "名称3"...]
    }
        ......
]

请问该如何实现？

Answer 1

用 SQLAlchemy 的 ORM 进行查询。那你就先要建立两张表的 ORM 模型，然后用 SQLAlchemy 的 join 功能进行连接，最后可以把结果转换为列表，再按所需进行组装。
代码如下：

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

Base = declarative_base()

class ATable(Base):
    __tablename__ = 'A'
    TID = Column(Integer, primary_key=True)
    PID = Column(Integer)
    Title = Column(String)

class BTable(Base):
    __tablename__ = 'B'
    GID = Column(Integer, primary_key=True)
    PID = Column(Integer)
    Name = Column(String)

engine = create_engine('sqlite:///:memory:')
Session = sessionmaker(bind=engine)
session = Session()

result = session.query(ATable, BTable).join(BTable, ATable.PID == BTable.PID).all()

data = []
for a, b in result:
    if len(data) == 0 or data[-1]['TID'] != a.TID:
        data.append({'TID': a.TID, 'PID': a.PID, 'Title': a.Title, 'Name': [b.Name]})
    else:
        data[-1]['Name'].append(b.Name)

print(data)

上面的代码使用了 SQLite 数据库，如果要使用其他数据库，要修改 create_engine 中的数据库链接字符串。