Add DB check before parsing start command

Signed-off-by: Ulincsys <[email protected]>

Author: Ulincsys

augur/application/cli/__init__.py

@@ -32,7 +32,7 @@ def new_func(ctx, *args, **kwargs):
                 You are not connected to the internet.\n \
                 Please connect to the internet to run Augur\n \
                 Consider setting http_proxy variables for limited access installations.")
-            sys.exit()        
+            sys.exit(-1)        
 
     return update_wrapper(new_func, function_internet_connection)
 
@@ -78,7 +78,7 @@ def new_func(ctx, *args, **kwargs):
                 print(f"\n\n{usage} command setup failed\nERROR: connecting to database\nHINT: The {incorrect_values} may be incorrectly specified in {location}\n")
 
             engine.dispose()
-            sys.exit()
+            sys.exit(-2)
 
     return update_wrapper(new_func, function_db_connection)
 

jumpstart/procman.py

@@ -41,6 +41,12 @@ def start(self, component: Union[Component, str], client: Client, *options):
             client.respond(Status.error(f"Invalid component for start: {component}"))
             return
 
+        check_db = run("augur db test-connection".split())
+        
+        if check_db.returncode != 0:
+            client.respond(Status.error(f"Could not communicate with the database: {check_db.returncode}"))
+            return
+        
         if c in (Component.api, Component.frontend, Component.all):
             if self.frontend:
                 client.respond(Status.information(f"The frontend/api is already running"))

jumpstart/tui.py

@@ -41,15 +41,16 @@ def compose(self):
                 with Vertical(classes="status_container"):
                     with Horizontal(classes="info_container"):
                         yield Label("Frontend")
-                        yield Label(id="frontend_label", classes="status_label")
+                        yield Label("X", id="frontend_label", classes="status_label")
                     with Horizontal(classes="info_container"):
                         yield Label("API")
-                        yield Label(id="api_label", classes="status_label")
+                        yield Label("X", id="api_label", classes="status_label")
                     with Horizontal(classes="info_container"):
                         yield Label("Collection")
-                        yield Label(id="collection_label", classes="status_label")
+                        yield Label("X", id="collection_label", classes="status_label")
 
-                yield Button("Status", id="statusbtn")
+                yield Button("Start", id="startbtn")
+                yield Button("Stop", id="stopbtn")
                 yield Button("Exit", variant="error", id="exitbtn")
 
             with Vertical():

jumpstart/tui.tcss

@@ -32,10 +32,13 @@
     dock: right;
     width: 2;
     background: red;
+    color: black;
+    text-align: center;
 }
 
 .label_up {
     background: green;
+    color: green;
 }
 
 #exitbtn {