SQLAlchemy - 查询多个表并返回嵌套对象

Question

假设我们在 Company 和 Employee 之间有一个简单的一对多关系，有没有办法查询所有公司并在每个公司的属性中都有一个员工列表？

class Company(Base):
    __tablename__ = 'company'

    id = Column(Integer, primary_key=True)
    name = Column(String)


class Employee(Base):
    __tablename__ = 'employee'

    id = Column(Integer, primary_key=True)
    first_name = Column(String)
    last_name = Column(String)
    company_id = Column(Integer, ForeignKey(Company.id))

我正在寻找这样的东西：

>>> result = db.session.query(Company).join(Employee).all()
>>> result[0].Employee
[<Employee object at 0x...>, <Employee object at 0x...>]

结果的大小应与公司表中的行数相同。

我尝试了以下方法，它给出了对象元组（这是有道理的）而不是漂亮的父/子结构：

>>> db.session.query(Company, Employee).filter(Company.id = Employee.company_id).all()

把它转换成我想要的对象结构并不难，只是想看看有没有捷径。

Answer 1

必须在父类中配置关系：

class Company(Base):
    __tablename__ = 'company'

    id = Column(Integer, primary_key=True)
    name = Column(String)
    employees = relationship('Employee', lazy='joined') # <<< Add this line

那么就可以不用join来查询了：

companies = session.query(Company).all()
print(companies[0].employees)

文档： https://docs.sqlalchemy.org/en/13/orm/loading_relationships.html

Answer 2

你可以这样做：

class Company(Base):
    __tablename__ = 'company'

    id = Column(Integer, primary_key=True)
    name = Column(String)

    employees = db.session.query(Company, Employee).filter(Company.id = self.id).all()
    self.employee_list = ['{0} {1}'.format(c.first_name, c.last_name) for c in employees]

然后您可以使用Company.employee_list[0] 访问员工姓名