fix(log): replace logger with loguru (again!)

This commit is contained in:
2025-10-26 23:07:43 +03:00
parent 9ecb6a83ba
commit 72f766ab46
6 changed files with 14 additions and 25 deletions
+9 -10
View File
@@ -1,6 +1,6 @@
import psycopg2
import os
import logging
from loguru import logger
from contextlib import contextmanager
from typing import Generator
@@ -8,8 +8,6 @@ from utils.loadDotEnv import initializeENV
initializeENV()
logger = logging.getLogger(__name__)
def PSQLConnect():
conn = psycopg2.connect(os.getenv('POSTDRESS_CONNECTION'))
return conn
@@ -24,16 +22,16 @@ def get_connection() -> Generator[psycopg2.extensions.connection, None, None]:
conn = None
try:
conn = PSQLConnect()
logger.debug("Подключение к БД установлено")
logger.info("Подключение к БД установлено")
yield conn
except psycopg2.OperationalError as e:
logger.error(f"Ошибка подключения к БД: {e}")
logger.error(f"Ошибка подключения к БД:\n{e}")
raise
except psycopg2.Error as e:
logger.error(f"Ошибка PostgreSQL: {e}")
logger.error(f"Ошибка PostgreSQL:\n{e}")
raise
except Exception as e:
logger.error(f"Неожиданная ошибка при работе с БД: {e}")
logger.error(f"Неожиданная ошибка при работе с БД:\n{e}")
raise
finally:
if conn:
@@ -41,7 +39,7 @@ def get_connection() -> Generator[psycopg2.extensions.connection, None, None]:
conn.close()
logger.debug("Соединение с БД закрыто")
except Exception as e:
logger.warning(f"Ошибка при закрытии соединения: {e}")
logger.warning(f"Ошибка при закрытии соединения:\n{e}")
def test_connection() -> bool:
try:
@@ -56,6 +54,7 @@ def test_connection() -> bool:
cur.close()
logger.debug("Курсор закрыт")
except Exception as e:
logger.error(f"Тест подключения к БД провален: {e}")
logger.error(f"Тест подключения к БД провален:\n{e}")
return False
print(test_connection())
logger.debug(test_connection())