27 lines
No EOL
752 B
Python
27 lines
No EOL
752 B
Python
|
|
from contextlib import asynccontextmanager
|
|
import logging
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
@asynccontextmanager
|
|
async def session_scope(session):
|
|
"""Provide a transactional scope around a series of operations."""
|
|
|
|
# session = AsyncSessionLocal()
|
|
try:
|
|
yield session
|
|
await session.commit()
|
|
except Exception as e:
|
|
await session.rollback()
|
|
logger.error(f"Session rollback due to: {str(e)}")
|
|
raise
|
|
finally:
|
|
await session.close()
|
|
|
|
|
|
async def add_entity(session, entity):
|
|
async with session_scope(session) as s: # Use your async session_scope
|
|
s.add(entity) # No need to commit; session_scope takes care of it
|
|
s.commit()
|
|
return "Successfully added entity" |