2026-moving-helper/app/main.py

from contextlib import asynccontextmanager

from fastapi import Depends, FastAPI, Form, HTTPException, Request, status
from fastapi.responses import RedirectResponse
from fastapi.staticfiles import StaticFiles
from fastapi.templating import Jinja2Templates
from sqlalchemy.orm import Session

from app.db import get_db, init_db
from app.models import Box, Item, SubItem

templates = Jinja2Templates(directory="app/templates")


def _clean_text(value: str | None) -> str | None:
    if value is None:
        return None
    cleaned = value.strip()
    return cleaned or None


def _parse_quantity(value: str | None) -> int | None:
    cleaned = _clean_text(value)
    if cleaned is None:
        return None
    return int(cleaned)


def _is_checked(value: str | None) -> bool:
    return value == "on"


def _get_box_or_404(db: Session, box_id: int) -> Box:
    box = db.get(Box, box_id)
    if box is None:
        raise HTTPException(status_code=404, detail="Box not found")
    return box


def _get_item_or_404(db: Session, item_id: int) -> Item:
    item = db.get(Item, item_id)
    if item is None:
        raise HTTPException(status_code=404, detail="Item not found")
    return item


def _get_subitem_or_404(db: Session, subitem_id: int) -> SubItem:
    subitem = db.get(SubItem, subitem_id)
    if subitem is None:
        raise HTTPException(status_code=404, detail="Sub-item not found")
    return subitem


def _require_container_item(item: Item) -> None:
    if not item.is_container:
        raise HTTPException(status_code=400, detail="Only container items can have sub-items")


