PersonalPublic/fastapi-sqlalchemy-asyncpg
1
0
Fork 0
mirror of https://github.com/grillazz/fastapi-sqlalchemy-asyncpg.git synced 2025-11-30 13:20:40 +03:00
Code Issues Packages Projects Releases Wiki Activity

wip: add exception handler

Browse Source
This commit is contained in:
grillazz
2025-08-23 18:22:35 +02:00
parent d722504e55
commit 69f3dc3fd8
5 changed files with 52 additions and 87 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
app/models/base.py
View File

@@ -20,14 +20,11 @@ class Base(DeclarativeBase):
return self.__name__.lower()
async def save(self, db_session: AsyncSession):
try:
db_session.add(self)
await db_session.flush()
await db_session.refresh(self)
return self
except SQLAlchemyError as ex:
await logger.aerror(f"Error inserting instance of {self}: {repr(ex)}")
raise # This will make the exception handler catch it
db_session.add(self)
await db_session.flush()
await db_session.refresh(self)
return self
async def delete(self, db_session: AsyncSession):
try:
@@ -61,5 +58,4 @@ class Base(DeclarativeBase):
status_code=status.HTTP_422_UNPROCESSABLE_ENTITY,
detail=repr(exception),
) from exception
finally:
await db_session.close()