Как ограничить максимальное значение числового поля в модели Django? - PullRequest
Новая
       17

Как ограничить максимальное значение числового поля в модели Django?

136 голосов
/ 11 мая 2009

Django имеет различные числовые поля, доступные для использования в моделях, например, DecimalField и PositiveIntegerField . Хотя первое может быть ограничено количеством сохраненных десятичных знаков и общим количеством сохраненных символов, есть ли способ ограничить его сохранением только чисел в определенном диапазоне, например 0,0-5,0?

Если это не так, есть ли способ ограничить PositiveIntegerField, чтобы хранить только, например, числа до 50?

Обновление: теперь, когда ошибка 6845 закрыта , этот вопрос StackOverflow может быть спорным. - сампаблокупер

Ответы [ 6 ]

272 голосов
/ 19 августа 2012

Вы можете использовать встроенные валидаторы Django - 

from django.db.models import IntegerField, Model
from django.core.validators import MaxValueValidator, MinValueValidator

class CoolModelBro(Model):
    limited_integer_field = IntegerField(
        default=1,
        validators=[
            MaxValueValidator(100),
            MinValueValidator(1)
        ]
     )

Редактировать : Хотя эти работают только при использовании модели в ModelForm , но не при использовании модели "самостоятельно", вздох.

125 голосов
/ 11 мая 2009

Вы также можете создать собственный тип поля модели - см. http://docs.djangoproject.com/en/dev/howto/custom-model-fields/#howto-custom-model-fields

В этом случае вы можете «унаследовать» от встроенного IntegerField и переопределить его логику проверки.

Чем больше я думаю об этом, тем больше понимаю, насколько это полезно для многих приложений Django. Возможно, тип IntegerRangeField мог бы быть представлен как патч для разработчиков Django, чтобы рассмотреть возможность добавления в trunk.

Это работает для меня: 

from django.db import models

class IntegerRangeField(models.IntegerField):
    def __init__(self, verbose_name=None, name=None, min_value=None, max_value=None, **kwargs):
        self.min_value, self.max_value = min_value, max_value
        models.IntegerField.__init__(self, verbose_name, name, **kwargs)
    def formfield(self, **kwargs):
        defaults = {'min_value': self.min_value, 'max_value':self.max_value}
        defaults.update(kwargs)
        return super(IntegerRangeField, self).formfield(**defaults)

Тогда в своем классе модели вы бы использовали его следующим образом (поле является модулем, в который вы поместили вышеуказанный код): 

size = fields.IntegerRangeField(min_value=1, max_value=50)

ИЛИ для диапазона отрицательных и положительных значений (например, диапазона осциллятора): 

size = fields.IntegerRangeField(min_value=-100, max_value=100)

Что было бы здорово, если бы его можно было вызывать с помощью оператора диапазона, например: 

size = fields.IntegerRangeField(range(1, 50))

Но для этого потребуется гораздо больше кода, поскольку вы можете указать параметр 'skip' - range (1, 50, 2) - Интересная идея, хотя ...

69 голосов
/ 19 августа 2012 
from django.db import models
from django.core.validators import MinValueValidator, MaxValueValidator

size = models.IntegerField(validators=[MinValueValidator(0),
                                       MaxValueValidator(5)])
50 голосов
/ 29 октября 2010

У меня была такая же проблема; вот мое решение: 

SCORE_CHOICES = zip( range(1,n), range(1,n) )
score = models.IntegerField(choices=SCORE_CHOICES, blank=True)
10 голосов
/ 11 мая 2009

Есть два способа сделать это. Одним из них является использование проверки формы, чтобы никогда не позволять пользователю вводить любое число свыше 50. Подтверждение формы документов .

Если в процессе нет пользователя или вы не используете форму для ввода данных, вам придется переопределить метод модели save, чтобы вызвать исключение или ограничить передачу данных в поле. .

4 голосов
/ 18 сентября 2014

Это лучшее решение, если вам нужна дополнительная гибкость и вы не хотите менять поле модели. Просто добавьте этот пользовательский валидатор: 

#Imports
from django.core.exceptions import ValidationError      

class validate_range_or_null(object):
    compare = lambda self, a, b, c: a > c or a < b
    clean = lambda self, x: x
    message = ('Ensure this value is between %(limit_min)s and %(limit_max)s (it is %(show_value)s).')
    code = 'limit_value'

    def __init__(self, limit_min, limit_max):
        self.limit_min = limit_min
        self.limit_max = limit_max

    def __call__(self, value):
        cleaned = self.clean(value)
        params = {'limit_min': self.limit_min, 'limit_max': self.limit_max, 'show_value': cleaned}
        if value:  # make it optional, remove it to make required, or make required on the model
            if self.compare(cleaned, self.limit_min, self.limit_max):
                raise ValidationError(self.message, code=self.code, params=params)

И его можно использовать как таковое: 

class YourModel(models.Model):

    ....
    no_dependents = models.PositiveSmallIntegerField("How many dependants?", blank=True, null=True, default=0, validators=[validate_range_or_null(1,100)])

Два параметра - это max и min, и это допускает нулевые значения. Вы можете настроить валидатор, если хотите, избавившись от отмеченного оператора if или изменив свое поле на пустое = False, null = False в модели. Это, конечно, потребует миграции.

Примечание: мне пришлось добавить валидатор, потому что Django не проверяет диапазон на PositiveSmallIntegerField, вместо этого он создает smallint (в postgres) для этого поля, и вы получаете ошибку БД, если указанное число выходит за пределы диапазона.

Надеюсь, это поможет :) Подробнее о Валидаторы в Django .

PS. Я основал свой ответ на BaseValidator в django.core.validators, но все отличается, кроме кода.

...