def create_app() -> FastAPI:
    @asynccontextmanager
    async def lifespan(app: FastAPI):
        init_db()
        yield

    app = FastAPI(title="搬家助手", lifespan=lifespan)
    app.mount("/static", StaticFiles(directory="app/static"), name="static")

    @app.get("/", include_in_schema=False)
    def root() -> RedirectResponse:
        return RedirectResponse(url="/boxes", status_code=status.HTTP_302_FOUND)

    @app.get("/boxes")
    def list_boxes(request: Request, db: Session = Depends(get_db)):
        boxes = db.query(Box).order_by(Box.id.desc()).all()
        return templates.TemplateResponse(
            request=request,
            name="boxes/index.html",
            context={"page_title": "箱子", "boxes": boxes},
        )

    @app.get("/boxes/new")
    def new_box_page(request: Request):
        return templates.TemplateResponse(
            request=request,
            name="boxes/form.html",
            context={
                "page_title": "新建箱子",
                "box": None,
                "form_action": "/boxes",
                "submit_label": "创建箱子",
            },
        )

    @app.post("/boxes")
    def create_box(
        name: str = Form(...),
        note: str | None = Form(default=None),
        room: str | None = Form(default=None),
        status_text: str | None = Form(default=None, alias="status"),
        db: Session = Depends(get_db),
    ) -> RedirectResponse:
        box = Box(
            name=name.strip(),
            note=_clean_text(note),
            room=_clean_text(room),
            status=_clean_text(status_text),
        )
        db.add(box)
        db.commit()
        db.refresh(box)
        return RedirectResponse(url=f"/boxes/{box.id}", status_code=status.HTTP_303_SEE_OTHER)

    @app.get("/boxes/{box_id}")
    def show_box(box_id: int, request: Request, db: Session = Depends(get_db)):
        box = _get_box_or_404(db, box_id)
        return templates.TemplateResponse(
            request=request,
            name="boxes/show.html",
            context={"page_title": box.name, "box": box},
        )

    @app.get("/boxes/{box_id}/edit")
    def edit_box_page(box_id: int, request: Request, db: Session = Depends(get_db)):
        box = _get_box_or_404(db, box_id)
        return templates.TemplateResponse(
            request=request,
            name="boxes/form.html",
            context={
                "page_title": f"编辑箱子：{box.name}",
                "box": box,
                "form_action": f"/boxes/{box.id}/update",
                "submit_label": "保存箱子",
            },
        )

    @app.post("/boxes/{box_id}/update")
    def update_box(
        box_id: int,
        name: str = Form(...),
        note: str | None = Form(default=None),
        room: str | None = Form(default=None),
        status_text: str | None = Form(default=None, alias="status"),
        db: Session = Depends(get_db),
    ) -> RedirectResponse:
        box = _get_box_or_404(db, box_id)
        box.name = name.strip()
        box.note = _clean_text(note)
        box.room = _clean_text(room)
        box.status = _clean_text(status_text)
        db.commit()
        return RedirectResponse(url=f"/boxes/{box.id}", status_code=status.HTTP_303_SEE_OTHER)

    @app.post("/boxes/{box_id}/delete")
    def delete_box(box_id: int, db: Session = Depends(get_db)) -> RedirectResponse:
        box = _get_box_or_404(db, box_id)
        db.delete(box)
        db.commit()
        return RedirectResponse(url="/boxes", status_code=status.HTTP_303_SEE_OTHER)

    @app.get("/boxes/{box_id}/items/new")
    def new_item_page(box_id: int, request: Request, db: Session = Depends(get_db)):
        box = _get_box_or_404(db, box_id)
        return templates.TemplateResponse(
            request=request,
            name="items/form.html",
            context={
                "page_title": f"为箱子“{box.name}”添加物品",
                "box": box,
                "item": None,
                "form_action": f"/boxes/{box.id}/items",
                "submit_label": "创建物品",
            },
        )

    @app.post("/boxes/{box_id}/items")
    def create_item(
        box_id: int,
        name: str = Form(...),
        note: str | None = Form(default=None),
        quantity: str | None = Form(default=None),
        is_container: str | None = Form(default=None),
        db: Session = Depends(get_db),
    ) -> RedirectResponse:
        box = _get_box_or_404(db, box_id)
        item = Item(
            box=box,
            name=name.strip(),
            note=_clean_text(note),
            quantity=_parse_quantity(quantity),
            is_container=_is_checked(is_container),
        )
        db.add(item)
        db.commit()
        db.refresh(item)
        return RedirectResponse(url=f"/items/{item.id}", 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)):
        item = _get_item_or_404(db, item_id)
        return templates.TemplateResponse(
            request=request,
            name="items/show.html",
            context={"page_title": item.name, "item": item},
        )

    @app.get("/items/{item_id}/edit")
    def edit_item_page(item_id: int, request: Request, db: Session = Depends(get_db)):
        item = _get_item_or_404(db, item_id)
        return templates.TemplateResponse(
            request=request,
            name="items/form.html",
            context={
                "page_title": f"编辑物品：{item.name}",
                "box": item.box,
                "item": item,
                "form_action": f"/items/{item.id}/update",
                "submit_label": "保存物品",
            },
        )

    @app.post("/items/{item_id}/update")
    def update_item(
        item_id: int,
        name: str = Form(...),
        note: str | None = Form(default=None),
        quantity: str | None = Form(default=None),
        is_container: str | None = Form(default=None),
        db: Session = Depends(get_db),
    ) -> RedirectResponse:
        item = _get_item_or_404(db, item_id)
        item.name = name.strip()
        item.note = _clean_text(note)
        item.quantity = _parse_quantity(quantity)
        item.is_container = _is_checked(is_container)
        if not item.is_container:
            item.subitems.clear()
        db.commit()
        return RedirectResponse(url=f"/items/{item.id}", status_code=status.HTTP_303_SEE_OTHER)

    @app.post("/items/{item_id}/delete")
    def delete_item(item_id: int, db: Session = Depends(get_db)) -> RedirectResponse:
        item = _get_item_or_404(db, item_id)
        box_id = item.box_id
        db.delete(item)
        db.commit()
        return RedirectResponse(url=f"/boxes/{box_id}", status_code=status.HTTP_303_SEE_OTHER)

    @app.get("/items/{item_id}/subitems/new")
    def new_subitem_page(item_id: int, request: Request, db: Session = Depends(get_db)):
        item = _get_item_or_404(db, item_id)
        _require_container_item(item)
        return templates.TemplateResponse(
            request=request,
            name="subitems/form.html",
            context={
                "page_title": f"为“{item.name}”添加子物品",
                "item": item,
                "subitem": None,
                "form_action": f"/items/{item.id}/subitems",
                "submit_label": "创建子物品",
            },
        )

    @app.post("/items/{item_id}/subitems")
    def create_subitem(
        item_id: int,
        name: str = Form(...),
        note: str | None = Form(default=None),
        quantity: str | None = Form(default=None),
        db: Session = Depends(get_db),
    ) -> RedirectResponse:
        item = _get_item_or_404(db, item_id)
        _require_container_item(item)
        subitem = SubItem(
            parent_item=item,
            name=name.strip(),
            note=_clean_text(note),
            quantity=_parse_quantity(quantity),
        )
        db.add(subitem)
        db.commit()
        db.refresh(subitem)
        return RedirectResponse(url=f"/items/{item.id}", status_code=status.HTTP_303_SEE_OTHER)

    @app.get("/subitems/{subitem_id}/edit")
    def edit_subitem_page(subitem_id: int, request: Request, db: Session = Depends(get_db)):
        subitem = _get_subitem_or_404(db, subitem_id)
        return templates.TemplateResponse(
            request=request,
            name="subitems/form.html",
            context={
                "page_title": f"编辑子物品：{subitem.name}",
                "item": subitem.parent_item,
                "subitem": subitem,
                "form_action": f"/subitems/{subitem.id}/update",
                "submit_label": "保存子物品",
            },
        )

    @app.post("/subitems/{subitem_id}/update")
    def update_subitem(
        subitem_id: int,
        name: str = Form(...),
        note: str | None = Form(default=None),
        quantity: str | None = Form(default=None),
        db: Session = Depends(get_db),
    ) -> RedirectResponse:
        subitem = _get_subitem_or_404(db, subitem_id)
        subitem.name = name.strip()
        subitem.note = _clean_text(note)
        subitem.quantity = _parse_quantity(quantity)
        db.commit()
        return RedirectResponse(
            url=f"/items/{subitem.parent_item_id}",
            status_code=status.HTTP_303_SEE_OTHER,
        )

    @app.post("/subitems/{subitem_id}/delete")
    def delete_subitem(subitem_id: int, db: Session = Depends(get_db)) -> RedirectResponse:
        subitem = _get_subitem_or_404(db, subitem_id)
        item_id = subitem.parent_item_id
        db.delete(subitem)
        db.commit()
        return RedirectResponse(url=f"/items/{item_id}", status_code=status.HTTP_303_SEE_OTHER)

    return app


app = create_app()