From 9239c8f40a7b97e2c48869ebb6deacdad52d7621 Mon Sep 17 00:00:00 2001 From: Jonas_Jones Date: Tue, 4 Mar 2025 18:51:02 +0100 Subject: [PATCH] fixed class path slug for relations --- puml_generator.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/puml_generator.py b/puml_generator.py index 9b1ef6a..93be27c 100644 --- a/puml_generator.py +++ b/puml_generator.py @@ -243,8 +243,9 @@ def class_to_puml(filename:str) -> JavaClass: class_methods[method_name] = get_method_visibility(javaline) next_line_setter = False elif javaline.startswith("import"): - importline_package_slug = javaline.strip().split(" ")[1].replace(";", "") - imported_classes[importline_package_slug.split(".")[-1]] = importline_package_slug + if "*" not in javaline: + importline_package_slug = javaline.strip().split(" ")[1].replace(";", "") + imported_classes[importline_package_slug] = importline_package_slug if javaline.startswith("@"): if "@Getter" in javaline and "@Setter" in javaline: