Django 数据库查询优化：在 filter() 中对同一 QuerySet 赋值给不同变量

Question

我有一个基础的 queryset，我的目标是利用这个基础的 queryset 来构建一个仪表盘统计数据，但需要根据每个变量的不同值进行过滤。

为了更好地理解，我简化了代码：

class Transaction(models.Model):
    """ Invoice model.
     Represents a basic income/outcome transaction. """
    
    user = models.ForeignKey(CustomUser, related_name="transactions", on_delete=models.CASCADE)
    title = models.CharField(max_length=32, verbose_name="Title")
    category = models.ForeignKey(Category, related_name="transactions", on_delete=models.CASCADE, null=True, blank=True)
    operation = models.CharField(max_length=8, choices=OPERATION_TYPE, verbose_name="operation")
    value = models.DecimalField(max_digits=14, decimal_places=2, verbose_name="value")
    date_created = models.DateTimeField(auto_now_add=True, blank=True, null=True) 

class Category(MPTTModel):
    """ Category model. 
    Represents a category where money have been spent/earned."""

    name = models.CharField(max_length=54, unique=True)
    parent = TreeForeignKey("self", on_delete=models.CASCADE, null=True, blank=True, related_name='children')

在上面的模型基础上，我创建了一个 ClassBasedView(ListView) ：

class DashboardView(ListView):
    """ View implements dashboard functionality. """
    model = Transaction
    template_name = "invoices/dashboard.html"
    ordering = "-date_created"


    def get_queryset(self) -> QuerySet[Any]:
        queryset = super().get_queryset()
        queryset = queryset.filter(user=self.request.user).select_related("category__parent")
        return queryset

    def get_context_data(self, **kwargs: Any) -> dict[str, Any]:
        data = super().get_context_data(**kwargs)

        # retrieving all transactons
        transactions_all = self.get_queryset()

        # retrieving incomes/expenses summary for current month
        incomes_this_month = transactions_all.filter(transaction_filter.transaction_date_filter(month="current"), operation="incomes")
        incomes_this_month_sum = incomes_this_month.aggregate(Sum("value")).get("value__sum")
        
        expenses_this_month = transactions_all.filter(transaction_filter.transaction_date_filter(month="current"), operation="expenses")                          
        expenses_this_month_sum = expenses_this_month.aggregate(Sum("value")).get("value__sum")
        
        # retrieving incomes/expenses summary for previous month
        transactions_prev_month = transactions_all.filter(transaction_filter.transaction_date_filter(month="previous"))
   
        incomes_previous_month = transactions_prev_month.filter(operation="incomes")
        incomes_previous_month_sum = incomes_previous_month.aggregate(Sum("value")).get("value__sum")

        expenses_previous_month = transactions_prev_month.filter(operation="expenses")
        expenses_previous_month_sum = expenses_previous_month.aggregate(Sum("value")).get("value__sum")

你可以看到 transaction_date_filter 在 filter() 管理器中。它只是用于日期过滤的 Q() 对象。

通过使用 select_related，我成功去掉了重复的数据，但仍然有一些“相似”的查询，因为 Django ORM 会单独请求查询来过滤 incomes_this_month、expenses_this_month、incomes_previous_month 和 expenses_previous_month，这些都是根据 date= 和 operation= 字段进行的。我知道 filter() 管理器会“重置”缓存的对象，所以我在尝试寻找一种更高效的方法来获取这些数据，而不需要通过额外的查询去访问数据库。我附上了 DjDT 查询计数器的截图，希望能让问题更清楚。

非常感谢对此主题的任何建议。

django orm 数据库优化过滤器 queryset 数据去重查询性能管理器

Django 数据库查询优化：在 filter() 中对同一 QuerySet 赋值给不同变量

1 个回答

撰写回答