Fix bug in refactoring of #55, add SVG output option

wireviz · Jul 5, 2020 · de1e45f · 17o2 · Jul 5, 2020 · de1e45f
1 parent 144c99e
commit de1e45f
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 1 deletion.
diff --git a/src/wireviz/Harness.py b/src/wireviz/Harness.py
@@ -264,6 +264,15 @@ def png(self):
         data.seek(0)
         return data.read()
 
+    @property
+    def svg(self):
+        from io import BytesIO
+        graph = self.create_graph()
+        data = BytesIO()
+        data.write(graph.pipe(format='svg'))
+        data.seek(0)
+        return data.read()
+
     def output(self, filename, directory='_output', view=False, cleanup=True, fmt='pdf', gen_bom=False):
         # graphical output
         graph = self.create_graph()

diff --git a/src/wireviz/wireviz.py b/src/wireviz/wireviz.py
@@ -205,8 +205,10 @@ def check_designators(what, where):
 
         if 'png' in return_types:
             returns.append(harness.png)
+        if 'svg' in return_types:
+            returns.append(harness.svg)
 
-        return tuple(returns) if len(returns) != 1 else returns
+        return tuple(returns) if len(returns) != 1 else returns[0]
 
 
 def parse_file(yaml_file, file_out=None, generate_bom=False):