Django ORM 执行复杂查询的技术与实践

小半 • 2023年12月24日下午8:08 • Python • 阅读 225

Django ORM（Object-Relational Mapping）是 Django 框架的核心组件之一，提供了一种高效、直观的方式来处理数据库操作。尽管简单查询在 Django ORM 中相对容易实现，但在面对复杂的数据请求时，需要更深入的了解和技巧。本文将探讨如何使用 Django ORM 执行复杂查询，包括链式查询、聚合查询、子查询以及原生 SQL 查询，旨在提高开发者在数据库处理方面的能力和灵活性。

1. Django ORM 基础

理解 QuerySet

在 Django 中，QuerySet 是进行数据库查询的主要工具，它代表了一个数据库查询的集合，并且可以链式调用，延迟执行。

示例：基本的 QuerySet 操作

from myapp.models import Article

# 筛选操作
articles = Article.objects.filter(pub_date__year=2020)

# 链式调用
articles = articles.exclude(title__startswith="Django").order_by('-pub_date')

2. 复杂查询的构建

链式查询

链式查询是构建复杂查询的强大工具，允许你以声明式的方式逐步构建查询。

from django.db.models import Q

# 使用 Q 对象进行复杂查找
articles = Article.objects.filter(
    Q(title__contains='tutorial') | Q(title__contains='guide')
)

聚合查询

聚合查询用于计算总数、平均值、最大值、最小值等。

from django.db.models import Count, Avg

# 计算每个作者发表的文章数量
author_article_count = Article.objects.values('author').annotate(Count('id'))

3. 子查询和相关查询

在某些情况下，可能需要根据一个查询的结果来构建另一个查询。

示例：子查询

from django.db.models import Subquery

# 首先获取所有作者的最新文章
latest_articles = Article.objects.filter(author=OuterRef('pk')).order_by('-pub_date')
authors = Author.objects.annotate(latest_article_id=Subquery(latest_articles.values('id')[:1]))

关联查询

使用 select_related 和 prefetch_related 进行关联查询可以优化性能。

# select_related 用于“一对一”或“多对一”关系
articles = Article.objects.select_related('author')

# prefetch_related 用于“多对多”或“一对多”关系
articles = Article.objects.prefetch_related('tags')

4. 使用原生 SQL 查询

虽然 Django ORM 强大，但在一些特定情况下，直接使用 SQL 查询可能更合适。

示例：原生 SQL

from django.db import connection

def my_custom_sql():
    with connection.cursor() as cursor:
        cursor.execute("SELECT * FROM myapp_article WHERE title = %s", ['Django'])
        return cursor.fetchall()