Add 'format' argument to UUIDField

To allow serializations to control the representation format of a
UUID value
This commit is contained in:
Thomas Stephenson 2015-03-26 04:56:55 +11:00
parent 5cf8230ba1
commit 236b731072
3 changed files with 42 additions and 3 deletions

View File

@ -183,6 +183,15 @@ A field that ensures the input is a valid UUID string. The `to_internal_value` m
"de305d54-75b4-431b-adb2-eb6b9e546013" "de305d54-75b4-431b-adb2-eb6b9e546013"
**Signature:** `UUIDField(format='hex_verbose')`
- `format`: Determines the representation format of the uuid value
- `'hex_verbose'` - The cannoncical hex representation, including hyphens
- `'hex'` - The compact hex representation of the UUID, not including hyphens
- `'int'` - A 128 bit integer representation of the UUID.
- `'urn'` - RFC 4122 URN representation of the UUID
Changing the `format` parameters only affects representation values. All formats are accepted by `to_internal_value`
--- ---
# Numeric fields # Numeric fields

View File

@ -639,20 +639,37 @@ class URLField(CharField):
class UUIDField(Field): class UUIDField(Field):
valid_formats = ('hex_verbose', 'hex', 'int', 'urn')
default_error_messages = { default_error_messages = {
'invalid': _('"{value}" is not a valid UUID.'), 'invalid': _('"{value}" is not a valid UUID.'),
} }
def __init__(self, **kwargs):
self.uuid_format = kwargs.pop('format', 'hex_verbose')
if self.uuid_format not in self.valid_formats:
raise ValueError(
'Invalid format for uuid representation. '
'Must be one of "{0}"'.format('", "'.join(self.valid_formats))
)
super(UUIDField, self).__init__(**kwargs)
def to_internal_value(self, data): def to_internal_value(self, data):
if not isinstance(data, uuid.UUID): if not isinstance(data, uuid.UUID):
try: try:
return uuid.UUID(data) if isinstance(data, six.integer_types):
return uuid.UUID(int=data)
else:
return uuid.UUID(hex=data)
except (ValueError, TypeError): except (ValueError, TypeError):
self.fail('invalid', value=data) self.fail('invalid', value=data)
return data return data
def to_representation(self, value): def to_representation(self, value):
if self.uuid_format == 'hex_verbose':
return str(value) return str(value)
else:
return getattr(value, self.uuid_format)
# Number types... # Number types...

View File

@ -526,7 +526,9 @@ class TestUUIDField(FieldValues):
""" """
valid_inputs = { valid_inputs = {
'825d7aeb-05a9-45b5-a5b7-05df87923cda': uuid.UUID('825d7aeb-05a9-45b5-a5b7-05df87923cda'), '825d7aeb-05a9-45b5-a5b7-05df87923cda': uuid.UUID('825d7aeb-05a9-45b5-a5b7-05df87923cda'),
'825d7aeb05a945b5a5b705df87923cda': uuid.UUID('825d7aeb-05a9-45b5-a5b7-05df87923cda') '825d7aeb05a945b5a5b705df87923cda': uuid.UUID('825d7aeb-05a9-45b5-a5b7-05df87923cda'),
'urn:uuid:213b7d9b-244f-410d-828c-dabce7a2615d': uuid.UUID('213b7d9b-244f-410d-828c-dabce7a2615d'),
284758210125106368185219588917561929842: uuid.UUID('d63a6fb6-88d5-40c7-a91c-9edf73283072')
} }
invalid_inputs = { invalid_inputs = {
'825d7aeb-05a9-45b5-a5b7': ['"825d7aeb-05a9-45b5-a5b7" is not a valid UUID.'] '825d7aeb-05a9-45b5-a5b7': ['"825d7aeb-05a9-45b5-a5b7" is not a valid UUID.']
@ -536,6 +538,17 @@ class TestUUIDField(FieldValues):
} }
field = serializers.UUIDField() field = serializers.UUIDField()
def _test_format(self, uuid_format, formatted_uuid_0):
field = serializers.UUIDField(format=uuid_format)
assert field.to_representation(uuid.UUID(int=0)) == formatted_uuid_0
assert field.to_internal_value(formatted_uuid_0) == uuid.UUID(int=0)
def test_formats(self):
self._test_format('int', 0)
self._test_format('hex_verbose', '00000000-0000-0000-0000-000000000000')
self._test_format('urn', 'urn:uuid:00000000-0000-0000-0000-000000000000')
self._test_format('hex', '0' * 32)
# Number types... # Number types...