Add all data from StellarBeat to each node

stellar · Jun 18, 2020 · cbff06d · cbff06d
1 parent 09a9449
commit cbff06d
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/scripts/OverlaySurvey.py b/scripts/OverlaySurvey.py
@@ -106,7 +106,12 @@ def augment(args):
     data = requests.get("https://api.stellarbeat.io/v1/nodes").json()
     for obj in data:
         if graph.has_node(obj["publicKey"]):
-            graph.add_node(obj["publicKey"], ip=obj["ip"], name=obj["name"])
+            # NetworkX doesn't allow dictionary properties,
+            # so we will turn them into strings.
+            dict_properties = ["geoData", "quorumSet", "statistics"]
+            for dict_property in dict_properties:
+                obj[dict_property] = json.dumps(obj[dict_property])
+            graph.add_node(obj["publicKey"], **obj)
     nx.write_graphml(graph, args.graphmlOutput)
     sys.exit(0)