Briq
/
drf-yasg
6
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Fix RelatedFieldInspector nesting issues

openapi3
Cristi Vîjdea 2018-09-10 05:04:03 +03:00
parent f50ad29a2b
commit 6820d6bc70
1 changed files with 13 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
src/drf_yasg/inspectors/field.py
View File

@ -232,9 +232,20 @@ class RelatedFieldInspector(FieldInspector):
# if the RelatedField has no queryset (e.g. read only), try to find the target model # if the RelatedField has no queryset (e.g. read only), try to find the target model
# from the view queryset or ModelSerializer model, if present # from the view queryset or ModelSerializer model, if present
view_queryset = getattr(self.view, 'queryset', None) view_queryset = getattr(self.view, 'queryset', None)
serializer_meta = getattr(get_parent_serializer(field), 'Meta', None) parent_serializer = get_parent_serializer(field)
this_model = getattr(view_queryset, 'model', None) or getattr(serializer_meta, 'model', None) if view_queryset is not None:
# make sure the view is actually using *this* serializer
try:
if type(parent_serializer) != self.view.get_serializer_class():
view_queryset = None
except Exception as e:
view_queryset = None
serializer_meta = getattr(parent_serializer, 'Meta', None)
this_model = getattr(serializer_meta, 'model', None) or getattr(view_queryset, 'model', None)
source = getattr(field, 'source', '') or field.field_name source = getattr(field, 'source', '') or field.field_name
if not source and isinstance(field.parent, serializers.ManyRelatedField):
source = field.parent.field_name
model = get_related_model(this_model, source) model = get_related_model(this_model, source)
model_field = get_model_field(model, target_field) model_field = get_model_field(model, target_field)