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

refactor base model crud meths

Browse Source
This commit is contained in:
Jakub Miazek 2023-04-28 13:19:03 +02:00
parent 8a87f15ff5
commit 63014e11db
1 changed files with 10 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
app/models/base.py
View File

@ -4,7 +4,7 @@ from asyncpg import UniqueViolationError
from fastapi import HTTPException, status
from sqlalchemy.exc import SQLAlchemyError, IntegrityError
from sqlalchemy.ext.asyncio import AsyncSession
from sqlalchemy.ext.declarative import as_declarative, declared_attr
from sqlalchemy.orm import as_declarative, declared_attr
@as_declarative()
@ -53,16 +53,19 @@ class Base:
except SQLAlchemyError as ex:
raise HTTPException(status_code=status.HTTP_422_UNPROCESSABLE_ENTITY, detail=repr(ex)) from ex
async def update(self, db_session: AsyncSession, **kwargs):
async def update(self, db: AsyncSession, **kwargs):
"""
:param db_session:
:param kwargs:
:param db:
:param kwargs
:return:
"""
try:
for k, v in kwargs.items():
setattr(self, k, v)
await self.save(db_session)
return await db.commit()
except SQLAlchemyError as ex:
raise HTTPException(status_code=status.HTTP_422_UNPROCESSABLE_ENTITY, detail=repr(ex)) from ex
async def save_or_update(self, db: AsyncSession):
try: