如何向Django QuerySet中添加自定义字典数据

嘿，我来帮你搞定这个问题！首先得澄清一个小误区：你用.values()得到的QuerySet不是字典，它是Django封装的可迭代结果集，里面是一个个字典，但本身是QuerySet对象，所以不能直接和普通字典合并。不过要实现你要的“如果没有当前年份就添加0值条目”的需求，有两个很实用的方法，我给你详细说下：

方法一：转成Python列表后操作（最直观简单） #

这个方法适合数据量不大的场景（比如年份统计这种数据肯定不多），操作起来特别容易：

1. 先把QuerySet转成普通的Python列表，这样就能像操作普通列表一样添加字典了：

from datetime import datetime
from django.db.models import Q, Sum, Count, TruncYear

# 你的原QuerySet代码不变
years = (Transactions.objects
                .annotate(year=TruncYear('timestamp'))
                .values('year')
                .order_by('-year')
                .annotate(total=Sum('amount'))
                .annotate(earnings=Sum('amount', filter=Q(amount__gt=0)))
                .annotate(spent=Sum('amount', filter=Q(amount__lt=0)))
                .annotate(count=Count('id'))
                .values('year', 'total', 'count', 'earnings', 'spent'))

# 转成列表
year_data = list(years)
# 获取当前年份
current_year = datetime.now().year

2. 检查列表里有没有包含当前年份的条目：
   注意哦，你的QuerySet里的year是TruncYear返回的datetime对象（比如datetime(2024, 1, 1, 0, 0)），所以对比的时候要取它的.year属性：

# 检查是否存在当前年份的条目
has_current_year = any(item['year'].year == current_year for item in year_data)

3. 如果没有的话，就往列表里append一个格式匹配的字典：

if not has_current_year:
    # 这里的year要和原QuerySet的格式一致，用当年1月1日的datetime对象
    year_data.append({
        'year': datetime(current_year, 1, 1),
        'total': 0,
        'count': 0,
        'earnings': 0,
        'spent': 0
    })

4. （可选）如果需要保持按年份倒序的排序，可以重新给列表排序：

year_data.sort(key=lambda x: x['year'], reverse=True)

之后你直接用year_data这个列表就可以了，里面包含了原有的数据和新增的当前年份0值条目。

方法二：在数据库层面用Union实现（保持QuerySet特性） #

如果你想保持结果还是QuerySet对象（比如要继续用Django的QuerySet方法），可以用数据库的Union操作，把原QuerySet和一个构造的“当前年份0值”QuerySet合并：

1. 先构造一个只包含当前年份0值的空QuerySet：

from django.db.models import Value, IntegerField, DateTimeField
from datetime import datetime
from django.db.models import Q, Sum, Count, TruncYear

current_year = datetime.now().year
# 构造和原QuerySet中year格式一致的datetime对象
current_year_start = datetime(current_year, 1, 1)

# 构造空的当前年份条目QuerySet
current_year_entry = Transactions.objects.none().annotate(
    year=Value(current_year_start, output_field=DateTimeField()),
    total=Value(0, output_field=IntegerField()),
    count=Value(0, output_field=IntegerField()),
    earnings=Value(0, output_field=IntegerField()),
    spent=Value(0, output_field=IntegerField())
).values('year', 'total', 'count', 'earnings', 'spent')

2. 检查原QuerySet是否已经有当前年份的条目，没有的话就用Union合并：

# 你的原QuerySet代码不变
years = (Transactions.objects
                .annotate(year=TruncYear('timestamp'))
                .values('year')
                .order_by('-year')
                .annotate(total=Sum('amount'))
                .annotate(earnings=Sum('amount', filter=Q(amount__gt=0)))
                .annotate(spent=Sum('amount', filter=Q(amount__lt=0)))
                .annotate(count=Count('id'))
                .values('year', 'total', 'count', 'earnings', 'spent'))

# 检查原QuerySet是否存在当前年份
has_current_year = years.filter(year=current_year_start).exists()

if not has_current_year:
    # all=True表示保留所有条目（这里不会有重复，只是保险）
    years = years.union(current_year_entry, all=True)
    # 合并后重新排序，因为Union会打乱原有的order_by
    years = years.order_by('-year')

这样得到的years还是一个QuerySet对象，可以继续用Django的QuerySet方法，比如分页、过滤等。

注意事项 #

• 一定要保证新增条目的字段类型和原QuerySet完全一致：比如原QuerySet里的year是datetime对象，你就不能直接传数字2024，必须构造一个当年1月1日的datetime，否则对比或者合并的时候会出错。
• 方法一因为转成了列表，会把所有数据加载到内存里，但年份统计这种数据量很小，完全不用担心性能问题；方法二更适合大数据量的场景，因为QuerySet是惰性求值的，不会一次性加载所有数据。

备注：内容来源于stack exchange，提问作者donbolli