encode/django-rest-framework
1
1
Fork 0
mirror of https://github.com/encode/django-rest-framework.git synced 2025-07-27 08:29:59 +03:00
Code Issues Packages Projects Releases Wiki Activity

Make use of distutils.util.strtobool in BooleanField class

Browse Source
This commit is contained in:
Hugo Cartwright 2021-04-01 15:38:44 +02:00
parent 96885dd9a7
commit 979f4d58ea
1 changed files with 14 additions and 25 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

39
rest_framework/fields.py
View File

@ -1,6 +1,7 @@
import copy
import datetime
import decimal
import distutils.util
import functools
import inspect
import re
@ -702,43 +703,31 @@ class BooleanField(Field):
}
default_empty_html = False
initial = False
TRUE_VALUES = {
't', 'T',
'y', 'Y', 'yes', 'Yes', 'YES',
'true', 'True', 'TRUE',
'on', 'On', 'ON',
'1', 1,
True
}
FALSE_VALUES = {
'f', 'F',
'n', 'N', 'no', 'No', 'NO',
'false', 'False', 'FALSE',
'off', 'Off', 'OFF',
'0', 0, 0.0,
False
}
NULL_VALUES = {'null', 'Null', 'NULL', '', None}
def to_internal_value(self, data):
try:
if data in self.TRUE_VALUES:
return True
elif data in self.FALSE_VALUES:
return False
elif data in self.NULL_VALUES and self.allow_null:
if data in self.NULL_VALUES and self.allow_null:
return None
elif isinstance(data, str):
try:
return bool(distutils.util.strtobool(data))
except ValueError:
pass
else:
return data
except TypeError: # Input is an unhashable type
pass
self.fail('invalid', input=data)
def to_representation(self, value):
if value in self.TRUE_VALUES:
return True
elif value in self.FALSE_VALUES:
return False
if value in self.NULL_VALUES and self.allow_null:
return None
elif isinstance(value, str):
try:
return bool(distutils.util.strtobool(value))
except ValueError:
pass
return bool(value)