7699 review changes

netbox-community · Oct 25, 2024 · 62358f6 · 62358f6
1 parent 277b175
commit 62358f6
Show file tree

Hide file tree

Showing 3 changed files with 14 additions and 2 deletions.
diff --git a/docs/models/virtualization/cluster.md b/docs/models/virtualization/cluster.md
@@ -25,4 +25,4 @@ The cluster's operational status.
 
 ### Scope
 
-The [region](../dcim/region.md), [site](../dcim/site.md) or [location](../dcim/location.md) with which this cluster is associated.
+The [region](../dcim/region.md), [site](../dcim/site.md), [site group](../dcim/sitegroup.md) or [location](../dcim/location.md) with which this cluster is associated.
diff --git a/netbox/dcim/graphql/types.py b/netbox/dcim/graphql/types.py
@@ -461,6 +461,10 @@ class LocationType(VLANGroupsMixin, ImageAttachmentsMixin, ContactsMixin, Organi
     devices: List[Annotated["DeviceType", strawberry.lazy('dcim.graphql.types')]]
     children: List[Annotated["LocationType", strawberry.lazy('dcim.graphql.types')]]
 
+    @strawberry_django.field
+    def clusters(self) -> List[Annotated["ClusterType", strawberry.lazy('virtualization.graphql.types')]]:
+        return self._clusters.all()
+
 
 @strawberry_django.type(
     models.Manufacturer,
@@ -704,6 +708,10 @@ class RegionType(VLANGroupsMixin, ContactsMixin, OrganizationalObjectType):
     def parent(self) -> Annotated["RegionType", strawberry.lazy('dcim.graphql.types')] | None:
         return self.parent
 
+    @strawberry_django.field
+    def clusters(self) -> List[Annotated["ClusterType", strawberry.lazy('virtualization.graphql.types')]]:
+        return self._clusters.all()
+
 
 @strawberry_django.type(
     models.Site,
@@ -748,6 +756,10 @@ class SiteGroupType(VLANGroupsMixin, ContactsMixin, OrganizationalObjectType):
     def parent(self) -> Annotated["SiteGroupType", strawberry.lazy('dcim.graphql.types')] | None:
         return self.parent
 
+    @strawberry_django.field
+    def clusters(self) -> List[Annotated["ClusterType", strawberry.lazy('virtualization.graphql.types')]]:
+        return self._clusters.all()
+
 
 @strawberry_django.type(
     models.VirtualChassis,

diff --git a/netbox/virtualization/constants.py b/netbox/virtualization/constants.py
@@ -1,4 +1,4 @@
-# models values for ContentTypes which may be CircuitTermination scope types
+# models values for ContentTypes which may be Cluster scope types
 CLUSTER_SCOPE_TYPES = (
     'region', 'sitegroup', 'site', 'location',
 )