From 6895b389e5a8702f165673c7c47e294db6cd36e6 Mon Sep 17 00:00:00 2001
From: Andrey Tikhonov <17@itishka.org>
Date: Tue, 6 May 2025 13:25:55 +0200
Subject: [PATCH] more prefetch for IPAddressViewSet

---
 netbox/ipam/api/views.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/netbox/ipam/api/views.py b/netbox/ipam/api/views.py
index 11cf70f84..dadbc2e2e 100644
--- a/netbox/ipam/api/views.py
+++ b/netbox/ipam/api/views.py
@@ -23,6 +23,7 @@ from netbox.api.viewsets.mixins import ObjectValidationMixin
 from netbox.config import get_config
 from netbox.constants import ADVISORY_LOCK_KEYS
 from utilities.api import get_serializer_for_model
+from virtualization.models import VMInterface
 from . import serializers
 
 
@@ -106,7 +107,10 @@ class IPAddressViewSet(NetBoxModelViewSet):
         GenericPrefetch(
             "assigned_object",
             [
-                Interface.objects.select_related("device"),
+                # serializers are taken according to IPADDRESS_ASSIGNMENT_MODELS
+                FHRPGroup.objects.all(),
+                Interface.objects.select_related("cable", "device"),
+                VMInterface.objects.select_related("virtual_machine"),
             ],
         ),
     )