Django如何编写PositiveDecimalField

时间:2016-07-06 14:53:01

标签: django django-models django-validation

我想写PositiveDecimalField。

这是我的代码:

from django.core.exceptions import ValidationError
from django.utils.translation import ugettext_lazy as _

from django.db.models import DecimalField
from decimal import *

def validate_positive_decimal(value):
    try:
        decimal_value = Decimal(value)

        if decimal_value < 0:
            raise ValidationError(_('{val} should be a positive decimal number'.format(val = decimal_value)), code='negative decimal')

    except (ValueError, TypeError):
        raise ValidationError(_('Enter a valid decimal or integer value'), code='invalid decimal')


class PositiveDecimalField(DecimalField):
    default_validators = [validate_positive_decimal]

    def __init__(self, *args, **kwargs):

        super(DecimalField, self).__init__(*args, **kwargs)

    def validators(self):
        return super(PositiveDecimalField, self).validators + [validate_positive_decimal]

然后,在我的models.py

class Service(models.Model):
    service_rate = PositiveDecimalField(max_digits = 6, decimal_places=2, blank=True) # e.g. 125.25 for water (demo phase), -125.25 should not be accepted

我得到的错误是:

super(DecimalField, self).__init__(*args, **kwargs)
TypeError: __init__() got an unexpected keyword argument 'max_digits'

我在文档中查看了Django的DecimalField的源代码。我还尝试将验证器validate_positive_decimal传递给DecimalField,但它没有对它进行验证。

我需要在这做什么?

2 个答案:

答案 0 :(得分:2)

我通常将这些字段定义为:

pos_float_field = models.FloatField(validators=[MinValueValidator(0.0)])

这样更干净。

Doc:MinValueValidator

答案 1 :(得分:1)

您对super()__init__的来电应为

super(PositiveDecimalField, self).__init__(*args, **kwargs)

而不是

super(DecimalField, self).__init__(*args, **kwargs)