Метод union класса Select
Метод union класса Select объединяет результаты
двух или более запросов SELECT в один результирующий набор.
По умолчанию метод удаляет дублирующиеся строки, то есть
работает как SQL-оператор UNION. Метод принимает
произвольное количество других запросов Select и
возвращает новый объект CompoundSelect. Чтобы
сохранить все строки, включая дубликаты, используйте
метод union_all.
Синтаксис
select(...).union(*others)
Пример
Давайте объединим результаты двух запросов к таблице
articles и получим уникальные статусы:
from sqlalchemy import create_engine, select
from sqlalchemy.orm import DeclarativeBase, Mapped, mapped_column, Session
class Base(DeclarativeBase):
pass
class Article(Base):
__tablename__ = 'articles'
id: Mapped[int] = mapped_column(primary_key=True)
title: Mapped[str]
text: Mapped[str]
status: Mapped[str]
num: Mapped[int]
engine = create_engine('sqlite:///:memory:')
Base.metadata.create_all(engine)
with Session(engine) as session:
session.add_all([
Article(title='article 1', text='text 1', status='draft', num=1),
Article(title='article 2', text='text 2', status='published', num=2),
Article(title='article 3', text='text 3', status='draft', num=3),
])
session.commit()
stmt1 = select(Article.status).where(Article.num < 3)
stmt2 = select(Article.status).where(Article.num > 1)
res = session.execute(stmt1.union(stmt2)).all()
print(res)
Результат выполнения кода:
[('draft',), ('published',)]
Пример
Давайте объединим три запроса и получим уникальные заголовки:
from sqlalchemy import create_engine, select
from sqlalchemy.orm import DeclarativeBase, Mapped, mapped_column, Session
class Base(DeclarativeBase):
pass
class Article(Base):
__tablename__ = 'articles'
id: Mapped[int] = mapped_column(primary_key=True)
title: Mapped[str]
text: Mapped[str]
status: Mapped[str]
num: Mapped[int]
engine = create_engine('sqlite:///:memory:')
Base.metadata.create_all(engine)
with Session(engine) as session:
session.add_all([
Article(title='article 1', text='text 1', status='draft', num=1),
Article(title='article 2', text='text 2', status='published', num=2),
Article(title='article 3', text='text 3', status='draft', num=3),
])
session.commit()
stmt1 = select(Article.title).where(Article.num == 1)
stmt2 = select(Article.title).where(Article.num == 2)
stmt3 = select(Article.title).where(Article.num == 3)
res = session.execute(stmt1.union(stmt2, stmt3)).all()
print(res)
Результат выполнения кода:
[('article 1',), ('article 2',), ('article 3',)]
Пример
Давайте сравним поведение union и union_all
на одинаковых запросах:
from sqlalchemy import create_engine, select
from sqlalchemy.orm import DeclarativeBase, Mapped, mapped_column, Session
class Base(DeclarativeBase):
pass
class Article(Base):
__tablename__ = 'articles'
id: Mapped[int] = mapped_column(primary_key=True)
title: Mapped[str]
text: Mapped[str]
status: Mapped[str]
num: Mapped[int]
engine = create_engine('sqlite:///:memory:')
Base.metadata.create_all(engine)
with Session(engine) as session:
session.add_all([
Article(title='article 1', text='text 1', status='draft', num=1),
Article(title='article 2', text='text 2', status='published', num=2),
])
session.commit()
stmt1 = select(Article.status).where(Article.num == 1)
stmt2 = select(Article.status).where(Article.num == 1)
res_union = session.execute(stmt1.union(stmt2)).all()
res_union_all = session.execute(stmt1.union_all(stmt2)).all()
print(res_union)
print(res_union_all)
Результат выполнения кода:
[('draft',)]
[('draft',), ('draft',)]
Пример
Давайте посмотрим итоговый SQL-запрос, который генерирует
метод union:
from sqlalchemy import create_engine, select
from sqlalchemy.orm import DeclarativeBase, Mapped, mapped_column
class Base(DeclarativeBase):
pass
class Article(Base):
__tablename__ = 'articles'
id: Mapped[int] = mapped_column(primary_key=True)
title: Mapped[str]
text: Mapped[str]
status: Mapped[str]
num: Mapped[int]
stmt1 = select(Article.title).where(Article.num == 1)
stmt2 = select(Article.title).where(Article.num == 2)
stmt = stmt1.union(stmt2)
print(stmt)
Итоговый SQL запрос:
SELECT articles.title
FROM articles
WHERE articles.num = :num_1
UNION
SELECT articles.title
FROM articles
WHERE articles.num = :num_2
