From c0f37f6ec1f7d97cbaa6ca71071eff026559bb2d Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Cristi=20V=C3=AEjdea?= Date: Wed, 25 Apr 2018 22:41:06 +0300 Subject: [PATCH] Avoid crash when hitting unknown model fields Fixes #102 --- src/drf_yasg/inspectors/field.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/src/drf_yasg/inspectors/field.py b/src/drf_yasg/inspectors/field.py index 4180c30..543bcf8 100644 --- a/src/drf_yasg/inspectors/field.py +++ b/src/drf_yasg/inspectors/field.py @@ -410,7 +410,9 @@ class ChoiceFieldInspector(FieldInspector): model = getattr(getattr(serializer, 'Meta'), 'model') model_field = get_model_field(model, field.source) if model_field: - enum_type = get_basic_type_info(model_field).get('type', enum_type) + model_type = get_basic_type_info(model_field) + if model_type: + enum_type = model_type.get('type', enum_type) if isinstance(field, serializers.MultipleChoiceField): return SwaggerType(