如何对Django模型一对多关联字段执行条件聚合计算钱包余额

最高效查询方案 #

核心思路是使用Django ORM的条件聚合能力，直接在数据库层完成余额计算，仅触发1次SQL查询，无N+1问题，也不需要修改现有表结构。

步骤1：导入依赖

from django.db.models import Sum, Q, F, DecimalField

步骤2：执行查询（Django 2.0+ 推荐写法）

wallets_with_balance = Wallet.objects.annotate(
    # 累计充值金额，无充值记录时默认返回0
    total_deposit=Sum(
        "transactions__amount",
        filter=Q(transactions__type="deposit"),
        default=0
    ),
    # 累计提现金额，无提现记录时默认返回0
    total_withdrawal=Sum(
        "transactions__amount",
        filter=Q(transactions__type="withdrawal"),
        default=0
    ),
    # 计算当前余额
    current_balance=F("total_deposit") - F("total_withdrawal")
).all()

查询完成后可以直接访问每个Wallet对象的current_balance属性获取余额：

for wallet in wallets_with_balance:
    print(wallet.id, wallet.current_balance)

低版本Django兼容写法（低于2.0不支持Sum的filter参数时使用）

wallets_with_balance = Wallet.objects.annotate(
    total_deposit=Sum(
        Case(
            When(transactions__type="deposit", then="transactions__amount"),
            default=0,
            output_field=DecimalField()
        )
    ),
    total_withdrawal=Sum(
        Case(
            When(transactions__type="withdrawal", then="transactions__amount"),
            default=0,
            output_field=DecimalField()
        )
    ),
    current_balance=F("total_deposit") - F("total_withdrawal")
).all()

方案优势

• 全程仅执行1次SQL查询，所有计算在数据库侧完成，性能远高于遍历钱包逐次查询、或者拉取全量交易在Python层计算的方案
• 自动处理无交易记录的钱包场景，默认返回余额0，不会出现空值异常
• 完全符合题目约束，不需要修改现有表结构、也不需要调整amount字段为负值的逻辑

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