Save output as pngs

1 files changed, 13 insertions, 3 deletions

diff --git a/secateurs.py b/secateurs.py
index 7111c04..c6a4f04 100644
--- a/secateurs.py
+++ b/secateurs.py
@@ -1,5 +1,6 @@
 import plantuml
 import sys
+import time
 
 f = open(sys.argv[1], 'r').readlines()
 p = plantuml.PlantUML(url="http://www.plantuml.com/plantuml/img/")
@@ -17,8 +18,16 @@ def add_preamble(offset):
 
 def generate_svg(text):
     print(f"generating: {text}")
-    res = p.get_url(plantuml_text=('\n'.join(preamble) + text))
-    print(res)
+    try:
+        res = p.processes(plantuml_text=('\n'.join(preamble) + text))
+    except Exception as e:
+        print(e)
+        print(p.get_url(plantuml_text=('\n'.join(preamble) + text)))
+        return
+
+    name = text.split(' ')
+    open('out/' + name[name.index('namespace') + 1] + '.png', 'wb').write(res)
+    time.sleep(1)
 
 
 def parse_namespace(offset):
@@ -39,6 +48,7 @@ def parse_namespace(offset):
 
 
 offset = 0
+offset = add_preamble(offset)
 while offset < len(f):
-    offset = add_preamble(offset)
+    #offset = add_preamble(offset)
     offset = parse_namespace(offset)