Add first vis - mindmap

visualisations/mermaid-mindmap.py

@@ -0,0 +1,36 @@
+#!/usr/bin/env python3
+"""
+Reads a GlobalTalk Scraper JSON file and create a mermaid graph
+with the data
+"""
+
+import argparse
+import sys
+import json
+
+
+def jsontomermaid(gts_obj) -> str:
+    map = ""
+    for zone in gts_obj["zones"]:
+        nodes = [x for x in gts_obj["nodes"] if x["zone"] == zone]
+        if not len(nodes):
+            continue
+        map += "  {0}\n{1}\n".format(
+            zone, "\n".join(set(["    " + x["object"] for x in nodes]))
+        )
+
+    return "```mermaid\nmindmap\nroot)GlobalTalk(\n{0}\n```".format(map)
+
+
+def main():
+    parser = argparse.ArgumentParser("mermaid")
+    parser.add_argument("json", type=argparse.FileType("r"))
+    parser.add_argument("--output", type=argparse.FileType("w"), default=sys.stdout)
+    args = parser.parse_args()
+
+    obj = json.load(args.json)
+    args.output.write(jsontomermaid(obj))
+
+
+if __name__ == "__main__":
+    main()