mirror of
https://github.com/netbox-community/netbox.git
synced 2024-05-10 07:54:54 +00:00
57 lines
1.7 KiB
Python
57 lines
1.7 KiB
Python
from django.contrib.contenttypes.models import ContentType
|
|
|
|
from rest_framework import serializers
|
|
|
|
from extras.models import CustomField, CustomFieldChoice, Graph
|
|
|
|
|
|
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']
|
|
|
|
|
|
class GraphSerializer(serializers.ModelSerializer):
|
|
embed_url = serializers.SerializerMethodField()
|
|
embed_link = serializers.SerializerMethodField()
|
|
|
|
class Meta:
|
|
model = Graph
|
|
fields = ['name', 'embed_url', 'embed_link']
|
|
|
|
def get_embed_url(self, obj):
|
|
return obj.embed_url(self.context['graphed_object'])
|
|
|
|
def get_embed_link(self, obj):
|
|
return obj.embed_link(self.context['graphed_object'])
|