Django Rest Framework:序列化/反序列化计算字段

时间:2015-04-26 21:04:09

标签: django django-rest-framework

我刚刚开始使用Django& Django REST框架。我有以下型号:

class Account(models.Model):
    name = models.CharField(max_length=100, blank=False)
    vat_perc = models.DecimalField(max_digits=4, decimal_places=2)
    def __str__(self):
        return "ACCOUNT: {0} -- {1}".format(self.name, str(self.vat_perc))


class Entry(models.Model):
    account = models.ForeignKey(Account)
    description = models.CharField(max_length=100)
    taxable_income = models.DecimalField(max_digits=10, decimal_places=2)
    total_amount = models.DecimalField(max_digits=10, decimal_places=2, null=True)

    def save(self, *args, **kwargs):
        selected_vat = Account.objects.get(pk=self.account).vat_perc
        self.total_amount = self.taxable_income * (100.00+selected_vat)/100.00
        super(Entry, self).save(*args, **kwargs)

这个想法是读取用户刚刚选择的vat_perc记录中的account值,并执行计算以确定total_amount值,然后将其保存在entry在数据库上的记录(我知道有些人会认为这是因为数据库中的数据重复而不是最理想的;无论如何都请关注我)。

total_amount字段应根据要求定期序列化。相反,序列化程序不应该为反序列化做任何事情,因为如果发生创建或修改,模型中的save方法的重写将负责更新值。如果我正确获得文档,则所有这一切都意味着将序列化程序类中的total_amount字段设置为read_only

现在,这些是我的序列化器:

class AccountSerializer(serializers.ModelSerializer):
    class Meta:
        model = Account
        fields = ('id', 'name', 'vat_perc',)


class EntrySerializer(serializers.ModelSerializer):
    class Meta:
        model = Entry
        fields = ('id', 'account', 'description', 'taxable_income', 'total_amount',)

    total_amount = serializers.ReadOnlyField()
    # alternatively: total_amount = serializers.FloatField(read_only=True)

但这是我得到的错误:

  

在致电TypeError时获得Entry.objects.create()。这可能是因为序列化程序类上的可写字段不是Entry.objects.create()的有效参数。您可能需要将该字段设置为只读,或重写EntrySerializer.create()方法以正确处理此字段。   原始异常文本是:int()参数必须是字符串,类似字节的对象或数字,而不是“帐户”。

最后一句话对我来说听起来特别模糊。我弄错了吗?任何提示? 提前谢谢。

1 个答案:

答案 0 :(得分:4)

感谢Claudiu。在序列化程序类和SlugRelatedField类型中使用了decimal.Decimal而不是float,因为我做得很少。以下代码现在可以使用:

class Account(models.Model):
    name = models.CharField(max_length=100, blank=False)
    vat_perc = models.DecimalField(max_digits=4, decimal_places=2)
    def __str__(self):
        return "ACCOUNT: {0} -- {1}".format(self.name, str(self.vat_perc))


class Entry(models.Model):
    account = models.ForeignKey(Account)
    description = models.CharField(max_length=100)
    taxable_income = models.DecimalField(max_digits=10, decimal_places=2)
    total_amount = models.DecimalField(max_digits=10, decimal_places=2, null=True)

    def save(self, *args, **kwargs):
        self.total_amount = self.taxable_income * (decimal.Decimal(100.00) + self.account.vat_perc) / decimal.Decimal(100.00)
        super(Entry, self).save(*args, **kwargs)

serializers.py

class AccountSerializer(serializers.ModelSerializer):
    class Meta:
         model = Account
         fields = ('id', 'name', 'vat_perc',)


class EntrySerializer(serializers.ModelSerializer):
    class Meta:
        model = Entry
        fields = ('id', 'account', 'description', 'taxable_income', 'total_amount',)

    total_amount = serializers.ReadOnlyField()
    account = serializers.SlugRelatedField(queryset=Account.objects.all(), slug_field="vat_perc")