Update get_model_field() to raise a clear error when relationship tra…

…versal isn't possible
wagtail · Feb 4, 2024 · ba498c3 · ba498c3
1 parent 07f60df
commit ba498c3
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/modelcluster/utils.py b/modelcluster/utils.py
@@ -66,9 +66,17 @@ def get_model_field(model, name):
                         subject_model=subject_model,
                     )
                 )
-            if hasattr(field, "related_model"):
+            if getattr(field, "related_model", None):
                 traversals.append(TraversedRelationship(subject_model, field))
                 subject_model = field.related_model
+            else:
+                raise FieldDoesNotExist(
+                    "Failed attempting to traverse from {from_field} to '{to_field}'."
+                    .format(
+                        from_field=subject_model._meta.app_label + '.' + field.name,
+                        to_field=field_name,
+                    )
+                )
         try:
             field = subject_model._meta.get_field(field_name)
         except FieldDoesNotExist: