From 7fc6b7232c488af2c220f536123b8a35eaff0e17 Mon Sep 17 00:00:00 2001
From: Donkie <daniel.cf.hultgren@gmail.com>
Date: Sun, 2 Apr 2023 14:27:39 +0200
Subject: [PATCH] Added full CRUD operations for vendors

---
 spoolson/api/v1/vendor.py   | 82 ++++++++++++++++++++++++++++++++++++-
 spoolson/database/vendor.py | 38 +++++++++++++++++
 2 files changed, 119 insertions(+), 1 deletion(-)

diff --git a/spoolson/api/v1/vendor.py b/spoolson/api/v1/vendor.py
index 418671d..5f7534c 100644
--- a/spoolson/api/v1/vendor.py
+++ b/spoolson/api/v1/vendor.py
@@ -1,7 +1,87 @@
 """Vendor related endpoints."""
 
-from fastapi import APIRouter
+from typing import Annotated, Optional, Union
+
+from fastapi import APIRouter, Depends
+from fastapi.exceptions import RequestValidationError
+from pydantic import BaseModel, Field
+from pydantic.error_wrappers import ErrorWrapper
+from sqlalchemy.ext.asyncio import AsyncSession
+
+from spoolson.api.v1.models import Message, Vendor
+from spoolson.database import vendor
+from spoolson.database.database import get_db_session
 
 router = APIRouter(
+    prefix="/vendor",
     tags=["vendor"],
 )
+
+# ruff: noqa: D103
+
+
+class VendorParameters(BaseModel):
+    name: str = Field(max_length=64, description="Vendor name.")
+    comment: Optional[str] = Field(max_length=1024, description="Free text comment about this vendor.")
+
+
+class VendorUpdateParameters(VendorParameters):
+    name: Optional[str] = Field(max_length=64, description="Vendor name.")
+    comment: Optional[str] = Field(max_length=1024, description="Free text comment about this vendor.")
+
+
+@router.get("/")
+async def find(_name: Union[int, None] = None) -> list[Vendor]:
+    return []
+
+
+@router.get("/{vendor_id}")
+async def get(
+    db: Annotated[AsyncSession, Depends(get_db_session)],
+    vendor_id: int,
+) -> Vendor:
+    db_item = await vendor.get_by_id(db, vendor_id)
+    return Vendor.from_db(db_item)
+
+
+@router.post("/")
+async def create(
+    db: Annotated[AsyncSession, Depends(get_db_session)],
+    body: VendorParameters,
+) -> Vendor:
+    db_item = await vendor.create(
+        db=db,
+        name=body.name,
+        comment=body.comment,
+    )
+
+    return Vendor.from_db(db_item)
+
+
+@router.patch("/{vendor_id}")
+async def update(
+    db: Annotated[AsyncSession, Depends(get_db_session)],
+    vendor_id: int,
+    body: VendorUpdateParameters,
+) -> Vendor:
+    patch_data = body.dict(exclude_unset=True)
+
+    if "name" in patch_data and body.name is None:
+        raise RequestValidationError([ErrorWrapper(ValueError("name cannot be unset"), ("query", "name"))])
+
+    db_item = await vendor.update(
+        db=db,
+        vendor_id=vendor_id,
+        data=patch_data,
+    )
+
+    return Vendor.from_db(db_item)
+
+
+@router.delete("/{vendor_id}")
+async def delete(
+    db: Annotated[AsyncSession, Depends(get_db_session)],
+    vendor_id: int,
+) -> Message:
+    await vendor.delete(db, vendor_id)
+    return Message(message="Success!")
diff --git a/spoolson/database/vendor.py b/spoolson/database/vendor.py
index 8effa82..01c1797 100644
--- a/spoolson/database/vendor.py
+++ b/spoolson/database/vendor.py
@@ -1,14 +1,52 @@
 """Helper functions for interacting with vendor database objects."""
 
+from typing import Optional
+
 from sqlalchemy.ext.asyncio import AsyncSession
 
 from spoolson.database import models
 from spoolson.exceptions import ItemNotFoundError
 
 
+async def create(
+    *,
+    db: AsyncSession,
+    name: Optional[str] = None,
+    comment: Optional[str] = None,
+) -> models.Vendor:
+    """Add a new vendor to the database."""
+    db_item = models.Vendor(
+        name=name,
+        comment=comment,
+    )
+    db.add(db_item)
+    await db.flush()
+    return db_item
+
+
 async def get_by_id(db: AsyncSession, vendor_id: int) -> models.Vendor:
     """Get a vendor object from the database by the unique ID."""
     vendor = await db.get(models.Vendor, vendor_id)
     if vendor is None:
         raise ItemNotFoundError(f"No vendor with ID {vendor_id} found.")
     return vendor
+
+
+async def update(
+    *,
+    db: AsyncSession,
+    vendor_id: int,
+    data: dict,
+) -> models.Vendor:
+    """Update the fields of a vendor object."""
+    vendor = await get_by_id(db, vendor_id)
+    for k, v in data.items():
+        setattr(vendor, k, v)
+    await db.flush()
+    return vendor
+
+
+async def delete(db: AsyncSession, vendor_id: int) -> None:
+    """Delete a vendor object."""
+    vendor = await get_by_id(db, vendor_id)
+    await db.delete(vendor)