Adjusted files for isort & autopep

This commit is contained in:
creyD
2024-11-15 11:39:59 +00:00
committed by github-actions[bot]
parent 9bba5b0a4e
commit 6f09c2ef4c

View File

@@ -11,28 +11,31 @@ from .models.base import Base
T = TypeVar("T", bound=Base) T = TypeVar("T", bound=Base)
@overload @overload
async def get_object_or_404( async def get_object_or_404(
db_class: Type[T], db_class: Type[T],
id: UUID | str, id: UUID | str,
db: AsyncSession, db: AsyncSession,
expunge: bool = False, expunge: bool = False,
lookup_column: str = "id" lookup_column: str = "id",
) -> T: ) -> T:
pass pass
@overload @overload
def get_object_or_404( def get_object_or_404(
db_class: Type[T], db_class: Type[T], id: UUID | str, db: Session, expunge: bool = False, lookup_column: str = "id"
id: UUID | str,
db: Session,
expunge: bool = False,
lookup_column: str = "id"
) -> T: ) -> T:
pass pass
def get_object_or_404( def get_object_or_404(
db_class: Type[T], id: UUID | str, db: Session | AsyncSession, expunge: bool = False, lookup_column: str = "id" db_class: Type[T],
id: UUID | str,
db: Session | AsyncSession,
expunge: bool = False,
lookup_column: str = "id",
) -> T: ) -> T:
async def _get_async_object() -> T: async def _get_async_object() -> T:
@@ -61,7 +64,6 @@ def get_object_or_404(
raise HTTPException(status_code=404, detail="Invalid session type. Expected Session or AsyncSession.") # type: ignore raise HTTPException(status_code=404, detail="Invalid session type. Expected Session or AsyncSession.") # type: ignore
# TODO: Add testing # TODO: Add testing
@overload @overload
async def create_obj_from_data( async def create_obj_from_data(
@@ -69,26 +71,20 @@ async def create_obj_from_data(
model: Type[T], model: Type[T],
db: AsyncSession, db: AsyncSession,
additional_data: dict = {}, additional_data: dict = {},
exclude: dict = {} exclude: dict = {},
) -> T: ) -> T:
pass pass
@overload @overload
def create_obj_from_data( def create_obj_from_data(
data: BaseModel, data: BaseModel, model: Type[T], db: Session, additional_data: dict = {}, exclude: dict = {}
model: Type[T],
db: Session,
additional_data: dict = {},
exclude: dict = {}
) -> T: ) -> T:
pass pass
def create_obj_from_data( def create_obj_from_data(
data: BaseModel, data: BaseModel, model: Type[T], db: Session | AsyncSession, additional_data={}, exclude={}
model: Type[T],
db: Session | AsyncSession,
additional_data={},
exclude={}
) -> T: ) -> T:
obj_data = data.model_dump(exclude=exclude) | additional_data obj_data = data.model_dump(exclude=exclude) | additional_data
obj = model(**obj_data) obj = model(**obj_data)
@@ -123,10 +119,11 @@ async def update_obj_from_data(
partial: bool = True, partial: bool = True,
ignore_fields: list = [], ignore_fields: list = [],
additional_data: dict = {}, additional_data: dict = {},
exclude: dict = {} exclude: dict = {},
) -> T: ) -> T:
pass pass
@overload @overload
def update_obj_from_data( def update_obj_from_data(
data: BaseModel, data: BaseModel,
@@ -136,10 +133,11 @@ def update_obj_from_data(
partial: bool = True, partial: bool = True,
ignore_fields: list = [], ignore_fields: list = [],
additional_data: dict = {}, additional_data: dict = {},
exclude: dict = {} exclude: dict = {},
) -> T: ) -> T:
pass pass
def update_obj_from_data( def update_obj_from_data(
data: BaseModel, data: BaseModel,
model: Type[T], model: Type[T],
@@ -179,23 +177,19 @@ def update_obj_from_data(
else: else:
raise HTTPException(status_code=404, detail="Invalid session type. Expected Session or AsyncSession.") # type: ignore raise HTTPException(status_code=404, detail="Invalid session type. Expected Session or AsyncSession.") # type: ignore
# TODO: Add testing # TODO: Add testing
@overload @overload
async def delete_object( async def delete_object(db_class: Type[T], id: UUID | str, db: AsyncSession) -> None:
db_class: Type[T], id: UUID | str, db: AsyncSession
) -> None:
pass pass
@overload @overload
def delete_object( def delete_object(db_class: Type[T], id: UUID | str, db: Session) -> None:
db_class: Type[T], id: UUID | str, db: Session
) -> None:
pass pass
def delete_object( def delete_object(db_class: Type[T], id: UUID | str, db: Session | AsyncSession) -> None:
db_class: Type[T], id: UUID | str, db: Session | AsyncSession
) -> None:
async def _delete_async_obj() -> None: async def _delete_async_obj() -> None:
query = select(db_class).filter(db_class.id == id) query = select(db_class).filter(db_class.id == id)
result = await db.execute(query) result = await db.execute(query)