diff options
| author | Tom Christie | 2014-09-23 14:30:17 +0100 | 
|---|---|---|
| committer | Tom Christie | 2014-09-23 14:30:17 +0100 | 
| commit | 0404f09a7e69f533038d47ca25caad90c0c2659f (patch) | |
| tree | eae2ceb609d868d02ba84c09d457b3ae6691f33b /rest_framework | |
| parent | f22d0afc3dfc7478e084d1d6ed6b53f71641dec6 (diff) | |
| download | django-rest-framework-0404f09a7e69f533038d47ca25caad90c0c2659f.tar.bz2 | |
NullBooleanField
Diffstat (limited to 'rest_framework')
| -rw-r--r-- | rest_framework/fields.py | 37 | ||||
| -rw-r--r-- | rest_framework/serializers.py | 2 | ||||
| -rw-r--r-- | rest_framework/utils/field_mapping.py | 2 | 
3 files changed, 38 insertions, 3 deletions
| diff --git a/rest_framework/fields.py b/rest_framework/fields.py index f5bae734..f859658a 100644 --- a/rest_framework/fields.py +++ b/rest_framework/fields.py @@ -289,6 +289,10 @@ class BooleanField(Field):      TRUE_VALUES = set(('t', 'T', 'true', 'True', 'TRUE', '1', 1, True))      FALSE_VALUES = set(('f', 'F', 'false', 'False', 'FALSE', '0', 0, 0.0, False)) +    def __init__(self, **kwargs): +        assert 'allow_null' not in kwargs, '`allow_null` is not a valid option. Use `NullBooleanField` instead.' +        super(BooleanField, self).__init__(**kwargs) +      def to_internal_value(self, data):          if data in self.TRUE_VALUES:              return True @@ -297,7 +301,38 @@ class BooleanField(Field):          self.fail('invalid', input=data)      def to_representation(self, value): -        if value is None: +        if value in self.TRUE_VALUES: +            return True +        elif value in self.FALSE_VALUES: +            return False +        return bool(value) + + +class NullBooleanField(Field): +    default_error_messages = { +        'invalid': _('`{input}` is not a valid boolean.') +    } +    default_empty_html = None +    TRUE_VALUES = set(('t', 'T', 'true', 'True', 'TRUE', '1', 1, True)) +    FALSE_VALUES = set(('f', 'F', 'false', 'False', 'FALSE', '0', 0, 0.0, False)) +    NULL_VALUES = set(('n', 'N', 'null', 'Null', 'NULL', '', None)) + +    def __init__(self, **kwargs): +        assert 'allow_null' not in kwargs, '`allow_null` is not a valid option.' +        kwargs['allow_null'] = True +        super(NullBooleanField, self).__init__(**kwargs) + +    def to_internal_value(self, data): +        if data in self.TRUE_VALUES: +            return True +        elif data in self.FALSE_VALUES: +            return False +        elif data in self.NULL_VALUES: +            return None +        self.fail('invalid', input=data) + +    def to_representation(self, value): +        if value in self.NULL_VALUES:              return None          if value in self.TRUE_VALUES:              return True diff --git a/rest_framework/serializers.py b/rest_framework/serializers.py index 949f5915..d8d72a4c 100644 --- a/rest_framework/serializers.py +++ b/rest_framework/serializers.py @@ -333,7 +333,7 @@ class ModelSerializer(Serializer):          models.FloatField: FloatField,          models.ImageField: ImageField,          models.IntegerField: IntegerField, -        models.NullBooleanField: BooleanField, +        models.NullBooleanField: NullBooleanField,          models.PositiveIntegerField: IntegerField,          models.PositiveSmallIntegerField: IntegerField,          models.SlugField: SlugField, diff --git a/rest_framework/utils/field_mapping.py b/rest_framework/utils/field_mapping.py index 1c718ccb..c208afdc 100644 --- a/rest_framework/utils/field_mapping.py +++ b/rest_framework/utils/field_mapping.py @@ -74,7 +74,7 @@ def get_field_kwargs(field_name, model_field):          kwargs['choices'] = model_field.flatchoices          return kwargs -    if model_field.null: +    if model_field.null and not isinstance(model_field, models.NullBooleanField):          kwargs['allow_null'] = True      if model_field.blank: | 
