chore: add non-existent database error check

server/app.py

@@ -1,4 +1,7 @@
 import os
+import pandas as pd
+import traceback
+import json
 
 from dotenv import load_dotenv
 from flask import Flask, jsonify, request
@@ -19,10 +22,6 @@ from server.auth import AuthManager
 from server.datasets import DatasetManager
 from server.utils import get_request_filters
 
-import pandas as pd
-import traceback
-import json
-
 app = Flask(__name__)
 
 # Env Variables

server/datasets.py

@@ -2,7 +2,6 @@ import pandas as pd
 from server.db.database import PostgresConnector
 from psycopg2.extras import Json
 from server.exceptions import NotAuthorisedException
-from flask_jwt_extended import get_jwt_identity
 
 class DatasetManager:
     def __init__(self, db: PostgresConnector):

server/db/database.py

@@ -1,8 +1,9 @@
 import os
 import psycopg2
-import pandas as pd
 from psycopg2.extras import RealDictCursor
-from psycopg2.extras import execute_batch, Json
+from psycopg2.extras import execute_batch
+
+from server.exceptions import DatabaseNotConfiguredException
 
 
 class PostgresConnector:
@@ -11,6 +12,8 @@ class PostgresConnector:
     """
 
     def __init__(self):
+
+        try:
             self.connection = psycopg2.connect(
                 host=os.getenv("POSTGRES_HOST", "localhost"),
                 port=os.getenv("POSTGRES_PORT", 5432),
@@ -18,6 +21,9 @@ class PostgresConnector:
                 password=os.getenv("POSTGRES_PASSWORD", "postgres"),
                 database=os.getenv("POSTGRES_DB", "postgres"),
             )
+        except psycopg2.OperationalError as e:
+            raise DatabaseNotConfiguredException(f"Ensure database is up and running: {e}")
+        
         self.connection.autocommit = False
 
     def execute(self, query, params=None, fetch=False) -> list:

server/exceptions.py

@@ -3,3 +3,6 @@ class NotAuthorisedException(Exception):
 
 class NotExistentDatasetException(Exception):
     pass
+
+class DatabaseNotConfiguredException(Exception):
+    pass