Better connection errors handling

Author: tisnik

src/quota/connect_pg.py

@@ -12,16 +12,20 @@
 def connect_pg(config: PostgreSQLDatabaseConfiguration) -> Any:
     """Initialize connection to PostgreSQL database."""
     logger.info("Connecting to PostgreSQL storage")
-    connection = psycopg2.connect(
-        host=config.host,
-        port=config.port,
-        user=config.user,
-        password=config.password.get_secret_value(),
-        dbname=config.db,
-        sslmode=config.ssl_mode,
-        # sslrootcert=config.ca_cert_path,
-        gssencmode=config.gss_encmode,
-    )
-    if connection is not None:
-        connection.autocommit = True
-    return connection
+    try:
+        connection = psycopg2.connect(
+            host=config.host,
+            port=config.port,
+            user=config.user,
+            password=config.password.get_secret_value(),
+            dbname=config.db,
+            sslmode=config.ssl_mode,
+            # sslrootcert=config.ca_cert_path,
+            gssencmode=config.gss_encmode,
+        )
+        if connection is not None:
+            connection.autocommit = True
+        return connection
+    except psycopg2.Error as e:
+        logger.exception("Error connecting to PostgreSQL database:\n%s", e)
+        raise

src/quota/connect_sqlite.py

@@ -17,10 +17,9 @@ def connect_sqlite(config: SQLiteDatabaseConfiguration) -> Any:
     connection = None
     try:
         connection = sqlite3.connect(database=config.db_path)
-    except sqlite3.Error as e:
         if connection is not None:
-            connection.close()
+            connection.autocommit = True
+        return connection
+    except sqlite3.Error as e:
         logger.exception("Error initializing SQLite cache:\n%s", e)
         raise
-    connection.autocommit = True
-    return connection