add display for WorkType

kontor-api/src/webapps/comic/route_comics.py

@@ -1,9 +1,13 @@
-from fastapi import APIRouter, Request
+from fastapi import APIRouter, Request, responses, status
 from fastapi.templating import Jinja2Templates
 from src.apis.utils import SessionDep
-from src.db.models.comic import Comic, Artist, Publisher, Issue
+from src.db.models.comic import Comic, Artist, Publisher, Issue, WorkType
 from typing import AnyStr
 
+from src.db.repository.comic import create_new_worktype
+from src.schema.comics.worktype import AddWorkType
+from src.webapps.comic.forms import AddWorktypeForm
+
 templates = Jinja2Templates(directory="src/templates")
 router = APIRouter(include_in_schema=False, prefix="/comic")
 
@@ -48,3 +52,33 @@ def get_issues(db: SessionDep, request: Request, msg: str = None):
 def issue_details(issue_id: AnyStr, request: Request, db: SessionDep):
     issue = db.get(Issue, issue_id)
     return templates.TemplateResponse("comic/issue_detail.html", {"request": request, "issue": issue})
+
+@router.get("/worktypes")
+def get_worktypes(db: SessionDep, request: Request, msg: str = None):
+    worktypes = db.query(WorkType).all()
+    return templates.TemplateResponse("comic/worktypes.html", {"request": request, "msg": msg, "worktypes": worktypes})
+
+@router.get("/worktypes/{worktype_id}")
+def worktype_detail(worktype_id: AnyStr, request: Request, db: SessionDep):
+    worktype = db.get(WorkType, worktype_id)
+    return templates.TemplateResponse("comic/worktype_detail.html", {"request": request, "worktype": worktype})
+
+@router.get("/add-worktype")
+def add_worktype(request: Request, db: SessionDep):
+    return templates.TemplateResponse("comic/add_worktype.html", {"request": request})
+
+@router.post("/add-worktype")
+async def post_worktype(request: Request, db: SessionDep):
+    form = AddWorktypeForm(request)
+    await form.load_data()
+    if form.is_valid():
+        try:
+            work = AddWorkType(**form.__dict__)
+            worktype = create_new_worktype(work=work, db=db)
+            return responses.RedirectResponse(f"/comic/worktypes/{worktype.id}", status_code=status.HTTP_302_FOUND)
+        except Exception as e:
+            print(e)
+            form.__dict__.get("errors").append("worktype already added")
+            return templates.TemplateResponse("comic/add_worktype.html", form.__dict__)
+    return templates.TemplateResponse("comic/add_worktype.html", form.__dict__)
+