From 7a5b02cb024d48d9ad944584028ae0495c1dbb48 Mon Sep 17 00:00:00 2001
From: Hrishikesh Bihani <hrishikesh.bihani@idfy.com>
Date: Wed, 22 May 2019 12:45:14 +0530
Subject: [PATCH] Handle FileNotFound exception for get_host_info()

---
 run_demo_server.py | 25 ++++++++++++++++---------
 1 file changed, 16 insertions(+), 9 deletions(-)

diff --git a/run_demo_server.py b/run_demo_server.py
index 59e5534d..979069e7 100755
--- a/run_demo_server.py
+++ b/run_demo_server.py
@@ -13,6 +13,8 @@
 import logging
 import collections
 
+import traceback
+
 logger = logging.getLogger(__name__)
 logger.setLevel(logging.INFO)
 
@@ -20,15 +22,20 @@
 @functools.lru_cache(maxsize=1)
 def get_host_info():
     ret = {}
-    with open('/proc/cpuinfo') as f:
-        ret['cpuinfo'] = f.read()
-
-    with open('/proc/meminfo') as f:
-        ret['meminfo'] = f.read()
-
-    with open('/proc/loadavg') as f:
-        ret['loadavg'] = f.read()
-
+    """Handle /proc/cpuinfo file not found on OSX.
+    The demo server will crash in such a case."""
+    try:
+        with open('/proc/cpuinfo') as f:
+            ret['cpuinfo'] = f.read()
+
+        with open('/proc/meminfo') as f:
+            ret['meminfo'] = f.read()
+
+        with open('/proc/loadavg') as f:
+            ret['loadavg'] = f.read()
+    except Exception:
+        print(traceback.format_exc())
+        pass
     return ret