Rename base Graphene types to match base models

This commit is contained in:
jeremystretch
2021-08-03 14:14:36 -04:00
parent 24e0f7adfa
commit ac7685dfba
6 changed files with 85 additions and 52 deletions

View File

@@ -1,6 +1,7 @@
from dcim.graphql.types import ComponentObjectType
from ipam.graphql.mixins import IPAddressesMixin, VLANGroupsMixin
from virtualization import filtersets, models
from netbox.graphql.types import ObjectType, PrimaryObjectType, TaggedObjectType
from netbox.graphql.types import OrganizationalObjectType, PrimaryObjectType
__all__ = (
'ClusterType',
@@ -19,7 +20,7 @@ class ClusterType(VLANGroupsMixin, PrimaryObjectType):
filterset_class = filtersets.ClusterFilterSet
class ClusterGroupType(VLANGroupsMixin, ObjectType):
class ClusterGroupType(VLANGroupsMixin, OrganizationalObjectType):
class Meta:
model = models.ClusterGroup
@@ -27,7 +28,7 @@ class ClusterGroupType(VLANGroupsMixin, ObjectType):
filterset_class = filtersets.ClusterGroupFilterSet
class ClusterTypeType(ObjectType):
class ClusterTypeType(OrganizationalObjectType):
class Meta:
model = models.ClusterType
@@ -43,7 +44,7 @@ class VirtualMachineType(PrimaryObjectType):
filterset_class = filtersets.VirtualMachineFilterSet
class VMInterfaceType(IPAddressesMixin, TaggedObjectType):
class VMInterfaceType(IPAddressesMixin, ComponentObjectType):
class Meta:
model = models.VMInterface