Add error handling for failed connection to database (#129)

NASA-AMMOS · Aug 22, 2019 · 46fbd83 · 46fbd83
1 parent 923cf61
commit 46fbd83
Showing 1 changed file with 9 additions and 5 deletions.
diff --git a/ait/gui/__init__.py b/ait/gui/__init__.py
@@ -164,11 +164,15 @@ def _db_connect(self):
                 break
 
         if datastore:
-            mod, cls = datastore.rsplit('.', 1)
-
-            # Connect to database
-            self.dbconn = getattr(importlib.import_module(mod), cls)()
-            self.dbconn.connect(**other_args)
+            try:
+                mod, cls = datastore.rsplit('.', 1)
+
+                # Connect to database
+                self.dbconn = getattr(importlib.import_module(mod), cls)()
+                self.dbconn.connect(**other_args)
+            except Exception as e:
+                log.error('Error connecting to datastore {}: {}'.format(datastore, e))
+                log.warn('Disabling telemetry playback.')
         else:
             msg = (
                 '[GUI Playback Configuration]'