From eed4117c30a5f2b714216e2280b2827f3e995f20 Mon Sep 17 00:00:00 2001 From: Renato Almeida de Oliveira Zaroubin Date: Thu, 27 Mar 2025 23:56:12 +0000 Subject: [PATCH] Add VLANGroup to ClusterGroupView get_related_models --- netbox/virtualization/views.py | 15 +++++++++++++-- 1 file changed, 13 insertions(+), 2 deletions(-) diff --git a/netbox/virtualization/views.py b/netbox/virtualization/views.py index 343d346e4..6cd36cd86 100644 --- a/netbox/virtualization/views.py +++ b/netbox/virtualization/views.py @@ -1,4 +1,5 @@ from django.contrib import messages +from django.contrib.contenttypes.models import ContentType from django.db import transaction from django.db.models import Prefetch, Sum from django.shortcuts import get_object_or_404, redirect, render @@ -10,7 +11,7 @@ from dcim.forms import DeviceFilterForm from dcim.models import Device from dcim.tables import DeviceTable from extras.views import ObjectConfigContextView, ObjectRenderConfigView -from ipam.models import IPAddress +from ipam.models import IPAddress, VLANGroup from ipam.tables import InterfaceVLANTable, VLANTranslationRuleTable from netbox.constants import DEFAULT_ACTION_PERMISSIONS from netbox.views import generic @@ -103,7 +104,17 @@ class ClusterGroupView(GetRelatedModelsMixin, generic.ObjectView): def get_extra_context(self, request, instance): return { - 'related_models': self.get_related_models(request, instance), + 'related_models': self.get_related_models( + request, + instance, + extra=( + ( + VLANGroup.objects.restrict(request.user, 'view').filter( + scope_type=ContentType.objects.get_for_model(ClusterGroup), + scope_id=instance.pk + ), 'cluster_group'), + ), + ), }