Sfoglia il codice sorgente

Add support for custom field data

jeremystretch 4 anni fa
parent
commit
930ca8d9a1
1 ha cambiato i file con 8 aggiunte e 1 eliminazioni
  1. 8 1
      netbox/netbox/graphql/types.py

+ 8 - 1
netbox/netbox/graphql/types.py

@@ -1,4 +1,5 @@
 import graphene
 import graphene
+from graphene.types.generic import GenericScalar
 from graphene_django import DjangoObjectType
 from graphene_django import DjangoObjectType
 
 
 __all__ = (
 __all__ = (
@@ -22,11 +23,17 @@ class BaseObjectType(DjangoObjectType):
 
 
 
 
 class ObjectType(BaseObjectType):
 class ObjectType(BaseObjectType):
-    # TODO: Custom fields support
+    """
+    Extends BaseObjectType with support for custom field data.
+    """
+    custom_fields = GenericScalar()
 
 
     class Meta:
     class Meta:
         abstract = True
         abstract = True
 
 
+    def resolve_custom_fields(self, info):
+        return self.custom_field_data
+
 
 
 class TaggedObjectType(ObjectType):
 class TaggedObjectType(ObjectType):
     """
     """