refactor

tests/app/api/nonsense.py

@@ -0,0 +1,40 @@
+from fastapi import APIRouter, Depends, status
+from sqlalchemy.ext.asyncio import AsyncSession
+
+from the_app.database import get_db
+from the_app.models.nonsense import Nonsense
+from the_app.schemas.nnonsense import NonsenseResponse, NonsenseSchema
+
+router = APIRouter(prefix="/v1/nonsense")
+
+
+@router.post("/", status_code=status.HTTP_201_CREATED, response_model=NonsenseResponse)
+async def create_nonsense(payload: NonsenseSchema, db_session: AsyncSession = Depends(get_db)):
+    nonsense = Nonsense(**payload.dict())
+    await nonsense.save(db_session)
+    return nonsense
+
+
+@router.get("/", response_model=NonsenseResponse)
+async def find_nonsense(
+    name: str,
+    db_session: AsyncSession = Depends(get_db),
+):
+    return await Nonsense.find(db_session, name)
+
+
+@router.delete("/")
+async def delete_nonsense(name: str, db_session: AsyncSession = Depends(get_db)):
+    nonsense = await Nonsense.find(db_session, name)
+    return await nonsense.delete(nonsense, db_session)
+
+
+@router.patch("/", response_model=NonsenseResponse)
+async def update_nonsense(
+    payload: NonsenseSchema,
+    name: str,
+    db_session: AsyncSession = Depends(get_db),
+):
+    nonsense = await Nonsense.find(db_session, name)
+    await nonsense.update(db_session, **payload.dict())
+    return nonsense

tests/app/api/shakespeare.py

@@ -0,0 +1,15 @@
+from fastapi import APIRouter, Depends, status
+from sqlalchemy.ext.asyncio import AsyncSession
+
+from the_app.database import get_db
+from the_app.models.shakespeare import Paragraph
+
+router = APIRouter(prefix="/v1/shakespeare")
+
+
+@router.get("/",)
+async def find_paragraph(
+    character: str,
+    db_session: AsyncSession = Depends(get_db),
+):
+    return await Paragraph.find(db_session=db_session, character=character)

tests/app/api/stuff.py

@@ -0,0 +1,58 @@
+from fastapi import APIRouter, Depends, HTTPException, status
+from sqlalchemy.exc import SQLAlchemyError
+from sqlalchemy.ext.asyncio import AsyncSession
+
+from the_app.database import get_db
+from the_app.models.stuff import Stuff
+from the_app.schemas.stuff import StuffResponse, StuffSchema
+from the_app.utils import get_logger
+
+router = APIRouter(prefix="/v1/stuff")
+
+logger = get_logger(__name__)
+
+
+@router.post("/add_many", status_code=status.HTTP_201_CREATED)
+async def create_multi_stuff(payload: list[StuffSchema], db_session: AsyncSession = Depends(get_db)):
+    try:
+        stuff_instances = [Stuff(name=stuf.name, description=stuf.description) for stuf in payload]
+        db_session.add_all(stuff_instances)
+        await db_session.commit()
+    except SQLAlchemyError as ex:
+        # logger.exception(ex)
+        raise HTTPException(status_code=status.HTTP_422_UNPROCESSABLE_ENTITY, detail=repr(ex))
+    else:
+        logger.info(f"{len(stuff_instances)} instances of Stuff inserted into database.")
+        return True
+
+
+@router.post("", status_code=status.HTTP_201_CREATED, response_model=StuffResponse)
+async def create_stuff(payload: StuffSchema, db_session: AsyncSession = Depends(get_db)):
+    stuff = Stuff(name=payload.name, description=payload.description)
+    await stuff.save(db_session)
+    return stuff
+
+
+@router.get("/{name}", response_model=StuffResponse)
+async def find_stuff(
+    name: str,
+    db_session: AsyncSession = Depends(get_db),
+):
+    return await Stuff.find(db_session, name)
+
+
+@router.delete("/{name}")
+async def delete_stuff(name: str, db_session: AsyncSession = Depends(get_db)):
+    stuff = await Stuff.find(db_session, name)
+    return await Stuff.delete(stuff, db_session)
+
+
+@router.patch("/{name}", response_model=StuffResponse)
+async def update_stuff(
+    payload: StuffSchema,
+    name: str,
+    db_session: AsyncSession = Depends(get_db),
+):
+    stuff = await Stuff.find(db_session, name)
+    await stuff.update(db_session, **payload.dict())
+    return stuff