python - примеры - sqlalchemy документация на русском




Применение LIMIT и OFFSET ко всем запросам в SQLAlchemy (2)

Я разрабатываю API с SQLAlchemy (запрос MySQL), и я хотел бы заставить все мои запросы иметь параметры page_size (LIMIT) и page_number (OFFSET).

Есть ли чистый способ сделать это с помощью SQLAlchemy? Возможно, создав какую-нибудь фабрику для создания пользовательского объекта Query? Или, может быть, есть хороший способ сделать это со классом mixin?

Я попытался сделать очевидную вещь, и это не сработало, потому что должны быть вызваны .limit () и .offset () после применения всех условий фильтра:

def q(page=0, page_size=None):
    q = session.query(...)
    if page_size: q = q.limit(page_size)
    if page: q = q.offset(page*page_size)
    return q

Когда я пытаюсь использовать это, я получаю исключение:

sqlalchemy.exc.InvalidRequestError: Query.filter() being called on a Query which already has LIMIT or OFFSET applied. To modify the row-limited results of a  Query, call from_self() first.  Otherwise, call filter() before limit() or offset() are applied.

Не вариант во время этого вопроса, начиная с версии 1.0.0, вы можете использовать события Query, чтобы гарантировать, что методы limit и offset всегда вызываются непосредственно перед компиляцией вашего объекта query после того, как любая манипуляция выполняется пользователями вашего q функция:

from sqlalchemy.event import listen


def q(page=0, page_size=None):
    query = session.query()
    listen(query, 'before_compile', apply_limit(page, page_size), retval=True)
    return query

def apply_limit(page, page_size):
    def wrapped(query):
        if page_size:
            query = query.limit(page_size)
            if page:
                query = query.offset(page * page_size)
        return query
    return wrapped

Попробуйте добавить первый, необходимый аргумент, который должен быть группой фильтров запросов. Таким образом,

# q({'id': 5}, 2, 50)
def q(filters, page=0, page_size=None):
    query = session.query(...).filter_by(**filters)
    if page_size:
        query = query.limit(page_size)
    if page: 
        query = query.offset(page*page_size)
    return query

или,

# q(Model.id == 5, 2, 50)
def q(filter, page=0, page_size=None):
    query = session.query(...).filter(filter)
    if page_size:
        query = query.limit(page_size)
    if page: 
        query = query.offset(page*page_size)
    return query




api-design