Как построить запрос SQLAlchemy, чтоб при значении None фильтра выдавались все строки таблицы?

Question

[REDkiy]

Есть несколько переменных.

a = 23
b = 45
c = 67

Делаю вот такой запрос:

query_result = User.query.filter(User.id == a, User.group == b, User.key == c).all()

Вопрос: Необходимо, чтоб при отсутствии переменной a или при a = None выполнялся запрос такого вида:

query_result = User.query.filter(User.group == b, User.key == c).all()

То есть выдавались строки со всеми возможными User.id.
Возможно ли обойтись без конструкции if ... elif ... else и нескольких разных запросов?

Answer 1

[DDDsa]

from sqlalchemy import or_

query_result = User.query.filter(or_(a == None, User.id == a),
                                 or_(b == None, User.group == b),
                                 or_(c == None, User.key == c))

Comments

[REDkiy]

Не совсем понял логику. Можете поподробней расписать?

[DDDsa]

REDkiy: функция or_ симулирует поведение оператора OR в SQL. Первый аргумент - одно условие, второй аргумент - второе условие.
То есть, для примера, код

SELECT *
FROM Users
WHERE user_id = 1 OR user_id = 2

В ORM будет выглядеть:

Users.query.filter(or_(User.user_id ==1, User.user_id ==2))

Теперь наш случай. Мы подставляем в функцию условие первое: User.id == a и условие второе: a == None. Условие обращается в True если id юзера равняется значению переменной a, либо если переменная a имеет значение None. Понимаете? Если a == None, это условие всегда True, то есть, оно как бы игнорируется, именно то, что вам и нужно:

User.query.filter(or_(User.id == a, a == None) <...>

и далее то же самое по остальным полям.

[DDDsa]

REDkiy: немного подкорректировал ответ

[REDkiy]

Спасибо! Я понял ход ваших мыслей.

Answer 2

[REDkiy]

Сделал так:
1. Создал объект запроса.
query_result = User.query
2. С помощью конструкции if добавляю фильтры и переопределяю запрос, чтоб наложить цепочку фильтров.

if a not None:
    query_result = query_result.filter(User.id == a)
if b not None:
    query_result = query_result.filter(User.group == b)
if c not None:
    query_result = query_result.filter(User.key == c)

3. Получаю из базы данные.
query_result = query_result.all()

Нужна конструктивная критика данного подхода.

Answer 3

[Вячеслав]

Если это код функции, то добавьте параметр a=None. Что то вроде
def func(a=None):

Comments

[REDkiy]

Не понял при чём тут это.

[Вячеслав]

REDkiy: запрос вида
query_result = User.query.filter(User.id == None, User.group == b, User.key == c).all()
отработает как нужно?

[REDkiy]

Нет. Будет пустой результат.