如何在Django ORM中调用自定义价格计算函数按年聚合数据？

由于自定义价格计算逻辑复杂无法直接通过ORM实现，推荐以下两种方案：

方案一：Python层面后处理（优先推荐） #

先通过ORM按年份分组获取会员数据，再在Python中遍历计算年度佣金总和，完全兼容复杂的自定义函数逻辑：

1. 先获取按年份分组的基础数据 #

from django.db.models import F

# 提取所有会员记录的年份分组
year_groups = Membership.objects.annotate(
    year=F('benefit_start__year')
).values('year').order_by('year')

2. 遍历分组计算年度佣金 #

提前预关联Package和PackageTier减少数据库查询次数，再调用自定义函数计算每笔会员记录的佣金并累加：

yearly_commissions = []
for group in year_groups:
    target_year = group['year']
    # 预关联相关模型，避免N+1查询
    memberships = Membership.objects.filter(
        benefit_start__year=target_year
    ).select_related('package').prefetch_related('package__packagetier_set')
    
    total_commission = 0.0
    for mem in memberships:
        # 调用自定义价格函数
        retail_price = get_package_price(mem.package, mem.member_count)
        wholesale_price = get_package_wholesale_price(mem.package, mem.member_count)
        # 根据业务逻辑计算单条佣金（示例为差价×数量，需按实际调整）
        single_commission = (retail_price - wholesale_price) * mem.quantity
        total_commission += single_commission
    
    yearly_commissions.append({
        'year': target_year,
        'commission': round(total_commission, 2)  # 按需保留小数
    })

方案二：自定义数据库函数（仅适用于逻辑可转SQL的场景） #

如果get_package_price和get_package_wholesale_price的逻辑可转化为SQL表达式，可通过Django的Case/When实现，直接在ORM聚合中计算：

示例（假设套餐层级定价是区间匹配）：

from django.db.models import Case, When, FloatField, Sum, F

yearly_commissions = Membership.objects.annotate(
    year=F('benefit_start__year'),
    # 用Case/When模拟层级定价逻辑（仅示例，需匹配实际函数逻辑）
    retail_price=Case(
        When(member_count__lte=10, then=F('package__packagetier_set__price')),
        When(member_count__lte=50, then=F('package__packagetier_set__price_2')),
        default=F('package__packagetier_set__price_3'),
        output_field=FloatField()
    ),
    wholesale_price=Case(
        # 同理实现批发价逻辑
        When(member_count__lte=10, then=F('package__packagetier_set__wholesale_price')),
        default=F('package__packagetier_set__wholesale_price_3'),
        output_field=FloatField()
    ),
    single_commission=(F('retail_price') - F('wholesale_price')) * F('quantity')
).values('year').annotate(
    commission=Sum('single_commission')
).order_by('year')

此方案性能更优，但仅适用于逻辑可完全转化为SQL的情况，若函数涉及复杂的Python逻辑（如外部依赖、非规则计算）则不适用。

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