Export class diagramms faster

1 files changed, 7 insertions, 2 deletions

diff --git a/secateurs.py b/secateurs.py
index ffa29b3..249651c 100644
--- a/secateurs.py
+++ b/secateurs.py
@@ -7,6 +7,7 @@ f = open(sys.argv[1], 'r').readlines()
 p = plantuml.PlantUML(url="http://www.plantuml.com/plantuml/svg/")
 
 preamble = []
+exported = set()
 
 
 def add_preamble(offset):
@@ -18,6 +19,11 @@ def add_preamble(offset):
 
 
 def generate_svg(text):
+    name = text.split(' ')
+    name = name[name.index('namespace') + 1]
+    if name in exported:
+        return
+    exported.add(name)
     print(f"generating: {text}")
     try:
         res = p.processes(plantuml_text=('\n'.join(preamble) + text))
@@ -26,8 +32,7 @@ def generate_svg(text):
         print(p.get_url(plantuml_text=('\n'.join(preamble) + text)))
         return
 
-    name = text.split(' ')
-    name = 'out/' + name[name.index('namespace') + 1]
+    name = 'out/' + name
     open(name + '.svg', 'wb').write(res)
     os.system(f"inkscape {name + '.svg'} -o {name + '.pdf'}")
     time.sleep(1)