NeoAnd/netbox
1
0
Fork 0
mirror of https://github.com/netbox-community/netbox.git synced 2025-07-27 10:58:37 -06:00
Code Issues Actions 4 Packages Projects Releases Wiki Activity

Generalize tags use case

Browse Source
This commit is contained in:
Renato Almeida de Oliveira Zaroubin 2025-03-07 21:38:25 +00:00
parent 603253ddd7
commit 68d9c3958d
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
netbox/utilities/api.py
View File

@ -1,6 +1,8 @@
from django.contrib.contenttypes.fields import GenericForeignKey
from django.core.exceptions import ( from django.core.exceptions import (
FieldDoesNotExist, FieldError, MultipleObjectsReturned, ObjectDoesNotExist, ValidationError, FieldDoesNotExist, FieldError, MultipleObjectsReturned, ObjectDoesNotExist, ValidationError,
) )
from django.db.models.fields.related import ManyToOneRel, RelatedField
from django.db.models import Prefetch from django.db.models import Prefetch
from django.urls import reverse from django.urls import reverse
from django.utils.module_loading import import_string from django.utils.module_loading import import_string
@ -108,10 +110,6 @@ def get_prefetches_for_serializer(serializer_class, fields_to_include=None, sour
except FieldDoesNotExist: except FieldDoesNotExist:
continue continue
# Prefetch for tags
if field.name == 'tags':
prefetch_fields.append('tags')
# If this field is represented by a nested serializer, recurse to resolve prefetches # If this field is represented by a nested serializer, recurse to resolve prefetches
# for the related object. # for the related object.
if serializer_field and source_field is None: if serializer_field and source_field is None:
@ -123,6 +121,8 @@ def get_prefetches_for_serializer(serializer_class, fields_to_include=None, sour
prefetch_fields.append(subfield) prefetch_fields.append(subfield)
else: else:
prefetch_fields.append(f'{field_name}__{subfield}') prefetch_fields.append(f'{field_name}__{subfield}')
elif isinstance(field, (RelatedField, ManyToOneRel, GenericForeignKey)):
prefetch_fields.append(field.name)
return prefetch_fields return prefetch_fields