Django开发分期应用时如何正确计算客户剩余余额

你代码里的余额统计混乱主要来自以下几个逻辑缺陷：

• 更新已存在的Payment记录时，save方法会重复累加amount：你没有区分是新增Payment还是修改已有Payment，比如某条Payment原来金额是100，已经累计到collect_amount，修改成200的时候直接加200，就会多算100的累计值
• Customer的save方法没有做空判断：如果prop_select为NULL时，直接访问self.prop_select.price会抛出AttributeError异常
• 冗余字段同步风险：remaining是完全可以通过prop_select.price和collect_amount计算得到的存储字段，一旦同步逻辑有遗漏就会出现数据不一致
• 批量操作失效：调用Django的QuerySet批量更新/删除方法（比如Payment.objects.filter(...).delete()）时，不会触发模型层的save/delete方法，会导致collect_amount统计错误

方案1：最小改动修复原有逻辑（保留现有字段设计） #

from django.db import models
from django.core.exceptions import ValidationError

class Property(models.Model):
    area = models.CharField(max_length=255)
    price = models.IntegerField(default=0)


class Customer(models.Model):
    name = models.CharField(max_length=255)
    prop_select = models.ForeignKey(Property, on_delete=models.SET_NULL, null=True)
    remaining = models.IntegerField(default=0)
    collect_amount = models.IntegerField(default=0)

    def save(self, *args, **kwargs):
        # 新增空判断
        if self.prop_select:
            self.remaining = self.prop_select.price - self.collect_amount
        else:
            self.remaining = 0
        super().save(*args, **kwargs)

class Payment(models.Model):
    customer = models.ForeignKey(Customer, null=True, on_delete=models.SET_NULL, blank=True, related_name='payments')
    amount = models.IntegerField(default=0)
    
    def save(self, *args, **kwargs):
        if not self.customer:
            raise ValidationError("还款记录必须关联客户")
        # 区分新增还是更新
        if self.pk:
            # 是更新操作，先减去旧的金额，再加新的金额
            old_payment = Payment.objects.get(pk=self.pk)
            self.customer.collect_amount = self.customer.collect_amount - old_payment.amount + self.amount
        else:
            # 是新增操作，直接加当前金额
            self.customer.collect_amount += self.amount
        self.customer.save()
        super().save(*args, **kwargs)

    def delete(self, *args, **kwargs):
        if self.customer:
            self.customer.collect_amount -= self.amount
            self.customer.save()
        super().delete(*args, **kwargs)

方案2：更稳妥的优化方案（消除冗余字段，避免同步问题） #

直接去掉remaining和collect_amount存储字段，改用动态属性计算，从根源避免数据不一致：

from django.db import models
from django.db.models import Sum

class Property(models.Model):
    area = models.CharField(max_length=255)
    price = models.IntegerField(default=0)


class Customer(models.Model):
    name = models.CharField(max_length=255)
    prop_select = models.ForeignKey(Property, on_delete=models.SET_NULL, null=True)

    @property
    def collect_amount(self):
        # 动态统计所有还款总额
        return self.payments.aggregate(total=Sum('amount'))['total'] or 0

    @property
    def remaining(self):
        # 动态计算剩余待还金额
        if not self.prop_select:
            return 0
        return self.prop_select.price - self.collect_amount

class Payment(models.Model):
    customer = models.ForeignKey(Customer, null=True, on_delete=models.SET_NULL, blank=True, related_name='payments')
    amount = models.IntegerField(default=0)

这个方案不需要处理任何字段同步逻辑，不管是新增、修改、删除还款，甚至批量操作，统计结果永远是准确的，唯一的成本是每次访问这两个属性的时候会执行一次聚合查询，对于分期类应用的访问频率来说完全可以接受。如果性能有要求可以加缓存策略。

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