mirror/netbox-community-netbox
1
0
Fork 0
mirror of https://github.com/netbox-community/netbox.git synced 2024-05-10 07:54:54 +00:00
Code Activity

Revise custom field API tests to check for single/multiple objects with/without custom field values

Browse Source
This commit is contained in:
Jeremy Stretch
2020-03-03 17:07:43 -05:00
parent e4abfd192e
commit 4611536ca9
2 changed files with 253 additions and 211 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
netbox/extras/api/customfields.py
View File

@@ -1,6 +1,7 @@
from datetime import datetime
from django.contrib.contenttypes.models import ContentType
from django.core.exceptions import ObjectDoesNotExist
from django.db import transaction
from rest_framework import serializers
from rest_framework.exceptions import ValidationError
@@ -29,10 +30,17 @@ class CustomFieldDefaultValues:
value = {}
for field in fields:
if field.default:
if field.type == CustomFieldTypeChoices.TYPE_SELECT:
field_value = field.choices.get(value=field.default).pk
if field.type == CustomFieldTypeChoices.TYPE_INTEGER:
field_value = int(field.default)
elif field.type == CustomFieldTypeChoices.TYPE_BOOLEAN:
field_value = bool(field.default)
# TODO: Fix default value assignment for boolean custom fields
field_value = False if field.default.lower() == 'false' else bool(field.default)
elif field.type == CustomFieldTypeChoices.TYPE_SELECT:
try:
field_value = field.choices.get(value=field.default).pk
except ObjectDoesNotExist:
# Invalid default value
field_value = None
else:
field_value = field.default
value[field.name] = field_value