ux refine

app/main.py

@@ -82,6 +82,10 @@ def _image_response_or_404(target) -> Response:
     return Response(content=target.image_blob, media_type=target.image_mime_type)
 
 
+def _wants_add_next(submit_action: str | None) -> bool:
+    return submit_action == "save_and_add_next"
+
+
 def _build_search_results(db: Session, query: str) -> list[dict]:
     keyword = f"%{query.lower()}%"
     results: list[dict] = []
@@ -333,6 +337,7 @@ def create_app() -> FastAPI:
         note: str | None = Form(default=None),
         quantity: str | None = Form(default=None),
         is_container: str | None = Form(default=None),
+        submit_action: str | None = Form(default=None),
         image_file: UploadFile | None = File(default=None),
         db: Session = Depends(get_db),
     ) -> RedirectResponse:
@@ -348,7 +353,13 @@ def create_app() -> FastAPI:
         db.add(item)
         db.commit()
         db.refresh(item)
-        return RedirectResponse(url=f"/items/{item.id}", status_code=status.HTTP_303_SEE_OTHER)
+        if _wants_add_next(submit_action):
+            redirect_url = f"/boxes/{box.id}/items/new"
+        elif item.is_container:
+            redirect_url = f"/items/{item.id}"
+        else:
+            redirect_url = f"/boxes/{box.id}"
+        return RedirectResponse(url=redirect_url, status_code=status.HTTP_303_SEE_OTHER)
 
     @app.get("/items/{item_id}")
     def show_item(item_id: int, request: Request, db: Session = Depends(get_db)):
@@ -436,6 +447,7 @@ def create_app() -> FastAPI:
         name: str = Form(...),
         note: str | None = Form(default=None),
         quantity: str | None = Form(default=None),
+        submit_action: str | None = Form(default=None),
         image_file: UploadFile | None = File(default=None),
         db: Session = Depends(get_db),
     ) -> RedirectResponse:
@@ -451,7 +463,11 @@ def create_app() -> FastAPI:
         db.add(subitem)
         db.commit()
         db.refresh(subitem)
-        return RedirectResponse(url=f"/items/{item.id}", status_code=status.HTTP_303_SEE_OTHER)
+        if _wants_add_next(submit_action):
+            redirect_url = f"/items/{item.id}/subitems/new"
+        else:
+            redirect_url = f"/items/{item.id}"
+        return RedirectResponse(url=redirect_url, status_code=status.HTTP_303_SEE_OTHER)
 
     @app.get("/subitems/{subitem_id}/image")
     def get_subitem_image(subitem_id: int, db: Session = Depends(get_db)) -> Response: