"""内容提交路由""" from fastapi import APIRouter, Depends, HTTPException, Query from sqlalchemy.orm import Session from typing import List, Optional from datetime import date from database import get_db from models import ( User, Submission, SubmissionHistory, Project, UserRole, PhaseGroup, WorkType, ContentType, SubmitTo ) from schemas import SubmissionCreate, SubmissionUpdate, SubmissionOut from auth import get_current_user, require_role router = APIRouter(prefix="/api/submissions", tags=["内容提交"]) def submission_to_out(s: Submission) -> SubmissionOut: return SubmissionOut( id=s.id, user_id=s.user_id, user_name=s.user.name if s.user else None, project_id=s.project_id, project_name=s.project.name if s.project else None, project_phase=s.project_phase.value if hasattr(s.project_phase, 'value') else s.project_phase, work_type=s.work_type.value if hasattr(s.work_type, 'value') else s.work_type, content_type=s.content_type.value if hasattr(s.content_type, 'value') else s.content_type, duration_minutes=s.duration_minutes, duration_seconds=s.duration_seconds, total_seconds=s.total_seconds, hours_spent=s.hours_spent, submit_to=s.submit_to.value if hasattr(s.submit_to, 'value') else s.submit_to, description=s.description, submit_date=s.submit_date, created_at=s.created_at, ) @router.get("/", response_model=List[SubmissionOut]) def list_submissions( project_id: Optional[int] = Query(None), user_id: Optional[int] = Query(None), start_date: Optional[date] = Query(None), end_date: Optional[date] = Query(None), db: Session = Depends(get_db), current_user: User = Depends(get_current_user) ): q = db.query(Submission) # 成员只能看自己的 if current_user.role == UserRole.MEMBER: q = q.filter(Submission.user_id == current_user.id) elif user_id: q = q.filter(Submission.user_id == user_id) if project_id: q = q.filter(Submission.project_id == project_id) if start_date: q = q.filter(Submission.submit_date >= start_date) if end_date: q = q.filter(Submission.submit_date <= end_date) subs = q.order_by(Submission.submit_date.desc(), Submission.created_at.desc()).all() return [submission_to_out(s) for s in subs] @router.post("/", response_model=SubmissionOut) def create_submission( req: SubmissionCreate, db: Session = Depends(get_db), current_user: User = Depends(get_current_user) ): # 校验项目存在 project = db.query(Project).filter(Project.id == req.project_id).first() if not project: raise HTTPException(status_code=404, detail="项目不存在") # 自动计算总秒数 total_seconds = (req.duration_minutes or 0) * 60 + (req.duration_seconds or 0) sub = Submission( user_id=current_user.id, project_id=req.project_id, project_phase=PhaseGroup(req.project_phase), work_type=WorkType(req.work_type), content_type=ContentType(req.content_type), duration_minutes=req.duration_minutes or 0, duration_seconds=req.duration_seconds or 0, total_seconds=total_seconds, hours_spent=req.hours_spent, submit_to=SubmitTo(req.submit_to), description=req.description, submit_date=req.submit_date, ) db.add(sub) db.commit() db.refresh(sub) return submission_to_out(sub) @router.put("/{submission_id}", response_model=SubmissionOut) def update_submission( submission_id: int, req: SubmissionUpdate, db: Session = Depends(get_db), current_user: User = Depends(require_role(UserRole.OWNER, UserRole.SUPERVISOR, UserRole.LEADER)) ): """高权限修改提交记录（需填写原因）""" sub = db.query(Submission).filter(Submission.id == submission_id).first() if not sub: raise HTTPException(status_code=404, detail="提交记录不存在") # 保存旧数据用于历史记录 old_data = { "project_phase": sub.project_phase.value if hasattr(sub.project_phase, 'value') else sub.project_phase, "work_type": sub.work_type.value if hasattr(sub.work_type, 'value') else sub.work_type, "content_type": sub.content_type.value if hasattr(sub.content_type, 'value') else sub.content_type, "duration_minutes": sub.duration_minutes, "duration_seconds": sub.duration_seconds, "total_seconds": sub.total_seconds, "hours_spent": sub.hours_spent, "submit_to": sub.submit_to.value if hasattr(sub.submit_to, 'value') else sub.submit_to, "description": sub.description, "submit_date": str(sub.submit_date), } # 更新字段 if req.project_phase is not None: sub.project_phase = PhaseGroup(req.project_phase) if req.work_type is not None: sub.work_type = WorkType(req.work_type) if req.content_type is not None: sub.content_type = ContentType(req.content_type) if req.duration_minutes is not None: sub.duration_minutes = req.duration_minutes if req.duration_seconds is not None: sub.duration_seconds = req.duration_seconds if req.hours_spent is not None: sub.hours_spent = req.hours_spent if req.submit_to is not None: sub.submit_to = SubmitTo(req.submit_to) if req.description is not None: sub.description = req.description if req.submit_date is not None: sub.submit_date = req.submit_date # 重算总秒数 sub.total_seconds = (sub.duration_minutes or 0) * 60 + (sub.duration_seconds or 0) # 保存新数据 new_data = { "project_phase": sub.project_phase.value if hasattr(sub.project_phase, 'value') else sub.project_phase, "work_type": sub.work_type.value if hasattr(sub.work_type, 'value') else sub.work_type, "content_type": sub.content_type.value if hasattr(sub.content_type, 'value') else sub.content_type, "duration_minutes": sub.duration_minutes, "duration_seconds": sub.duration_seconds, "total_seconds": sub.total_seconds, "hours_spent": sub.hours_spent, "submit_to": sub.submit_to.value if hasattr(sub.submit_to, 'value') else sub.submit_to, "description": sub.description, "submit_date": str(sub.submit_date), } # 写入修改历史 history = SubmissionHistory( submission_id=sub.id, changed_by=current_user.id, change_reason=req.change_reason, old_data=old_data, new_data=new_data, ) db.add(history) db.commit() db.refresh(sub) return submission_to_out(sub) @router.get("/{submission_id}/history") def get_submission_history( submission_id: int, db: Session = Depends(get_db), current_user: User = Depends(require_role(UserRole.OWNER, UserRole.SUPERVISOR, UserRole.LEADER)) ): """查看提交的修改历史""" records = db.query(SubmissionHistory).filter( SubmissionHistory.submission_id == submission_id ).order_by(SubmissionHistory.created_at.desc()).all() return [ { "id": r.id, "changed_by": r.changed_by, "change_reason": r.change_reason, "old_data": r.old_data, "new_data": r.new_data, "created_at": r.created_at, } for r in records ]