mirror of
https://github.com/netbox-community/netbox.git
synced 2025-07-27 19:08:38 -06:00
Removed prefetch for model fields, since its a Serializer and add for tags
This commit is contained in:
parent
53e8da674f
commit
603253ddd7
@ -1,8 +1,6 @@
|
|||||||
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
|
||||||
@ -107,12 +105,13 @@ def get_prefetches_for_serializer(serializer_class, fields_to_include=None, sour
|
|||||||
# If the serializer field does not map to a discrete model field, skip it.
|
# If the serializer field does not map to a discrete model field, skip it.
|
||||||
try:
|
try:
|
||||||
field = model._meta.get_field(model_field_name)
|
field = model._meta.get_field(model_field_name)
|
||||||
if (isinstance(field, (RelatedField, ManyToOneRel, GenericForeignKey)) and
|
|
||||||
not issubclass(type(serializer_field), Serializer)):
|
|
||||||
prefetch_fields.append(field.name)
|
|
||||||
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:
|
||||||
@ -124,7 +123,6 @@ 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}')
|
||||||
|
|
||||||
return prefetch_fields
|
return prefetch_fields
|
||||||
|
|
||||||
|
|
||||||
|
Loading…
Reference in New Issue
Block a user