153 lines
4.8 KiB
Python
153 lines
4.8 KiB
Python
from app.model.crud import project_type_crud as ptc
|
|
from app.model.crud import project_label_crud as plc
|
|
from app.model.crud import project_info_crud as pic
|
|
from app.model.crud import project_image_crud as pimc
|
|
from app.service import project_service as ps
|
|
from app.model.schemas.project_info_schemas import ProjectInfoIn, ProjectInfoPager
|
|
from app.model.schemas.project_label_schemas import ProjectLabel
|
|
from app.model.schemas.project_image_schemas import ProjectImgLeaferLabel
|
|
from app.model.bussiness_model import ProjectLabel as pl
|
|
from app.db.db_session import get_db
|
|
from app.common.jwt_check import get_user_id
|
|
from app.common import reponse_code as rc
|
|
|
|
from typing import List
|
|
from fastapi import APIRouter, Depends, Request, UploadFile, File, Form
|
|
from sqlalchemy.orm import Session
|
|
|
|
"""项目管理API"""
|
|
project = APIRouter()
|
|
|
|
|
|
@project.get("/types")
|
|
def get_type_list(session: Session = Depends(get_db)):
|
|
"""获取项目类别"""
|
|
type_list = ptc.get_list(session)
|
|
return rc.response_success(data=type_list)
|
|
|
|
|
|
@project.post("/list")
|
|
def project_pager(info: ProjectInfoPager, session: Session = Depends(get_db)):
|
|
pager = pic.get_project_pager(info, session)
|
|
return rc.response_success_pager(pager)
|
|
|
|
|
|
@project.post("/add")
|
|
def add_project(request: Request, info: ProjectInfoIn, session: Session = Depends(get_db)):
|
|
"""新建项目"""
|
|
if pic.check_project_name(info.project_name, session):
|
|
return rc.response_error("已经存在相同名称的项目")
|
|
user_id = get_user_id(request)
|
|
ps.add_project(info, session, user_id)
|
|
return rc.response_success(msg="新建成功")
|
|
|
|
|
|
@project.get("/label_list/{project_id}")
|
|
def get_label_list(project_id: int, session: Session = Depends(get_db)):
|
|
"""
|
|
根据项目id查询项目标签列表
|
|
:param project_id:
|
|
:param session:
|
|
:return:
|
|
"""
|
|
label_list = plc.get_label_list(project_id, session)
|
|
return rc.response_success(msg="查询成功", data=label_list)
|
|
|
|
|
|
@project.post("/add_label")
|
|
def add_label(label: ProjectLabel, session: Session = Depends(get_db)):
|
|
"""
|
|
新增标签
|
|
:param label:
|
|
:param session:
|
|
:return:
|
|
"""
|
|
if plc.check_label_name(label.project_id, label.label_name, session):
|
|
return rc.response_error("标签名称已经存在，不能重复")
|
|
label_save = pl(**label.dict())
|
|
plc.add_label(label_save, session)
|
|
return rc.response_success(msg="保存成功")
|
|
|
|
|
|
@project.post("/up_label")
|
|
def up_label(label: ProjectLabel, session: Session = Depends(get_db)):
|
|
"""
|
|
修改标签
|
|
:param label:
|
|
:param session:
|
|
:return:
|
|
"""
|
|
if plc.check_label_name(label.project_id, label.label_name, session, label.id):
|
|
return rc.response_error("修改的标签名称已经存在，不能重复")
|
|
label_save = pl(**label.dict())
|
|
plc.update_label(label_save, session)
|
|
return rc.response_success(msg="修改成功")
|
|
|
|
|
|
@project.post("/del_label/{label_id}")
|
|
def del_label(label_id: int, session: Session = Depends(get_db)):
|
|
"""
|
|
删除标签
|
|
:param label_id:
|
|
:param session:
|
|
:return:
|
|
"""
|
|
row_del = plc.del_label(label_id, session)
|
|
if row_del > 0:
|
|
return rc.response_success(msg="删除成功")
|
|
else:
|
|
return rc.response_error("删除失败")
|
|
|
|
|
|
@project.post("/up_proj_img")
|
|
def upload_project_image(project_id: int = Form(...), files: List[UploadFile] = File(...), session: Session = Depends(get_db)):
|
|
"""
|
|
上传项目图片
|
|
:param files: 文件图片
|
|
:param project_id:
|
|
:param session:
|
|
:return:
|
|
"""
|
|
project_info = pic.get_project_by_id(project_id, session)
|
|
if project_info is None:
|
|
return rc.response_error("项目查询错误，请刷新页面后再试")
|
|
ps.upload_project_image(project_info, files, session)
|
|
return rc.response_success(msg="上传成功")
|
|
|
|
|
|
@project.get("/img_list/{project_id}")
|
|
def get_image_list(project_id: int, session: Session = Depends(get_db)):
|
|
"""
|
|
获取项目图片列表
|
|
:param project_id: 项目id
|
|
:param session:
|
|
:return:
|
|
"""
|
|
image_list = pimc.get_image_list(project_id, session)
|
|
return rc.response_success(data=image_list)
|
|
|
|
|
|
@project.post("/save_img_label")
|
|
def save_img_label(img_leafer_label: ProjectImgLeaferLabel, session: Session = Depends(get_db)):
|
|
"""
|
|
保存图片的标签框选信息
|
|
:param img_leafer_label:
|
|
:param session:
|
|
:return:
|
|
"""
|
|
ps.save_img_label(img_leafer_label, session)
|
|
return rc.response_success(msg="保存成功")
|
|
|
|
|
|
@project.get("/get_img_leafer/{imager_id}")
|
|
def get_img_leafer(image_id: int, session: Session = Depends(get_db)):
|
|
"""
|
|
根据图片id查询图片的leafer信息
|
|
:param image_id:
|
|
:param session:
|
|
:return:
|
|
"""
|
|
img_leafer_out = ps.get_img_leafer(image_id, session)
|
|
return rc.response_success(data=img_leafer_out)
|
|
|