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

Moved custom field serializers to their own module to avoid circular dependency

This commit is contained in:
Jeremy Stretch
2017-03-08 16:18:41 -05:00
parent f43fbffdf7
commit a6ceaf8d96
6 changed files with 53 additions and 50 deletions

View File

@ -1,49 +1,7 @@
from django.contrib.contenttypes.models import ContentType
from rest_framework import serializers
# from dcim.api.serializers import NestedSiteSerializer
from extras.models import CustomField, CustomFieldChoice, Graph, TopologyMap
#
# Custom fields
#
class CustomFieldSerializer(serializers.BaseSerializer):
"""
Extends ModelSerializer to render any CustomFields and their values associated with an object.
"""
def to_representation(self, manager):
# Initialize custom fields dictionary
data = {f.name: None for f in self.parent._custom_fields}
# Assign CustomFieldValues from database
for cfv in manager.all():
data[cfv.field.name] = cfv.value
return data
class CustomFieldModelSerializer(serializers.ModelSerializer):
custom_fields = CustomFieldSerializer(source='custom_field_values')
def __init__(self, *args, **kwargs):
super(CustomFieldModelSerializer, self).__init__(*args, **kwargs)
# Cache the list of custom fields for this model
content_type = ContentType.objects.get_for_model(self.Meta.model)
self._custom_fields = CustomField.objects.filter(obj_type=content_type)
class CustomFieldChoiceSerializer(serializers.ModelSerializer):
class Meta:
model = CustomFieldChoice
fields = ['id', 'value']
from dcim.api.serializers import NestedSiteSerializer
from extras.models import Graph, TopologyMap
#
@ -69,8 +27,8 @@ class GraphSerializer(serializers.ModelSerializer):
# Topology maps
#
class TopologyMapSerializer(CustomFieldModelSerializer):
# site = NestedSiteSerializer()
class TopologyMapSerializer(serializers.ModelSerializer):
site = NestedSiteSerializer()
class Meta:
model = TopologyMap