оптимизация работы с БД
This commit is contained in:
+17
-17
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user