150 lines
5.3 KiB
Python
150 lines
5.3 KiB
Python
"""每周重点工作：单条录入、按周跟踪的短期任务。
|
||
|
||
和项目管理（Project + Update）分开：不用先建项目档案，一条就是一个活儿。
|
||
字段按需求——工作内容 / 说明 / 预计完成时间（日历日期）/ 执行状态 / 结果反馈 / 是否延期。
|
||
period_id 是「录入周」——新建时顶栏选中的那一周，和项目管理/定制项目的周期归属同一个意思；
|
||
预计完成时间只存 due_date，不再按日期反查周期。本周录一条两周后要干的事，本周照样看得见。
|
||
「是否延期」是人工开关（delayed）：更早周次录入、标了延期又没做完的，结转进当周视图。
|
||
录入整理在「每周重点工作」页，汇总在「工作看板」。
|
||
"""
|
||
from __future__ import annotations
|
||
|
||
from datetime import date
|
||
|
||
from fastapi import APIRouter, Depends, HTTPException, Query
|
||
from sqlalchemy.orm import Session
|
||
|
||
from .. import constants
|
||
from ..db import get_db
|
||
from ..models import Period, WeeklyFocus
|
||
from ..schemas import FocusIn, FocusPatch
|
||
|
||
router = APIRouter()
|
||
|
||
|
||
def _week_period(db: Session, year: int, month: int, week: int) -> Period:
|
||
"""取 periods 里的这一周，没有就补一条。"""
|
||
sort_key = year * 10000 + month * 100 + week
|
||
target = (
|
||
db.query(Period)
|
||
.filter(Period.kind == "week", Period.sort_key == sort_key)
|
||
.one_or_none()
|
||
)
|
||
if target is None:
|
||
target = Period(
|
||
label=f"{month}月第{week}周",
|
||
kind="week",
|
||
year=year,
|
||
month=month,
|
||
week=week,
|
||
sort_key=sort_key,
|
||
)
|
||
db.add(target)
|
||
db.flush()
|
||
return target
|
||
|
||
|
||
def _entry_period(db: Session, period_id: int | None) -> Period:
|
||
"""录入周：前端传顶栏当前选中的那一周；没传（脚本/API 直接调）就按今天所在的周。"""
|
||
if period_id:
|
||
existed = db.get(Period, period_id)
|
||
if existed:
|
||
return existed
|
||
raise HTTPException(400, "录入周不存在")
|
||
return _week_period(db, *constants.week_of_date(date.today()))
|
||
|
||
|
||
def _clean_date(value: str | None) -> str | None:
|
||
"""校验日历日期，存 ISO 字符串。"""
|
||
raw = (value or "").strip()
|
||
if not raw:
|
||
return None
|
||
try:
|
||
return date.fromisoformat(raw[:10]).isoformat()
|
||
except ValueError:
|
||
raise HTTPException(400, f"预计完成时间格式不对：{raw}（应为 YYYY-MM-DD）")
|
||
|
||
|
||
@router.get("/focus")
|
||
def list_focus(
|
||
archived: str = Query("false", pattern="^(true|false|all)$"),
|
||
status: str | None = None,
|
||
keyword: str | None = None,
|
||
db: Session = Depends(get_db),
|
||
):
|
||
q = db.query(WeeklyFocus)
|
||
if archived == "false":
|
||
q = q.filter(WeeklyFocus.archived.is_(False))
|
||
elif archived == "true":
|
||
q = q.filter(WeeklyFocus.archived.is_(True))
|
||
if status:
|
||
q = q.filter(WeeklyFocus.status == status)
|
||
if keyword:
|
||
q = q.filter(WeeklyFocus.content.contains(keyword) | WeeklyFocus.detail.contains(keyword))
|
||
return [f.to_dict() for f in q.order_by(WeeklyFocus.id.desc()).all()]
|
||
|
||
|
||
@router.post("/focus")
|
||
def create_focus(payload: FocusIn, db: Session = Depends(get_db)):
|
||
content = (payload.content or "").strip()
|
||
if not content:
|
||
raise HTTPException(400, "请填写工作内容")
|
||
if payload.status not in constants.STATUS_CODES:
|
||
raise HTTPException(400, f"未知状态：{payload.status}")
|
||
due = _clean_date(payload.due_date)
|
||
item = WeeklyFocus(
|
||
content=content,
|
||
detail=(payload.detail or "").strip() or None,
|
||
period_id=_entry_period(db, payload.period_id).id,
|
||
due_date=due,
|
||
result=(payload.result or "").strip() or None,
|
||
delayed=bool(payload.delayed),
|
||
status=payload.status,
|
||
)
|
||
db.add(item)
|
||
db.commit()
|
||
db.refresh(item)
|
||
return item.to_dict()
|
||
|
||
|
||
@router.patch("/focus/{fid}")
|
||
def patch_focus(fid: int, payload: FocusPatch, db: Session = Depends(get_db)):
|
||
item = db.get(WeeklyFocus, fid)
|
||
if not item:
|
||
raise HTTPException(404, "条目不存在")
|
||
data = payload.model_dump(exclude_unset=True)
|
||
if "content" in data:
|
||
content = (data["content"] or "").strip()
|
||
if not content:
|
||
raise HTTPException(400, "请填写工作内容")
|
||
data["content"] = content
|
||
if "detail" in data:
|
||
data["detail"] = (data["detail"] or "").strip() or None
|
||
if "status" in data and data["status"] not in constants.STATUS_CODES:
|
||
raise HTTPException(400, f"未知状态：{data['status']}")
|
||
if "result" in data:
|
||
data["result"] = (data["result"] or "").strip() or None
|
||
if "due_date" in data:
|
||
data["due_date"] = _clean_date(data["due_date"])
|
||
# 改预计完成时间不再把条目搬去别的周，只有显式指定录入周才动 period_id
|
||
if data.get("period_id"):
|
||
data["period_id"] = _entry_period(db, data["period_id"]).id
|
||
else:
|
||
data.pop("period_id", None)
|
||
for key, value in data.items():
|
||
setattr(item, key, value)
|
||
db.commit()
|
||
db.refresh(item)
|
||
return item.to_dict()
|
||
|
||
|
||
@router.delete("/focus/{fid}")
|
||
def delete_focus(fid: int, db: Session = Depends(get_db)):
|
||
item = db.get(WeeklyFocus, fid)
|
||
if not item:
|
||
raise HTTPException(404, "条目不存在")
|
||
content = item.content
|
||
db.delete(item)
|
||
db.commit()
|
||
return {"id": fid, "content": content}
|