2024-03-21 14:34:37 -04:00
|
|
|
from django.db.models import ManyToOneRel
|
2018-02-02 13:32:16 -05:00
|
|
|
|
2024-03-21 16:12:44 -04:00
|
|
|
__all__ = (
|
|
|
|
'get_related_models',
|
|
|
|
)
|
|
|
|
|
2022-11-02 16:26:26 -04:00
|
|
|
|
2023-11-16 12:02:32 -05:00
|
|
|
def get_related_models(model, ordered=True):
|
|
|
|
"""
|
|
|
|
Return a list of all models which have a ForeignKey to the given model and the name of the field. For example,
|
|
|
|
`get_related_models(Tenant)` will return all models which have a ForeignKey relationship to Tenant.
|
|
|
|
"""
|
|
|
|
related_models = [
|
|
|
|
(field.related_model, field.remote_field.name)
|
|
|
|
for field in model._meta.related_objects
|
|
|
|
if type(field) is ManyToOneRel
|
|
|
|
]
|
|
|
|
|
|
|
|
if ordered:
|
2023-12-01 11:23:38 -05:00
|
|
|
return sorted(related_models, key=lambda x: x[0]._meta.verbose_name.lower())
|
2023-11-16 12:02:32 -05:00
|
|
|
|
|
|
|
return related_models
|