fix(log): replace logger with loguru (again!)
This commit is contained in:
+9
-10
@@ -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())
|
||||
@@ -1,11 +1,9 @@
|
||||
from typing import List, Optional
|
||||
from datetime import datetime, date
|
||||
import logging
|
||||
from loguru import logger
|
||||
from db.connection import get_connection
|
||||
from model.ai_prediction import AIPrediction
|
||||
|
||||
logger = logging.getLogger(__name__)
|
||||
|
||||
class AIPredictionsRepository:
|
||||
def get_all(self) -> List[AIPrediction]:
|
||||
try:
|
||||
|
||||
@@ -1,12 +1,10 @@
|
||||
# db/repositories/inventory_repository.py
|
||||
from typing import List, Optional, Tuple
|
||||
from datetime import datetime
|
||||
import logging
|
||||
from loguru import logger
|
||||
from db.connection import get_connection
|
||||
from model.inventory import InventoryRecord
|
||||
|
||||
logger = logging.getLogger(__name__)
|
||||
|
||||
class InventoryRepository:
|
||||
def get_all(self) -> List[InventoryRecord]:
|
||||
try:
|
||||
|
||||
@@ -1,10 +1,8 @@
|
||||
from typing import List, Optional
|
||||
import logging
|
||||
from loguru import logger
|
||||
from db.connection import get_connection
|
||||
from model.product import Product
|
||||
|
||||
logger = logging.getLogger(__name__)
|
||||
|
||||
class ProductRepository:
|
||||
def get_all(self) -> List[Product]:
|
||||
try:
|
||||
|
||||
@@ -1,10 +1,8 @@
|
||||
from typing import List, Optional
|
||||
import logging
|
||||
from loguru import logger
|
||||
from db.connection import get_connection
|
||||
from model.robot import Robot
|
||||
|
||||
logger = logging.getLogger(__name__)
|
||||
|
||||
class RobotRepository:
|
||||
def get_all(self) -> List[Robot]:
|
||||
try:
|
||||
|
||||
@@ -1,10 +1,8 @@
|
||||
from typing import List, Optional
|
||||
import logging
|
||||
from loguru import logger
|
||||
from model.user import User
|
||||
from db.connection import get_connection
|
||||
|
||||
logger = logging.getLogger(__name__)
|
||||
|
||||
class UserRepository:
|
||||
def get_all(self) -> List[User]:
|
||||
try:
|
||||
|
||||
Reference in New Issue
Block a user