Fix timestamp output

algorand · Aug 5, 2024 · 2892bc6 · 2892bc6
1 parent 4bb53b4
commit 2892bc6
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 2 deletions.
diff --git a/test/heapwatch/topology-extract-ws-e2e.py b/test/heapwatch/topology-extract-ws-e2e.py
@@ -94,7 +94,11 @@ def main():
                                 remote_name = resolved
                             source = remote_name
                             target = node_name
-                            edges.append((source, target))
+                            if args.timestamp:
+                                # datetime is not serializable, so we store it as string for now
+                                edge = (source, target, {'dt': data["time"]})
+                            else:
+                                edge = (source, target)
 
                         # Check for outgoing connections
                         elif "Made outgoing connection to peer" in data.get('msg', ""):

diff --git a/test/heapwatch/topology-extract-ws.py b/test/heapwatch/topology-extract-ws.py
@@ -68,7 +68,11 @@ def main():
                             remote_name = ip_to_name[remote_ip]
                             source = remote_name
                             target = mapped
-                            edges.append((source, target))
+                            if args.timestamp:
+                                # datetime is not serializable, so we store it as string for now
+                                edge = (source, target, {'dt': data["time"]})
+                            else:
+                                edge = (source, target)
 
                         # Check for outgoing connections
                         elif "Made outgoing connection to peer" in data.get('msg', ""):