[Django ORM] Group By trong Django

Group by trong Django

Mình sẽ sử dụng User model đã hỗ trợ sẵn của Django nằm trong django.contrib.auth app.

Count

SQL:

SELECT COUNT(*) FROM auth_user

Django ORM:

User.objects.count()

Aggregate Function

Để sử dụng được các aggregate functions, chúng ta cần import các function đó nằm trong mục from django.db.models import Count.
Trong ví dụ này mình sẽ sử dụng Count function, ngoài ra còn cách aggregate function khác như: Max, Min, Sum, Avg.

SQL:

SELECT COUNT(id) AS id__count FROM auth_user

Django ORM:

from django.db.models import Count

User.objects.aggregate(Count('id'))

=> {"id__count": 318}

Bạn cũng có thể thay đổi tên của giá trị trả về. Ví dụ mình sẽ đổi từ id__count sang total. SQL:

SELECT COUNT(id) AS total FROM auth_user

Django ORM:

from django.db.models import Count

User.objects.aggregate(total=Count('id'))

=> {"total": 318}

Group by với aggregate function

SQL:

SELECT
    is_active,
    COUNT(id) AS total
FROM
    auth_user
GROUP BY
    is_active

Django ORM:

User.objects
.values('is_active')
.annotate(total=Count('id'))

• values('is_active'): trường muốn group by.
• annotate(total=Count('id')): group by theo aggregate function nào.

=> Thứ tự gọi là values trước annotate.

Filter a QuerySet với Group by

Ví dụ này sẽ đếm số lượng những user active hoặc inactive có role là staff.

SQL:

SELECT
    is_active,
    COUNT(id) AS total
FROM
    auth_user
WHERE
    is_staff = True
GROUP BY
    is_active

Django ORM:

User.objects
.values('is_active')
.filter(is_staff=True)
.annotate(total=Count('id'))

Sort a QuerySet với Group by

SQL:

SELECT
    is_active,
    COUNT(id) AS total
FROM
    auth_user
GROUP BY
    is_active
ORDER BY
    is_active,
    total

Django ORM:

User.objects
.values('is_active')
.annotate(total=Count('id'))
.order_by('is_active', 'total')

Kết hợp nhiều Aggregate Function với Group By

Ví dụ này sẽ đếm số lượng những user active hoặc inactive và với từng loại active hoặc inactive sẽ lấy ra last_joined.

SQL:

SELECT
    is_active,
    COUNT(id) AS total,
    MAX(date_joined) AS last_joined
FROM
    auth_user
GROUP BY
    is_active

Django ORM:

from django.db.models import Max

User.objects
.values('is_active')
.annotate(
    total=Count('id'),
    last_joined=Max('date_joined'),
)

Group By nhiều trường

Kết quả của ví dụ này sẽ bao gồm is_active, is_staff và số lượng users với mỗi group.

SQL:

SELECT
    is_active,
    is_staff,
    COUNT(id) AS total
FROM
    auth_user
GROUP BY
    is_active,
    is_staff

Django ORM:

User.objects
.values('is_active', 'is_staff')
.annotate(total=Count('id'))

Group By với Expression

SQL:

SELECT
    EXTRACT('year' FROM date_joined),
    COUNT(id) AS total
FROM
    auth_user
GROUP BY
    EXTRACT('year' FROM date_joined)

Django ORM:

User.objects
.values('date_joined__year')
.annotate(total=Count('id'))

Sử dụng với Conditional Aggregation

SQL:

SELECT
    EXTRACT('year' FROM date_joined),

    COUNT(id) FILTER (
        WHERE is_staff = True
    ) AS staff_users,

    COUNT(id) FILTER (
        WHERE is_staff = False
    ) AS non_staff_users

FROM
    auth_user
GROUP BY
    EXTRACT('year' FROM date_joined)

Django ORM:

from django.db.models import F, Q

User.objects
.values('date_joined__year')
.annotate(
    staff_users=(
        Count('id', filter=Q(is_staff=True))
    ),
    non_staff_users=(
        Count('id', filter=Q(is_staff=False))
    ),
)

Sử dụng với Having

SQL:

SELECT
    is_active,
    COUNT(id) AS total
FROM
    auth_user
GROUP BY
    is_active
HAVING
    COUNT(id) > 100

Django ORM:

User.objects
.annotate(year_joined=F('date_joined__year'))
.values('is_active')
.annotate(total=Count('id'))
.filter(total__gt=100)

Distinct với Group By

SQL:

SELECT
    is_active,
    COUNT(id) AS total,
    COUNT(DISTINCT last_name) AS unique_names
FROM
    auth_user
GROUP BY
    is_active

Django ORM:

User.objects
.values('is_active')
.annotate(
    total=Count('id'),
    unique_names=Count('last_name', distinct=True),
)

Join với Group By

SQL:

SELECT
    p.type,
    COUNT(u.id) AS total
FROM
    auth_user u
    JOIN user_profile p ON u.id = p.user_id
GROUP BY
    p.type

Django ORM:

User.objects
.values('user_profile__type')
.annotate(total=Count('id'))

Group By với Many to Many Relationship

SQL:

SELECT
    u.id,
    COUNT(ug.group_id) AS memberships
FROM
    auth_user
    LEFT OUTER JOIN auth_user_groups ug ON (
        u.id = ug.user_id
    )
GROUP BY
    u.id

Django ORM:

User.objects
.annotate(memberships=Count('groups'))
.values('id', 'memberships')

Hi vọng bài viết giúp ích được các bạn trong việc làm quen với Django ORM.

Mình có tham khảo và dịch lại bài viết tại blog:
https://hakibenita.com/django-group-by-sql