style(database): Changed logging

db/connection.py

@@ -1,6 +1,6 @@
-# db/connection.py
 import psycopg2
 import os
+import logging
 from contextlib import contextmanager
 from typing import Generator
 
@@ -8,6 +8,8 @@ from utils.loadDotEnv import initializeENV
 
 initializeENV()
 
+logger = logging.getLogger(__name__)
+
 def PSQLConnect():
     conn = psycopg2.connect(os.getenv('POSTDRESS_CONNECTION'))
     return conn
@@ -22,30 +24,38 @@ def get_connection() -> Generator[psycopg2.extensions.connection, None, None]:
     conn = None
     try:
         conn = PSQLConnect()
-        print("Подключение к БД установлено")
+        logger.info("Подключение к БД установлено")
         yield conn
     except psycopg2.OperationalError as e:
-        print(f"Ошибка подключения к БД: {e}")
+        logger.error(f"Ошибка подключения к БД: {e}")
+        raise
+    except psycopg2.Error as e:
+        logger.error(f"Ошибка PostgreSQL: {e}")
         raise
     except Exception as e:
-        print(f"Неожиданная ошибка: {e}")
+        logger.error(f"Неожиданная ошибка при работе с БД: {e}")
         raise
     finally:
         if conn:
-            conn.close()
-            print("Соединение с БД закрыто")
+            try:
+                conn.close()
+                logger.info("Соединение с БД закрыто")
+            except Exception as e:
+                logger.warning(f"Ошибка при закрытии соединения: {e}")
 
 def test_connection() -> bool:
     try:
         with get_connection() as conn:
             cur = PSQLCursor(conn)
-            cur.execute("SELECT version();")
-            version = cur.fetchone()
-            print(f"Версия PostgreSQL: {version[0]}")
-            cur.close()
-            return True
+            try:
+                cur.execute("SELECT version();")
+                version = cur.fetchone()
+                logger.info(f"Версия PostgreSQL: {version[0]}")
+                return True
+            finally:
+                cur.close()
+                logger.debug("Курсор закрыт")
     except Exception as e:
-        print(f"Тест подключения к БД провален: {e}")
+        logger.error(f"Тест подключения к БД провален: {e}")
         return False
-
 print(test_connection())