Uprava prace s DB, refaktoring

This commit is contained in:
Michal Kváček
2023-09-13 10:04:06 +02:00
parent 64cb544e2d
commit 1760dba99d
27 changed files with 195 additions and 182 deletions
+2 -14
View File
@@ -1,8 +1,4 @@
from contextlib import asynccontextmanager
from typing import AsyncGenerator
from sqlalchemy.ext.asyncio import AsyncSession
from database import async_session
@@ -13,18 +9,10 @@ async def get_session():
try:
yield session
await session.commit()
except:
except Exception as e:
await session.rollback()
print(f"ERROR: {e}")
raise
finally:
session.expunge_all()
await session.close()
async def db_session():
async with async_session() as session:
async with session.begin():
yield session
await session.flush()
await session.commit()
print("CCCCCCCCCCCCCOOOOOOOOOOOOOOMMMMMMMMMMIIIIIIIIIIIITTTTTTTTTTTT")