add CRUD for WorkType

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

@@ -1,10 +1,13 @@
 from fastapi import APIRouter, Request, responses, status
 from fastapi.templating import Jinja2Templates
+from fastapi.responses import RedirectResponse
+
 from src.apis.utils import SessionDep
 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.db.repository.comic import create_new_worktype, update_worktype
+from src.main import logger
 from src.schema.comics.worktype import AddWorkType
 from src.webapps.comic.forms import AddWorktypeForm
 
@@ -59,26 +62,54 @@ def get_worktypes(db: SessionDep, request: Request, msg: str = None):
     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):
+def worktype_detail(db: SessionDep, request: Request, worktype_id: AnyStr):
     worktype = db.get(WorkType, worktype_id)
     return templates.TemplateResponse("comic/worktype_detail.html", {"request": request, "worktype": worktype})
 
-@router.get("/add-worktype")
+@router.get("/worktype/add")
 def add_worktype(request: Request, db: SessionDep):
-    return templates.TemplateResponse("comic/add_worktype.html", {"request": request})
+    return templates.TemplateResponse("comic/worktype_edit.html", {"request": request})
 
-@router.post("/add-worktype")
-async def post_worktype(request: Request, db: SessionDep):
+@router.post("/worktype/add")
+async def add_worktype(db: SessionDep, request: Request):
     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)
+            logger.info(f"add_worktype: redirect to /comic/worktypes/{worktype.id}")
+            return RedirectResponse(f"/comic/worktypes/{worktype.id}", status_code=status.HTTP_303_SEE_OTHER)
         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__)
+            return templates.TemplateResponse("comic/worktype_edit.html", form.__dict__)
+    print("form is not valid")
+    return templates.TemplateResponse("comic/worktype_edit.html", form.__dict__)
 
+@router.get("/worktype/edit/{worktype_id}")
+def edit_worktype(db: SessionDep, request: Request, worktype_id: str):
+    worktype = db.get(WorkType, worktype_id)
+    return templates.TemplateResponse("comic/worktype_edit.html", {"request": request, "worktype": worktype.name})
+
+@router.post("/worktype/edit/{worktype_id}")
+async def edit_worktype(request: Request, db: SessionDep, worktype_id: str):
+    form = AddWorktypeForm(request)
+    await form.load_data()
+    if form.is_valid():
+        try:
+            work = AddWorkType(**form.__dict__)
+            worktype = update_worktype(work=work, worktype_id=worktype_id, db=db)
+            return RedirectResponse(f"/comic/worktypes/{worktype.id}", status_code=status.HTTP_303_SEE_OTHER)
+        except Exception as e:
+            print(e)
+            form.__dict__.get("errors").append("worktype already added")
+            return templates.TemplateResponse("comic/worktype_edit.html", form.__dict__)
+    return templates.TemplateResponse("comic/worktype_edit.html", form.__dict__)
+
+@router.get("/worktype/delete/{worktype_id}")
+async def delete_worktype(db: SessionDep, request: Request, worktype_id: str):
+    worktype = db.get(WorkType, worktype_id)
+    db.delete(worktype)
+    db.commit()
+    return RedirectResponse("/comic/worktypes", status_code=status.HTTP_303_SEE_OTHER)