Handle KeyError in ApiEndpoint

Author: Ivaylo-Bachvarov-chichi

rest_framework_docs/api_endpoint.py

@@ -19,7 +19,7 @@ def __init__(self, pattern, parent_pattern=None):
         self.errors = None
         self.serializer_class = self.__get_serializer_class__()
         if self.serializer_class:
-            self.serializer = self.serializer_class()
+            self.serializer = self.__get_serializer__()
             self.fields = self.__get_serializer_fields__(self.serializer)
             self.fields_json = self.__get_serializer_fields_json__()
 
@@ -40,6 +40,12 @@ def __get_permissions_class__(self):
         for perm_class in self.pattern.callback.cls.permission_classes:
             return perm_class.__name__
 
+    def __get_serializer__(self):
+        try:
+            return self.serializer_class()
+        except KeyError as e:
+            self.errors = e
+
     def __get_serializer_class__(self):
         if hasattr(self.callback.cls, 'serializer_class'):
             return self.callback.cls.serializer_class
@@ -51,19 +57,14 @@ def __get_serializer_fields__(self, serializer):
         fields = []
 
         if hasattr(serializer, 'get_fields'):
-            try:
-                for key, field in serializer.get_fields().items():
-                    sub_fields = self.__get_serializer_fields__(field) if isinstance(field, BaseSerializer) else None
-                    fields.append({
-                        "name": key,
-                        "type": str(field.__class__.__name__),
-                        "sub_fields": sub_fields,
-                        "required": field.required
-                    })
-            except KeyError as e:
-                self.errors = e
-                fields = []
-
+            for key, field in serializer.get_fields().items():
+                sub_fields = self.__get_serializer_fields__(field) if isinstance(field, BaseSerializer) else None
+                fields.append({
+                    "name": key,
+                    "type": str(field.__class__.__name__),
+                    "sub_fields": sub_fields,
+                    "required": field.required
+                })
             # FIXME:
             # Show more attibutes of `field`?