ModelFormset批量保存Transaction时关联Share余额未累加更新问题

你碰到的这个问题我之前也踩过坑，核心是Django ORM的对象缓存机制在作祟。当你通过ModelFormset循环处理每一笔交易时，每个self.share其实都是从初始查询结果里拿到的同一个对象实例——它的balance值始终停留在数据库的初始状态，完全没跟上前面交易对余额的更新。

举个直观的例子：假设某Share初始余额是100，你提交两笔+50的交易。第一笔交易里你把余额改成150并保存到数据库，但第二笔交易里的share对象还是缓存里的旧实例，它的balance还是100，最后数据库里的余额就变成了100+50=150，而不是预期的200。

下面给你几个可行的解决方案，按推荐程度排序：

方案1：用F表达式原子更新（最安全，推荐） #

如果你的系统存在并发操作的可能（比如多个用户同时更新同一个Share的余额），直接用Django的F()表达式是最优解——它会跳过ORM的对象缓存，直接在数据库层面执行原子更新，彻底避免竞态条件。

修改你的Transaction.save()方法：

from django.db.models import F

def save(self, *args, **kwargs):
    if not self.pk:
        # 保留你原有的初始化逻辑
        ...
    if self.share:
        if self in self.share.transaction_set.all():
            logging.error('Transaction %s already posted' % self.id)
            return False
        amt = self.amt if self.db_cr == 'cr' else -self.amt
        # 直接在数据库层面更新余额，绕过本地缓存
        Share.objects.filter(pk=self.share.pk).update(balance=F('balance') + amt)
        # 可选：刷新本地share对象的余额，避免后续代码使用旧值
        self.share.refresh_from_db()
    # 别忘了调用父类的save方法，保存Transaction本身
    super().save(*args, **kwargs)

方案2：每次更新前重新查询Share对象 #

如果你的系统没有并发场景，或者暂时不想用F表达式，也可以在更新余额前，重新从数据库拉取最新的Share实例：

def save(self, *args, **kwargs):
    if not self.pk:
        ...
    if self.share:
        if self in self.share.transaction_set.all():
            logging.error('Transaction %s already posted' % self.id)
            return False
        # 关键：重新查询最新的Share对象，拿到当前数据库里的余额
        share = Share.objects.get(pk=self.share.pk)
        amt = self.amt if self.db_cr == 'cr' else -self.amt
        share.balance = share.balance + amt
        share.save()
        # 同步本地self.share的余额值
        self.share.balance = share.balance
    super().save(*args, **kwargs)

方案3：视图层批量处理（性能优化） #

如果你的表单集经常要处理大量交易，上面两种方法每次处理一笔交易就操作一次数据库，性能会有点差。可以在视图层先收集所有交易对应的金额，然后批量更新：

修改视图里的表单处理逻辑：

if tran_formset.is_valid():
    # 先收集所有待保存的交易和对应的Share金额
    share_updates = {}
    pending_transactions = []
    for form in tran_formset:
        tran = form.save(commit=False)
        tran.account = account
        tran.origin = 'tt'
        pending_transactions.append(tran)
        
        if tran.share:
            amt = tran.amt if tran.db_cr == 'cr' else -tran.amt
            # 累加同一个Share的总金额
            share_pk = tran.share.pk
            share_updates[share_pk] = share_updates.get(share_pk, 0) + amt
    
    # 批量保存所有交易，减少数据库写入次数
    Transaction.objects.bulk_create(pending_transactions)
    
    # 批量更新Share余额
    from django.db.models import F
    for share_pk, total_amt in share_updates.items():
        Share.objects.filter(pk=share_pk).update(balance=F('balance') + total_amt)

这种方法能大幅减少数据库交互次数，适合批量处理大量交易的场景。

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