diff --git a/puml_generator.py b/puml_generator.py index da7fb49..1a23c74 100644 --- a/puml_generator.py +++ b/puml_generator.py @@ -204,7 +204,8 @@ def gen_puml_code_from_class(java_class:JavaClass, no_pkgs:bool=False) -> str: for related_class, relation in uml_relations.items(): if not related_class == "*": - puml_code += f"{class_package + "." + class_name} {relation}-- {related_class}\n" + cardinalities = ("1", "1") + puml_code += f"{class_package + "." + class_name} {"\"" + cardinalities[1] + "\""} {relation}-- {"\"" + cardinalities[0] + "\""} {related_class}\n" puml_code += "\n"