Attach version information to each node

stellar · Jun 15, 2020 · b91e443 · b91e443
1 parent f5b2957
commit b91e443
Showing 1 changed file with 12 additions and 4 deletions.
diff --git a/scripts/OverlaySurvey.py b/scripts/OverlaySurvey.py
@@ -9,10 +9,10 @@
 import time
 
 
-def add_new_node(graph, label):
-    if graph.has_node(label):
+def add_new_node(graph, label, version=""):
+    if graph.has_node(label) and version is "":
         return
-    graph.add_node(label, label=label)
+    graph.add_node(label, label=label, version=version)
 
 
 def add_new_edge(graph, u, v):
@@ -51,7 +51,7 @@ def update_results(graph, parent_info, parent_key, results, is_inbound):
 
         results[direction_tag][other_key] = peer
         add_new_node(graph, parent_key)
-        add_new_node(graph, other_key)
+        add_new_node(graph, other_key, peer["version"])
         add_new_edge(graph, other_key, parent_key)
 
     if "numTotalInboundPeers" in parent_info:
@@ -124,6 +124,14 @@ def run_survey(args):
     duration = int(args.duration)
     PARAMS = {'duration': duration}
 
+    add_new_node(G,
+                 requests
+                 .get(URL + "/scp?limit=0&fullkeys=true")
+                 .json()
+                 ["you"],
+                 requests.get(URL + "/info").json()["info"]["build"])
+
+
     # reset survey
     r = requests.get(url=STOP_SURVEY)