NeoAnd/netbox
1
0
Fork 0
mirror of https://github.com/netbox-community/netbox.git synced 2025-07-28 19:36:26 -06:00
Code Issues Actions 8 Packages Projects Releases Wiki Activity

Add get_annotations_for_subfields and remove recursion

Browse Source
This commit is contained in:
Renato Almeida de Oliveira Zaroubin 2025-03-15 02:24:29 +00:00
parent 68d9c3958d
commit 603c269a87
1 changed files with 18 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

32
netbox/utilities/api.py
View File

@ -77,7 +77,7 @@ def get_view_name(view):
return drf_get_view_name(view) return drf_get_view_name(view)
def get_prefetches_for_serializer(serializer_class, fields_to_include=None, source_field=None): def get_prefetches_for_serializer(serializer_class, fields_to_include=None):
""" """
Compile and return a list of fields which should be prefetched on the queryset for a serializer. Compile and return a list of fields which should be prefetched on the queryset for a serializer.
""" """
@ -89,13 +89,6 @@ def get_prefetches_for_serializer(serializer_class, fields_to_include=None, sour
prefetch_fields = [] prefetch_fields = []
# If this serializer is nested, get annotations and prefetches for the nested serializer
if source_field is not None:
nested_annotations = get_annotations_for_serializer(serializer_class, fields_to_include=fields_to_include)
if nested_annotations:
related_prefetch = Prefetch(source_field, queryset=model.objects.all().annotate(**nested_annotations))
prefetch_fields.append(related_prefetch)
for field_name in fields_to_include: for field_name in fields_to_include:
serializer_field = serializer_class._declared_fields.get(field_name) serializer_field = serializer_class._declared_fields.get(field_name)
@ -112,20 +105,31 @@ def get_prefetches_for_serializer(serializer_class, fields_to_include=None, sour
# 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:
if issubclass(type(serializer_field), Serializer): if issubclass(type(serializer_field), Serializer):
# Determine which fields to prefetch for the nested object # Determine which fields to prefetch for the nested object
subfields = serializer_field.Meta.brief_fields if serializer_field.nested else None subfields = serializer_field.Meta.brief_fields if serializer_field.nested else None
for subfield in get_prefetches_for_serializer(type(serializer_field), subfields, field_name): annotated_prefetch = get_annotations_for_subfields(
if isinstance(subfield, Prefetch): serializer_field, field_name, fields_to_include=subfields)
prefetch_fields.append(subfield) if annotated_prefetch:
else: prefetch_fields.append(annotated_prefetch)
prefetch_fields.append(f'{field_name}__{subfield}') else:
prefetch_fields.append(field_name)
elif isinstance(field, (RelatedField, ManyToOneRel, GenericForeignKey)): elif isinstance(field, (RelatedField, ManyToOneRel, GenericForeignKey)):
prefetch_fields.append(field.name) prefetch_fields.append(field.name)
return prefetch_fields return prefetch_fields
def get_annotations_for_subfields(serializer_class, source_field, fields_to_include=None):
nested_annotations = get_annotations_for_serializer(serializer_class, fields_to_include=fields_to_include)
model = serializer_class.Meta.model
related_prefetch = None
if nested_annotations:
related_prefetch = Prefetch(source_field, queryset=model.objects.all().annotate(**nested_annotations))
return related_prefetch
def get_annotations_for_serializer(serializer_class, fields_to_include=None): def get_annotations_for_serializer(serializer_class, fields_to_include=None):
""" """
Return a mapping of field names to annotations to be applied to the queryset for a serializer. Return a mapping of field names to annotations to be applied to the queryset for a serializer.