mirror of
https://github.com/netbox-community/netbox.git
synced 2025-12-18 18:26:27 +00:00
a20715f229
* Fixes #19321: Reduce redundant database queries during bulk creation of devices * Add test for test_get_prefetchable_fields
35 lines
1.0 KiB
Python
35 lines
1.0 KiB
Python
from django.contrib.contenttypes.fields import GenericRelation
|
|
from django.db.models import ManyToManyField
|
|
from django.db.models.fields.related import ForeignObjectRel
|
|
from taggit.managers import TaggableManager
|
|
|
|
__all__ = (
|
|
'get_prefetchable_fields',
|
|
)
|
|
|
|
|
|
def get_prefetchable_fields(model):
|
|
"""
|
|
Return a list containing the names of all fields on the given model which support prefetching.
|
|
"""
|
|
field_names = []
|
|
|
|
for field in model._meta.get_fields():
|
|
# Forward relations (e.g. ManyToManyFields)
|
|
if isinstance(field, ManyToManyField):
|
|
field_names.append(field.name)
|
|
|
|
# Reverse relations (e.g. reverse ForeignKeys, reverse M2M)
|
|
elif isinstance(field, ForeignObjectRel):
|
|
field_names.append(field.get_accessor_name())
|
|
|
|
# Generic relations
|
|
elif isinstance(field, GenericRelation):
|
|
field_names.append(field.name)
|
|
|
|
# Tags
|
|
elif isinstance(field, TaggableManager):
|
|
field_names.append(field.name)
|
|
|
|
return field_names
|