From 20ecdf70d05b08e54958ffaed0011be82c28d9f2 Mon Sep 17 00:00:00 2001
From: Blake Harnden <32446120+bharnden@users.noreply.github.com>
Date: Tue, 21 Apr 2020 13:22:21 -0700
Subject: [PATCH] pygui fixed emane link updates when rejoining session

---
 daemon/core/gui/graph/graph.py | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/daemon/core/gui/graph/graph.py b/daemon/core/gui/graph/graph.py
index e4963f45..40a941b1 100644
--- a/daemon/core/gui/graph/graph.py
+++ b/daemon/core/gui/graph/graph.py
@@ -268,8 +268,11 @@ class CanvasGraph(tk.Canvas):
             return
         network_id = link.network_id if link.network_id else None
         token = create_edge_token(src.id, dst.id, network_id)
-        edge = self.wireless_edges[token]
-        edge.middle_label_text(link.label)
+        if token not in self.wireless_edges:
+            self.add_wireless_edge(src, dst, link)
+        else:
+            edge = self.wireless_edges[token]
+            edge.middle_label_text(link.label)
 
     def draw_session(self, session: core_pb2.Session):
         """