How to convert with annotate String to Bool with Django ORM

Question

My question is as follows. There is a request

Thread.objects.all().annotate(is_marked=Count('mark', Q(mark__username='gou')))

in SQL it looks like this

SELECT "api_thread"."id",
       "api_thread"."conference_id",
       "api_thread"."title",
       "api_thread"."description",
       "api_thread"."is_closed",
       "api_thread"."is_hidden",
       "api_thread"."is_pinned",
       "api_thread"."has_voting",
       "api_thread"."creator_uid",
       "api_thread"."created_at",
       "api_thread"."expired",
       COUNT("api_thread_mark"."user_id") FILTER (WHERE "auth_user"."username" = 'gou') AS "is_marked"
FROM "api_thread"
         LEFT OUTER JOIN "api_thread_mark" ON ("api_thread"."id" = "api_thread_mark"."thread_id")
         LEFT OUTER JOIN "auth_user" ON ("api_thread_mark"."user_id" = "auth_user"."id")
GROUP BY "api_thread"."id"

What do I need to do to turn a number into an boolean. In SQL it look like

COUNT("api_thread_mark"."user_id") FILTER (WHERE "auth_user"."username" = 'gou') > 0 AS "is_marked"

Can you please share the relevant models? – Willem Van Onsem Dec 19 '19 at 10:47 — Willem Van Onsem, Dec 19 '19 at 10:47

score 5 · Accepted Answer · edited Sep 27 '21 at 14:40

5

Something like this?

from django.db.models import Case, When, Value, BooleanField, Count, Q

is_marked__count = Count('mark', Q(mark__username='gou'))
is_marked_bool = Case(When(is_marked__count__gt=0, then=Value(True)), default=Value(False), output_field=BooleanField())

Thread.objects.all().annotate(is_marked__count=is_marked__count).annotate(is_marked_bool=is_marked_bool

Ref: Conditional Expressions

edited Sep 27 '21 at 14:40

Mark Mishyn

3,921
2
28
30

answered Dec 19 '19 at 10:52

JPG

82,442
19
127
206

1

Thanks, dude. But you forgot **output_field** – Руслан Тютин Dec 19 '19 at 11:17

How to convert with annotate String to Bool with Django ORM

1 Answers1