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

Merge pull request #85 from grillazz/ci-test

Browse Source 
save and update
This commit is contained in:
Jakub Miazek
2023-03-29 21:55:54 +02:00
committed by GitHub
parent 2138e3e2a8 2bcd95556d
commit 336f1e8456
1 changed files with 8 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
app/models/base.py
View File

@@ -1,5 +1,6 @@
from typing import Any from typing import Any
from asyncpg import UniqueViolationError
from fastapi import HTTPException, status from fastapi import HTTPException, status
from sqlalchemy.exc import SQLAlchemyError, IntegrityError from sqlalchemy.exc import SQLAlchemyError, IntegrityError
from sqlalchemy.ext.asyncio import AsyncSession from sqlalchemy.ext.asyncio import AsyncSession
@@ -68,17 +69,16 @@ class Base:
await self.save(db_session) await self.save(db_session)
async def save_or_update(self, db: AsyncSession): async def save_or_update(self, db: AsyncSession):
# TODO: this will be successor of update meth
_success = False
try: try:
db.add(self) db.add(self)
_success = await db.commit() return await db.commit()
return _success
except IntegrityError as exception: except IntegrityError as exception:
if not _success: if isinstance(exception.orig, UniqueViolationError):
# TODO: check if exception is instance of class 'asyncpg.exceptions.UniqueViolationError'
return await db.merge(self) return await db.merge(self)
else: else:
raise exception raise HTTPException(
status_code=status.HTTP_422_UNPROCESSABLE_ENTITY,
detail=repr(exception),
) from exception
finally: finally:
await db.close() await db.close()