Метод count класса Query
Метод count класса Query возвращает
целое число - количество строк, которое вернул бы
текущий запрос без учета лимитов и смещений.
Метод не принимает обязательных параметров и
выполняет отдельный SQL-запрос с функцией
COUNT. Его удобно использовать, когда нужно
узнать общее количество записей в таблице или
число строк, удовлетворяющих условию.
Синтаксис
Query.count()
Пример
Давайте создадим модель Article и посчитаем
общее количество статей в таблице:
from sqlalchemy import create_engine, String
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] = mapped_column(String(100))
status: Mapped[str] = mapped_column(String(50))
engine = create_engine('sqlite:///:memory:')
Base.metadata.create_all(engine)
with Session(engine) as session:
session.add_all([
Article(title='article 1', status='published'),
Article(title='article 2', status='draft'),
Article(title='article 3', status='published'),
])
session.commit()
res = session.query(Article).count()
print(res)
Результат выполнения кода:
3
Пример
Давайте посчитаем количество статей с определенным
статусом при помощи метода filter:
from sqlalchemy import create_engine, String
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] = mapped_column(String(100))
status: Mapped[str] = mapped_column(String(50))
engine = create_engine('sqlite:///:memory:')
Base.metadata.create_all(engine)
with Session(engine) as session:
session.add_all([
Article(title='article 1', status='published'),
Article(title='article 2', status='draft'),
Article(title='article 3', status='published'),
])
session.commit()
res = session.query(Article).filter(Article.status == 'published').count()
print(res)
Результат выполнения кода:
2
Пример
Давайте посмотрим, какой SQL-запрос формирует
метод count при включенном логировании:
from sqlalchemy import create_engine, String
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] = mapped_column(String(100))
status: Mapped[str] = mapped_column(String(50))
engine = create_engine('sqlite:///:memory:', echo=True)
Base.metadata.create_all(engine)
with Session(engine) as session:
session.add_all([
Article(title='article 1', status='published'),
Article(title='article 2', status='draft'),
])
session.commit()
res = session.query(Article).filter(Article.status == 'published').count()
print(res)
Итоговый SQL запрос:
SELECT count(*) AS count_1
FROM articles
WHERE articles.status = ?
