sqlalchemy2.0 先union后分页，然后joinedload|极客教程

sqlalchemy2.0 先union后分页，然后joinedload

在使用 SQLAlchemy 进行数据库操作时，有时候我们需要先对多个查询结果进行合并，然后再进行分页操作。同时，我们可能还需要在查询的同时进行关联加载（joinedload）来减少数据库查询次数，提高查询效率。本文将详细介绍如何在 SQLAlchemy 2.0 中实现先 union 后分页，然后再进行 joinedload。

案例背景

假设我们有一个学生表和一个老师表，学生和老师是多对多的关系。现在我们需要查询所有学生和老师的信息，并按照学生姓名进行分页显示。同时，我们希望在查询结果中关联加载学生和老师的其他信息，以减少数据库查询次数。

实现步骤

创建数据库模型

首先，我们需要定义学生表（Student）和老师表（Teacher）的模型，并创建学生和老师之间的多对多关系。

from sqlalchemy import Table, Column, Integer, String, ForeignKey
from sqlalchemy.orm import relationship
from sqlalchemy.ext.declarative import declarative_base

Base = declarative_base()

class Student(Base):
    __tablename__ = 'students'

    id = Column(Integer, primary_key=True)
    name = Column(String)
    teachers = relationship('Teacher', secondary='students_teachers')

class Teacher(Base):
    __tablename__ = 'teachers'

    id = Column(Integer, primary_key=True)
    name = Column(String)
    students = relationship('Student', secondary='students_teachers')

students_teachers = Table('students_teachers', Base.metadata,
    Column('student_id', ForeignKey('students.id'), primary_key=True),
    Column('teacher_id', ForeignKey('teachers.id'), primary_key=True)
)

创建查询

接下来，我们需要创建一个查询，先将学生和老师的信息 union 起来，然后再进行分页操作。

from sqlalchemy import create_engine, select
from sqlalchemy.orm import sessionmaker
from sqlalchemy.sql import union

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

# 创建查询
query1 = select([Student.id, Student.name]).\
    where(Student.name.like('A%'))

query2 = select([Teacher.id, Teacher.name]).\
    where(Teacher.name.like('B%'))

full_query = union(query1, query2).alias()

# 分页查询
page_num = 1
page_size = 10
results = session.execute(full_query.limit(page_size).offset((page_num - 1) * page_size))

# 输出结果
for result in results:
    print(result)

关联加载

最后，我们可以使用 joinedload 方法来关联加载学生和老师的其他信息，以减少数据库查询次数。

from sqlalchemy.orm import joinedload

# 创建查询
query1 = select([Student.id, Student.name]).\
    where(Student.name.like('A%'))

query2 = select([Teacher.id, Teacher.name]).\
    where(Teacher.name.like('B%'))

full_query = union(query1, query2).alias()

# 分页查询并关联加载
page_num = 1
page_size = 10
results = session.query(full_query).options(joinedload(Student.teachers), joinedload(Teacher.students)).\
    limit(page_size).offset((page_num - 1) * page_size)

# 输出结果
for result in results:
    print(result)