fix issues with the class name

osl-incubator · Jun 25, 2024 · 3163706 · 3163706
1 parent 243fb52
commit 3163706
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/src/umlizer/class_graph.py b/src/umlizer/class_graph.py
@@ -219,7 +219,10 @@ def _get_entity_class_uml(klass: ClassDef) -> str:
     class_name = klass.name
 
     if base_classes:
-        class_name += f' ({base_classes})'
+        if len(base_classes) < 20:  # noqa: PLR2004
+            class_name += f' ({base_classes})'
+        else:
+            class_name += ' (\\n' + base_classes.replace(', ', ',\\n  ') + ')'
 
     # Formatting fields and methods
     fields_struct = klass.fields