From 5e69ea48b3cc4091102a5f41f9c1c3b3eb4725fb Mon Sep 17 00:00:00 2001
From: Blake Harnden <32446120+bharnden@users.noreply.github.com>
Date: Thu, 14 May 2020 23:07:21 -0700
Subject: [PATCH] pygui: fixed tracking for throughputs when joining a session

---
 daemon/core/gui/graph/graph.py | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/daemon/core/gui/graph/graph.py b/daemon/core/gui/graph/graph.py
index 220e122f..2920f9b0 100644
--- a/daemon/core/gui/graph/graph.py
+++ b/daemon/core/gui/graph/graph.py
@@ -327,9 +327,15 @@ class CanvasGraph(tk.Canvas):
                     self.edges[edge.token] = edge
                     self.core.links[edge.token] = edge
                     if link.HasField("interface_one"):
+                        self.core.interface_to_edge[
+                            (node_one.id, link.interface_one.id)
+                        ] = token
                         canvas_node_one.interfaces.append(link.interface_one)
                         edge.src_interface = link.interface_one
                     if link.HasField("interface_two"):
+                        self.core.interface_to_edge[
+                            (node_two.id, link.interface_two.id)
+                        ] = edge.token
                         canvas_node_two.interfaces.append(link.interface_two)
                         edge.dst_interface = link.interface_two
                 elif link.options.unidirectional: