Diagram generator can handle empty files

netsatan · Nov 23, 2022 · 7f70a0f · 7f70a0f
1 parent 5167940
commit 7f70a0f
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 1 deletion.
diff --git a/specifipy/file_scanners/__init__.py b/specifipy/file_scanners/__init__.py
diff --git a/specifipy/file_scanners/directory_scanner.py b/specifipy/file_scanners/directory_scanner.py
diff --git a/specifipy/parsers/diagram_generator.py b/specifipy/parsers/diagram_generator.py
@@ -63,5 +63,10 @@ def generate_diagram(
         joined_links_to_generate = "\t" + "\n\t".join(links_to_generate)
 
         diagram_definition = f'with Diagram("{source_file_name}", show=False, filename="{base_path if base_path else ""}{source_file_name}"):\n'
-        generated_file = f"{self.header}\n{diagram_definition}\n{joined_elements_to_generate}\n{joined_links_to_generate}"
+        if len(links_to_generate) == 0:
+            generated_file = f"{self.header}\n{diagram_definition}\n\tpass"
+        elif len(elements_to_generate) == 0:
+            generated_file = f"{self.header}\n{diagram_definition}\n\tpass"
+        else:
+            generated_file = f"{self.header}\n{diagram_definition}\n{joined_elements_to_generate}\n{joined_links_to_generate}"
         exec(f"{generated_file}")