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
-35
View File
@@ -1,35 +0,0 @@
import asyncio
import functools
from typing import Callable
from database import async_session
def transactional(func: Callable) -> Callable:
@functools.wraps(func)
async def _wrapper(*args, **kwargs):
# db_session = db_session_context.get()
# if db_session:
# return func(*args, **kwargs)
db_session = async_session()
print("STARTUJI TRANSAKCI")
db_session.begin()
# db_session_context.set(db_session)
try:
kwargs['db'] = db_session
result = await func(*args, **kwargs)
await db_session.commit()
print("KONEC TRANSAKCE V DEKORATORU")
except Exception as e:
await db_session.rollback()
raise
finally:
await db_session.close()
# db_session_context.set(None)
return result
return _wrapper
+1 -1
View File
@@ -9,6 +9,6 @@ def error_logging(func):
try:
return await func(*args, **kwargs)
except NoResultFound as e:
raise GraphQLError(f"Not found", original_error=e)
raise GraphQLError("Not found", original_error=e)
return decorator