Django创建自定义模型字段

时间:2015-02-15 17:55:50

标签: python django django-models django-forms django-1.7

我正在尝试在Django中创建一个自定义字段,它将采用十进制货币值(例如:£1.56)并将其作为整数(例如:156)保存在数据库中以存储货币值。

这是我到目前为止(我已经将固定值用于测试)

class CurrencyField(models.DecimalField):
    __metaclass__ = models.SubfieldBase

    def get_internal_type(self):
        return 'PositiveIntegerField'

    def to_python(self, value):   
        print "CurrentField to_python"

        return Decimal(value)/100

    def get_db_prep_value(self, value, connection, prepared=False):
        print "CurrentField get_db_prep_value"
        if value is None:
            return value

        print type(value)

        return int(value*100)

我关注Python 1.7 documentation for creating custom model fields

在上面的代码中,当我使用此字段保存模型时,永远不会调用get_db_prep_value方法。 to_python方法工作正常(如果我在DB中手动输入值将返回正确的小数),但它永远不会保存正确的值。

我也尝试使用get_prep_value,结果相同。

如何保存正确的值?

1 个答案:

答案 0 :(得分:7)

Integer based MoneyField

我一直在使用这个Django片段,但后来意识到在保存数据库之前转换为整数时会引入浮点伪像。

此外,从Django 1.9 SubfieldBase has been deprecated开始,我们应该使用Field.from_db_value。所以我编辑了代码以解决这两个问题。

class CurrencyField(models.IntegerField):
  description = "A field to save dollars as pennies (int) in db, but act like a float"

  def get_db_prep_value(self, value, *args, **kwargs):
    if value is None:
      return None
    return int(round(value * 100))

  def to_python(self, value):
    if value is None or isinstance(value, float):
      return value
    try:
      return float(value) / 100
    except (TypeError, ValueError):
      raise ValidationError("This value must be an integer or a string represents an integer.")

  def from_db_value(self, value, expression, connection, context):
    return self.to_python(value)

  def formfield(self, **kwargs):
    from django.forms import FloatField
    defaults = {'form_class': FloatField}
    defaults.update(kwargs)
    return super(CurrencyField, self).formfield(**defaults)