factor out serializer name estimation for easier modification #976

tfranzel · Jul 9, 2023 · 1a9be59 · 1a9be59
1 parent 31665a3
commit 1a9be59
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/drf_spectacular/openapi.py b/drf_spectacular/openapi.py
@@ -1517,6 +1517,9 @@ def _get_response_headers_for_code(self, status_code, direction='response') -> d
 
         return result
 
+    def get_serializer_name(self, serializer, direction):
+        return serializer.__class__.__name__
+
     def _get_serializer_name(self, serializer, direction, bypass_extensions=False):
         serializer_extension = OpenApiSerializerExtension.get_match(serializer)
         if serializer_extension and not bypass_extensions:
@@ -1538,7 +1541,7 @@ def _get_serializer_name(self, serializer, direction, bypass_extensions=False):
         elif is_list_serializer(serializer):
             return self._get_serializer_name(serializer.child, direction)
         else:
-            name = serializer.__class__.__name__
+            name = self.get_serializer_name(serializer, direction)
 
         if name.endswith('Serializer'):
             name = name[:-10]