Битовый запрос с F?

Question

[Владимир Куц]

Есть некая модель User с целочисленным полем alert_flags.
Выполняю побитовые операции с каждым инстансом - все четко:

In [71]: User.objects.all().count()
Out[71]: 554

In [72]: len(filter(lambda x: x!=0, [im.alert_flags & 0b00010000 for im in User.objects.all()]))
Out[72]: 517

In [73]: len(filter(lambda x: x==0, [im.alert_flags & 0b00010000 for im in User.objects.all()]))
Out[73]: 37

Выполнение запроса к ОРМ приводит к совершенно неожиданным результатам:

In [74]: User.objects.filter(alert_flags=F('alert_flags').bitand(0b00010000)).count()
Out[74]: 63

Каким образом можно корректно осуществить побитовый запрос в базу через ОРМ?

Comments

[Владимир Куц]

Pavel Denisov, я понимаю что запрос некорректный - но сама операция в 517 случаях должна выставляться в ненулевое значение (а именно - в 16), и в 37 - в нулевое.
Откуда берется число 63 в этом случае - вообще загадка.
А сама суть вопроса как раз и состоит в том чтобы построить корректную инструкцию в этом случае.

Answer 1

[Roman K]

Какой-то бессмысленный у тебя запрос в SQL получается. Примерно вот такой формулировки:

"Покажи мне всех юзеров, у которых флаги равны флагам & 0b00010000". В выборку попадут все юзеры, такие, что маска не меняет их alert_flags с помощью побитового AND.

Используй annotate, а потом filter. Только правильно.

Comments

[Владимир Куц]

Я знаю что он неправильный. Был бы правильный - этого вопроса не было бы )

[Roman K]

Владимир Куц, Ну так ответ я уже дал.

mask = 0b00010000
User.objects.annotate(masked_flags=F('alert_flags').bitand(mask)).filter(masked_flags=mask)

Этот запрос возьмёт всех юзеров, у которых alert_flags ПОЛНОСТЬЮ отвечает заданной маске

[Владимир Куц]

Roman Kitaev, проблема еще в том что у меня старая версия Джанго

AttributeError: 'ExpressionNode' object has no attribute 'lookup'

За подсказку спасибо - придется видимо с помощью raw query заморачиваться...

[Roman K]

Владимир Куц, тогда можно попробовать с помощью .extra

[Владимир Куц]

Roman Kitaev, дело в том что я собираю запрос по куче параметров, что-то типа:

q = Q()
...
куча разных фильтров в зависимости от всяких состояний
...
q &= Q(alert_flags=F('alert_flags').bitand(0b11101111))
...
res = queryset.filter(q)

так что наверное нужно еще подумать...

[Roman K]

Владимир Куц, Ну так .extra(where=...) будет подставлен через AND

[Владимир Куц]

Roman Kitaev, да - через .extra(where=... получилось