From eae6f7d4d39cbff77fc0785be47d5735c1ceb2c4 Mon Sep 17 00:00:00 2001
From: coderick14
Date: Sun, 25 Jun 2017 19:47:04 +0530
Subject: Catch key not found Exception in pi_ip_map

---
 experiment/views.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/experiment/views.py b/experiment/views.py
index c2424a1..20173fd 100644
--- a/experiment/views.py
+++ b/experiment/views.py
@@ -29,7 +29,9 @@ def initial_login(req):
     except Exception as e:
         return HttpResponse(json.dumps({"STATUS": 400, "MESSAGE": {"IS_IP":"1","DATA":"Invalid username"}}))
             
-    rpi_ip = settings.pi_ip_map[str(assigned_mid)]
+    rpi_ip = settings.pi_ip_map.get(str(assigned_mid))
+    if rpi_ip is None:
+        return HttpResponse(json.dumps({"STATUS": 400, "MESSAGE": {"IS_IP": "1", "DATA":"Board is currently offline"}}))
 
     return HttpResponse(json.dumps({"STATUS": 200, "MESSAGE": {"IS_IP":"1","DATA":rpi_ip}}))
 #    return HttpResponse(key)
-- 
cgit