如何通过Django的annotate按规则更新模型日期字段年份并生成新日期？

问题描述 #

我有一个Django模型：

class MyModel(models.Model):
    """My Model that stores date."""

    date = models.DateField()

我的API接收一个review_month参数，需要获取MyModel的所有条目，并按规则替换date字段的年份：当date的月份大于review_month的月份时，年份设为当前年，否则设为上一年。我需要为查询集中的每个条目生成包含原日期的月、日和新年份的updated_date字段，且该字段需为DateField类型以支持排序。目前我编写的代码仅能单独得到更新后的年份，无法生成目标日期字段：

MyModel.objects.annotate(
   updated_year=Case(
       When(
           Q(review_date__month__gt=review_month.month), then=previous_year),
           default=current_year,
           output_field=IntegerField(),
       ),
   )

实现方案 #

可以利用Django的MakeDate数据库函数，结合年份判断逻辑，将计算后的年份与原日期的月、日组合成合法的DateField类型字段。

步骤1：导入所需依赖 #

from datetime import datetime
from django.db.models import Case, When, IntegerField, F
from django.db.models.functions import MakeDate, ExtractMonth, ExtractDay

步骤2：编写查询集逻辑 #

先定义当前年和上一年的变量：

current_year = datetime.now().year
previous_year = current_year - 1

然后通过annotate生成目标字段：

# 方式1：先计算updated_year，再组合成日期
queryset = MyModel.objects.annotate(
    updated_year=Case(
        When(
            date__month__gt=review_month.month,
            then=current_year
        ),
        default=previous_year,
        output_field=IntegerField()
    ),
    updated_date=MakeDate(
        F('updated_year'),
        ExtractMonth('date'),
        ExtractDay('date')
    )
)

或者简化为直接在MakeDate中嵌套年份判断：

# 方式2：直接在MakeDate中嵌入年份逻辑，省去单独的updated_year字段
queryset = MyModel.objects.annotate(
    updated_date=MakeDate(
        Case(
            When(
                date__month__gt=review_month.month,
                then=current_year
            ),
            default=previous_year,
            output_field=IntegerField()
        ),
        ExtractMonth('date'),
        ExtractDay('date')
    )
)

关键说明 #

• ExtractMonth('date')和ExtractDay('date')分别提取原date字段的月份和日期数值；
• MakeDate函数将年份、月份、日期三个数值拼接为标准日期对象，输出类型为DateField，完全支持排序、过滤等ORM操作；
• 生成的updated_date可直接用于排序：

  sorted_queryset = queryset.order_by('updated_date')
  

内容的提问来源于stack exchange，提问作者Noyan Aziz