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,13 +12,18 @@ class PostgresConnector:
     """
 
     def __init__(self):
-        self.connection = psycopg2.connect(
-            host=os.getenv("POSTGRES_HOST", "localhost"),
-            port=os.getenv("POSTGRES_PORT", 5432),
-            user=os.getenv("POSTGRES_USER", "postgres"),
-            password=os.getenv("POSTGRES_PASSWORD", "postgres"),
-            database=os.getenv("POSTGRES_DB", "postgres"),
-        )
+
+        try:
+            self.connection = psycopg2.connect(
+                host=os.getenv("POSTGRES_HOST", "localhost"),
+                port=os.getenv("POSTGRES_PORT", 5432),
+                user=os.getenv("POSTGRES_USER", "postgres"),
+                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

@@ -2,4 +2,7 @@ class NotAuthorisedException(Exception):
     pass
 
 class NotExistentDatasetException(Exception):
+    pass
+
+class DatabaseNotConfiguredException(Exception):
     pass