avnazv/toolbox
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

создана часть бекенда

Browse Source
This commit is contained in:
Macbook
2025-11-30 19:31:06 +03:00
parent 4ae09c36b3
commit dc85e7c0c9
35 changed files with 1573 additions and 149 deletions
Download Patch File Download Diff File Expand all files Collapse all files
db/__init__.py
+16 -16
View File
@@ -4,7 +4,7 @@ from sqlalchemy.ext.asyncio import create_async_engine, async_sessionmaker
from sqlalchemy.exc import InvalidRequestError
from sqlalchemy.pool import NullPool
import config
from utils import loggerDB
from utils import logger
DATABASE_URL = f"postgresql+asyncpg://{config.DB_USER}:{config.DB_PASS}@{config.DB_HOST}:{config.DB_PORT}/{config.DB_NAME}"
@@ -22,43 +22,43 @@ class CRUD:
is_lst = isinstance(db_data, list)
async with SessionLocal() as db:
if is_lst:
loggerDB.info(f"Создаю {len(db_data)} записей")
logger.info(f"Создаю {len(db_data)} записей")
try:
db.add_all(db_data)
except InvalidRequestError:
for data in db_data:
await db.merge(data)
else:
loggerDB.info("Создаю запись")
logger.info("Создаю запись")
db.add(db_data)
await db.commit()
if refresh:
if is_lst:
loggerDB.info(f"Обновляю {len(db_data)} записей")
logger.info(f"Обновляю {len(db_data)} записей")
for data in db_data:
await db.refresh(data)
else:
loggerDB.info("Обновляю запись")
logger.info("Обновляю запись")
await db.refresh(db_data)
loggerDB.info("Запись создана")
logger.info("Запись создана")
return db_data if refresh else None
except Exception as e:
loggerDB.error(f"Ошибка создания: {str(e)}", exc_info=True)
logger.error(f"Ошибка создания: {str(e)}", exc_info=True)
return None
async def read(query, all: bool = False):
try:
async with SessionLocal() as db:
loggerDB.info(f"Чтение записей. Все: {all}")
logger.info(f"Чтение записей. Все: {all}")
results = await db.execute(query)
loggerDB.info(f"Чтение завершено")
logger.info(f"Чтение завершено")
return (
results.unique().scalars().all()
if all
else results.unique().scalars().first()
)
except Exception as e:
loggerDB.error(f"Ошибка чтения: {str(e)}", exc_info=True)
logger.error(f"Ошибка чтения: {str(e)}", exc_info=True)
return None
async def delete(db_data) -> bool:
@@ -87,18 +87,18 @@ class CRUD:
async with SessionLocal() as db:
try:
if isinstance(db_data, list):
loggerDB.info(f"Удаляю записей: {len(db_data)}")
logger.info(f"Удаляю записей: {len(db_data)}")
for data in db_data:
await deleteFromDB(data, db)
else:
loggerDB.info("Удаляю запись")
logger.info("Удаляю запись")
await deleteFromDB(db_data, db)
await db.commit()
loggerDB.info("Запись удалена")
logger.info("Запись удалена")
return True
except Exception as e:
await db.rollback()
loggerDB.error(f"Ошибка удаления: {str(e)}", exc_info=True)
logger.error(f"Ошибка удаления: {str(e)}", exc_info=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()
loggerDB.info("Запись обновлена")
logger.info("Запись обновлена")
return item
except Exception as e:
await db.rollback()
loggerDB.error(f"Ошибка обновления: {str(e)}", exc_info=True)
logger.error(f"Ошибка обновления: {str(e)}", exc_info=True)
return None