Add vlan_groups to Region, SiteGroup, Site, Location, Rack, ClusterGroup, Cluster

This commit is contained in:
jeremystretch
2021-08-03 11:49:22 -04:00
parent c42dab14c5
commit 8c7ef2a95e
6 changed files with 59 additions and 9 deletions

View File

@@ -1,4 +1,4 @@
from ipam.graphql.mixins import IPAddressesMixin
from ipam.graphql.mixins import IPAddressesMixin, VLANGroupsMixin
from virtualization import filtersets, models
from netbox.graphql.types import ObjectType, TaggedObjectType
@@ -11,7 +11,7 @@ __all__ = (
)
class ClusterType(TaggedObjectType):
class ClusterType(VLANGroupsMixin, TaggedObjectType):
class Meta:
model = models.Cluster
@@ -19,7 +19,7 @@ class ClusterType(TaggedObjectType):
filterset_class = filtersets.ClusterFilterSet
class ClusterGroupType(ObjectType):
class ClusterGroupType(VLANGroupsMixin, ObjectType):
class Meta:
model = models.ClusterGroup