1 files changed, 20 insertions, 0 deletions
diff --git a/trygvis/eda/__init__.py b/trygvis/eda/__init__.py
new file mode 100644
index 0000000..e6ccb44
--- /dev/null
+++ b/trygvis/eda/__init__.py
@@ -0,0 +1,20 @@
+import sys
+
+from os.path import isfile
+from . import cli
+
+
+def write_graph(gen_g, filename=None, force_write=False):
+    if filename is not None:
+        if force_write or not isfile(filename):
+            g = gen_g()
+            bs = g.serialize(encoding='utf-8', format='turtle')
+            with open(filename, "wb") as f:
+                f.write(bs)
+            cli.info("Wrote %s" % filename)
+        else:
+            cli.info("Skipped writing %s, already exists" % filename)
+    else:
+        g = gen_g()
+        bs = g.serialize(encoding='utf-8', format='turtle')
+        sys.stdout.buffer.write(bs)