116 lines
4.6 KiB
Python
116 lines
4.6 KiB
Python
|
from fastapi import APIRouter, Depends, Query, HTTPException
|
||
|
|
from sqlalchemy import desc
|
||
|
|
from sqlalchemy.orm import Session
|
||
|
|
from datetime import datetime
|
||
|
|
from pydantic import BaseModel
|
||
|
|
from app.core.database import get_db
|
||
|
|
from app.models.requirement import Requirement
|
||
|
|
from app.models.project import Project
|
||
|
|
from app.utils.common_util import Success, SuccessListPage, get_total_page, get_offset
|
||
|
|
from app.core.security import get_current_user
|
||
|
|
|
||
|
|
|
||
|
|
class RequirementEdit(BaseModel):
|
||
|
|
id: int = 0
|
||
|
|
title: str
|
||
|
|
description: str
|
||
|
|
project_id: int
|
||
|
|
|
||
|
|
|
||
|
|
router = APIRouter(prefix="/requirement", tags=["需求管理"])
|
||
|
|
|
||
|
|
|
||
|
|
@router.post("/save", description="新增需求")
|
||
|
|
async def save_requirement(requirement: RequirementEdit,
|
||
|
|
db: Session = Depends(get_db)):
|
||
|
|
try:
|
||
|
|
params = Requirement(
|
||
|
|
title=requirement.title,
|
||
|
|
description=requirement.description,
|
||
|
|
project_id=requirement.project_id,
|
||
|
|
gmt_create=datetime.now().strftime("%Y-%m-%d %H:%M:%S")
|
||
|
|
)
|
||
|
|
db.add(params)
|
||
|
|
db.commit()
|
||
|
|
except Exception as e:
|
||
|
|
db.rollback()
|
||
|
|
raise HTTPException(status_code=400, detail=str(e))
|
||
|
|
return Success("新增成功")
|
||
|
|
|
||
|
|
|
||
|
|
@router.delete("/delete", description="删除需求")
|
||
|
|
async def delete_requirement(ids: str = Query(..., description="删除的ID列表", min_length=1),
|
||
|
|
current_user: str = Depends(get_current_user),
|
||
|
|
db: Session = Depends(get_db)):
|
||
|
|
try:
|
||
|
|
id_list = ids.split(",")
|
||
|
|
db.query(Requirement).filter(Requirement.id.in_(id_list)).delete()
|
||
|
|
db.commit()
|
||
|
|
except Exception as e:
|
||
|
|
db.rollback()
|
||
|
|
raise HTTPException(status_code=400, detail=str(e))
|
||
|
|
return Success("删除成功")
|
||
|
|
|
||
|
|
|
||
|
|
@router.put("/update/{requirement_id}", description="修改需求")
|
||
|
|
async def update_requirement(requirement_id: int,
|
||
|
|
requirement: RequirementEdit,
|
||
|
|
current_user: str = Depends(get_current_user),
|
||
|
|
db: Session = Depends(get_db)):
|
||
|
|
try:
|
||
|
|
params = db.query(Requirement).filter(Requirement.id == requirement_id).first()
|
||
|
|
if not params:
|
||
|
|
raise HTTPException(status_code=404, detail="数据不存在")
|
||
|
|
params.title = requirement.title
|
||
|
|
params.description = requirement.description
|
||
|
|
params.project_id = requirement.project_id
|
||
|
|
db.commit()
|
||
|
|
except Exception as e:
|
||
|
|
db.rollback()
|
||
|
|
raise HTTPException(status_code=400, detail=str(e))
|
||
|
|
return Success("修改成功")
|
||
|
|
|
||
|
|
|
||
|
|
@router.get("/get/{requirement_id}", description="获取需求详情")
|
||
|
|
async def get_requirement(requirement_id: int,
|
||
|
|
current_user: str = Depends(get_current_user),
|
||
|
|
db: Session = Depends(get_db)):
|
||
|
|
data = db.query(Requirement).filter(Requirement.id == requirement_id).first()
|
||
|
|
if not data:
|
||
|
|
raise HTTPException(status_code=404, detail="数据不存在")
|
||
|
|
return data
|
||
|
|
|
||
|
|
|
||
|
|
@router.get("/list", description="需求列表")
|
||
|
|
async def list_requirement(db: Session = Depends(get_db),
|
||
|
|
current_user: str = Depends(get_current_user), ):
|
||
|
|
return db.query(Requirement).order_by(desc(Requirement.gmt_create)).all()
|
||
|
|
|
||
|
|
|
||
|
|
@router.get("/listpage", description="需求分页列表")
|
||
|
|
async def listpage_requirement(db: Session = Depends(get_db),
|
||
|
|
current_user: str = Depends(get_current_user),
|
||
|
|
page: int = Query(1, ge=1, description="当前页面"),
|
||
|
|
page_size: int = Query(10, ge=10, description="每页条数"),
|
||
|
|
title: str = Query(None, description="标题"),
|
||
|
|
proj_id: str = Query(None, description="项目ID")):
|
||
|
|
query = db.query(Requirement)
|
||
|
|
if title:
|
||
|
|
query = query.filter(Requirement.title.ilike(f"%{title}%"))
|
||
|
|
if proj_id:
|
||
|
|
query = query.filter(Requirement.project_id == proj_id)
|
||
|
|
total = query.count()
|
||
|
|
offset = get_offset(page, page_size)
|
||
|
|
total_page = get_total_page(page, page_size)
|
||
|
|
data_list = query.offset(offset).limit(page_size).all()
|
||
|
|
project_ids = list([])
|
||
|
|
for data in data_list:
|
||
|
|
project_ids.append(data.project_id)
|
||
|
|
if len(project_ids) > 0:
|
||
|
|
project_list = db.query(Project).filter(Project.id.in_(project_ids)).all()
|
||
|
|
for project in project_list:
|
||
|
|
for data in data_list:
|
||
|
|
if project.id == data.project_id:
|
||
|
|
data.project_title = project.title
|
||
|
|
return SuccessListPage(page, total, total_page, data_list)
|