通过在django orm中使用多对多关系来执行组?

iswrvxsc  于 2021-08-09  发布在  Java
关注(0)|答案(1)|浏览(312)

我有以下型号:

class Tag(models.Model):
    name = models.CharField(max_length=255, unique=True, default="")

class Problem(models.Model):
    tags = models.ManyToManyField(Tag, related_name="problems")
    index = models.CharField(max_length=5)
    name = models.CharField(max_length=255)
    rating = models.IntegerField(default=-1)

我要执行以下查询:

SELECT
    "p"."index",
    "p"."name",
    "p"."rating"
FROM
    problem p
WHERE
    p.id IN (
        SELECT
            pt.problem_id
        FROM
            problem_tags pt
            JOIN tag t ON pt.tag_id = t.id
        WHERE
            t.name IN ('math', 'binary search', 'implementation')
        GROUP BY
            pt.problem_id
        HAVING
            COUNT(*) = 3
    )
ORDER BY
    rating,
    index;

我用了这样的方法:

Problem.tags.through.objects.values("problem_id").filter(
    tag__name__in=("math", "binary search", "implementation")
).annotate(count=models.Count("*")).filter(count=3)

但它发出以下查询 COUNT(*)SELECT 是多余和错误的:

SELECT
    "api_problem_tags"."problem_id",
    COUNT(*) AS "count"
FROM
    "api_problem_tags"
    INNER JOIN "api_tag" ON ("api_problem_tags"."tag_id" = "api_tag"."id")
WHERE
    "api_tag"."name" IN ('math', 'binary search', 'implementation')
GROUP BY
    "api_problem_tags"."problem_id"
HAVING
    COUNT(*) = 3

我怎样才能摆脱第一个 COUNT(*)

rdrgkggo

rdrgkggo1#

我想你可以用 Count 以及 filter 参数。

Problem.objects.annotate(
   tag_count=Count(
       'tags',
       filter=Q(tags__name__in=["math", "binary search", "implementation"],
   )
).filter(tag_count=3)

相关问题