оптимизация работы с БД

This commit is contained in:
2025-12-06 13:48:42 +03:00
parent f07843de5a
commit 54bf21d52d
8 changed files with 80 additions and 63 deletions
+17 -17
View File
@@ -22,47 +22,47 @@ class CRUD:
is_lst = isinstance(db_data, list)
async with SessionLocal() as db:
if is_lst:
logger.info(f"Создаю {len(db_data)} записей")
logger.debug(f"Создаю {len(db_data)} записей")
try:
db.add_all(db_data)
except InvalidRequestError:
for data in db_data:
await db.merge(data)
else:
logger.info("Создаю запись")
logger.debug("Создаю запись")
db.add(db_data)
await db.commit()
if refresh:
if is_lst:
logger.info(f"Обновляю {len(db_data)} записей")
logger.debug(f"Обновляю {len(db_data)} записей")
for data in db_data:
await db.refresh(data)
else:
logger.info("Обновляю запись")
logger.debug("Обновляю запись")
await db.refresh(db_data)
logger.info("Запись создана")
logger.debug("Запись создана")
return db_data if refresh else None
except Exception as e:
logger.error(f"Ошибка создания: {str(e)}", exc_info=True)
logger.error(f"Ошибка создания: {str(e)}", exc_debug=True)
return None
async def read(query, all: bool = False):
try:
async with SessionLocal() as db:
logger.info(f"Чтение записей. Все: {all}")
logger.debug(f"Чтение записей. Все: {all}")
results = await db.execute(query)
logger.info(f"Чтение завершено")
logger.debug(f"Чтение завершено")
return (
results.unique().scalars().all()
if all
else results.unique().scalars().first()
)
except Exception as e:
logger.error(f"Ошибка чтения: {str(e)}", exc_info=True)
logger.error(f"Ошибка чтения: {str(e)}", exc_debug=True)
return None
async def delete(db_data) -> bool:
def itemInfo(instance):
def itemdebug(instance):
from sqlalchemy import inspect
state = inspect(instance)
@@ -78,7 +78,7 @@ class CRUD:
return {"key": pKey, "value": pValue, "class": instance.__class__}
async def deleteFromDB(data, db):
itemData = itemInfo(data)
itemData = itemdebug(data)
query = delete(itemData["class"]).where(
getattr(itemData["class"], itemData["key"]) == itemData["value"]
)
@@ -87,18 +87,18 @@ class CRUD:
async with SessionLocal() as db:
try:
if isinstance(db_data, list):
logger.info(f"Удаляю записей: {len(db_data)}")
logger.debug(f"Удаляю записей: {len(db_data)}")
for data in db_data:
await deleteFromDB(data, db)
else:
logger.info("Удаляю запись")
logger.debug("Удаляю запись")
await deleteFromDB(db_data, db)
await db.commit()
logger.info("Запись удалена")
logger.debug("Запись удалена")
return True
except Exception as e:
await db.rollback()
logger.error(f"Ошибка удаления: {str(e)}", exc_info=True)
logger.error(f"Ошибка удаления: {str(e)}", exc_debug=True)
return False
async def update(db_data, id, **kwargs):
@@ -107,9 +107,9 @@ class CRUD:
query = update(db_data).where(db_data.id == id).values(**kwargs)
item = await db.execute(query)
await db.commit()
logger.info("Запись обновлена")
logger.debug("Запись обновлена")
return await db.get(db_data, id)
except Exception as e:
await db.rollback()
logger.error(f"Ошибка обновления: {str(e)}", exc_info=True)
logger.error(f"Ошибка обновления: {str(e)}", exc_debug=True)
return None