Merge pull request #132 from openclimatefix/issue/one-connection

only open one session
openclimatefix · Dec 1, 2023 · 982bb8e · 982bb8e
2 parents 320194c + 5d6e20c
commit 982bb8e
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/pv_site_api/session.py b/pv_site_api/session.py
@@ -4,13 +4,17 @@
 
 from pvsite_datamodel.connection import DatabaseConnection
 
+try:
+    connection = DatabaseConnection(url=os.getenv("DB_URL", "not_set"))
+except Exception as e:
+    print(e)
+    print("Could not connect to database")
+
 
 def get_session():
     """Get database settion"""
     if int(os.environ.get("FAKE", 0)):
         yield None
     else:
-        connection = DatabaseConnection(url=os.getenv("DB_URL", "not_set"))
-
         with connection.get_session() as s:
             yield s