quaik8/sdb2-backend
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Refactor error logging in database functions for consistency and clarity

Browse Source
This commit is contained in:
JSC
2025-07-25 11:09:50 +02:00
parent d8aaedf851
commit c5c3e8442e
1 changed files with 6 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
app/core/database.py
View File

@@ -24,12 +24,13 @@ engine: AsyncEngine = create_async_engine(
async def get_db() -> AsyncGenerator[AsyncSession, None]: async def get_db() -> AsyncGenerator[AsyncSession, None]:
"""Get a database session for dependency injection."""
logger = get_logger(__name__) logger = get_logger(__name__)
async with AsyncSession(engine) as session: async with AsyncSession(engine) as session:
try: try:
yield session yield session
except Exception as e: except Exception:
logger.exception("Database session error: %s", e) logger.exception("Database session error")
await session.rollback() await session.rollback()
raise raise
finally: finally:
@@ -37,12 +38,13 @@ async def get_db() -> AsyncGenerator[AsyncSession, None]:
async def init_db() -> None: async def init_db() -> None:
"""Initialize the database and create tables if they do not exist."""
logger = get_logger(__name__) logger = get_logger(__name__)
try: try:
logger.info("Initializing database tables") logger.info("Initializing database tables")
async with engine.begin() as conn: async with engine.begin() as conn:
await conn.run_sync(SQLModel.metadata.create_all) await conn.run_sync(SQLModel.metadata.create_all)
logger.info("Database tables created successfully") logger.info("Database tables created successfully")
except Exception as e: except Exception:
logger.exception("Failed to initialize database: %s", e) logger.exception("Failed to initialize database")
raise raise