Implemented static writable ModelSerializers for all models

netbox/extras/api/serializers.py

@@ -1,35 +1,42 @@
 from rest_framework import serializers
 
-from extras.models import CF_TYPE_SELECT, CustomFieldChoice, Graph
+from extras.models import CF_TYPE_SELECT, CustomFieldChoice, CustomFieldValue, Graph
 
 
-class CustomFieldSerializer(serializers.Serializer):
-    """
-    Extends a ModelSerializer to render any CustomFields and their values associated with an object.
-    """
-    custom_fields = serializers.SerializerMethodField()
+# class CustomFieldSerializer(serializers.ModelSerializer):
+#     """
+#     Extends ModelSerializer to render any CustomFields and their values associated with an object.
+#     """
+#     custom_fields = serializers.SerializerMethodField()
+#
+#     def get_custom_fields(self, obj):
+#
+#         # Gather all CustomFields applicable to this object
+#         fields = {cf.name: None for cf in self.context['custom_fields']}
+#         custom_field_choices = self.context['custom_field_choices']
+#
+#         # Attach any defined CustomFieldValues to their respective CustomFields
+#         for cfv in obj.custom_field_values.all():
+#
+#             # Attempt to suppress database lookups for CustomFieldChoices by using the cached choice set from the view
+#             # context.
+#             if cfv.field.type == CF_TYPE_SELECT:
+#                 cfc = {
+#                     'id': int(cfv.serialized_value),
+#                     'value': custom_field_choices[int(cfv.serialized_value)]
+#                 }
+#                 fields[cfv.field.name] = CustomFieldChoiceSerializer(instance=cfc).data
+#             else:
+#                 fields[cfv.field.name] = cfv.value
+#
+#         return fields
 
-    def get_custom_fields(self, obj):
 
-        # Gather all CustomFields applicable to this object
-        fields = {cf.name: None for cf in self.context['custom_fields']}
-        custom_field_choices = self.context['custom_field_choices']
+class CustomFieldValueSerializer(serializers.ModelSerializer):
 
-        # Attach any defined CustomFieldValues to their respective CustomFields
-        for cfv in obj.custom_field_values.all():
-
-            # Attempt to suppress database lookups for CustomFieldChoices by using the cached choice set from the view
-            # context.
-            if cfv.field.type == CF_TYPE_SELECT:
-                cfc = {
-                    'id': int(cfv.serialized_value),
-                    'value': custom_field_choices[int(cfv.serialized_value)]
-                }
-                fields[cfv.field.name] = CustomFieldChoiceSerializer(instance=cfc).data
-            else:
-                fields[cfv.field.name] = cfv.value
-
-        return fields
+    class Meta:
+        model = CustomFieldValue
+        fields = ['field', 'serialized_value']
 
 
 class CustomFieldChoiceSerializer(serializers.ModelSerializer):