From fd5d23cf229dd4fb77ab1d969d22904cb71186dc Mon Sep 17 00:00:00 2001
From: forest <forest.n.johnson@gmail.com>
Date: Wed, 17 Feb 2021 20:56:52 -0600
Subject: [PATCH] prevent keyError when double_check_capsul_address returns
 None

---
 capsulflask/console.py             | 3 +++
 capsulflask/templates/capsuls.html | 2 ++
 2 files changed, 5 insertions(+)

diff --git a/capsulflask/console.py b/capsulflask/console.py
index b0d4c41..8632e1b 100644
--- a/capsulflask/console.py
+++ b/capsulflask/console.py
@@ -63,6 +63,9 @@ def index():
     if result is not None:
       vm["ipv4"] = result.ipv4
       vm["state"] = result.state
+    else:
+      vm["state"] = "unknown"
+
 
   vms = list(map(
     lambda x: dict(
diff --git a/capsulflask/templates/capsuls.html b/capsulflask/templates/capsuls.html
index ea8dfc6..161c000 100644
--- a/capsulflask/templates/capsuls.html
+++ b/capsulflask/templates/capsuls.html
@@ -38,6 +38,8 @@
               <td class="capsul-status waiting-pulse">•</td>
             {% elif vm['state'] == 'crashed' or vm['state'] == 'blocked' or vm['state'] == 'stopped'  %}
               <td class="capsul-status red">•</td>
+            {% elif vm['state'] == 'unknown' %}
+              <td class="capsul-status">?</td>
             {% else  %}
               <td class="capsul-status green">•</td>
             {% endif  %}