sun15684712291/aicheckv2
1
1
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

完善推理模块的主体功能

Browse Source
This commit is contained in:
sunyg
2025-03-05 16:16:12 +08:00
parent 5f4ed473d7
commit 05e2e8a59a
3 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
app/api/business/project_detect_api.py
View File

@ -34,8 +34,8 @@ def add_detect(detect_in: ProjectDetectIn, session: Session = Depends(get_db)):
:param session: :param session:
:return: :return:
""" """
pds.add_detect(detect_in, session) detect = pds.add_detect(detect_in, session)
return rc.response_success("新增成功") return rc.response_success(msg="新增成功", data=detect.id)
@detect.post("/get_img_list") @detect.post("/get_img_list")
@ -89,7 +89,7 @@ def del_detect_img(detect_img_id: int, session: Session = Depends(get_db)):
return rc.response_error(msg="删除失败") return rc.response_error(msg="删除失败")
@detect.get("/run_detect_yolo") @detect.post("/run_detect_yolo")
def run_detect_yolo(detect_log_in: ProjectDetectLogIn, session: Session = Depends(get_db)): def run_detect_yolo(detect_log_in: ProjectDetectLogIn, session: Session = Depends(get_db)):
""" """
开始执行训练 开始执行训练

2
app/api/sys/login_api.py
View File

@ -14,7 +14,7 @@ from app.common.redis_cli import redis_conn
login = APIRouter() login = APIRouter()
@login.post("/") @login.post("")
def login_check(user: SysUserLogin, session: Session = Depends(get_db)): def login_check(user: SysUserLogin, session: Session = Depends(get_db)):
""" """
登录验证，并返回token 登录验证，并返回token

2
app/application/token_middleware.py
View File

@ -5,7 +5,7 @@ from jwt import PyJWTError
from app.common import reponse_code as rc from app.common import reponse_code as rc
from app.common import jwt_check as jc from app.common import jwt_check as jc
green = ['/login', '/view_img', 'view_thumb', 'view_detect_img', 'view_detect_thumb', 'view_log_img'] green = ['login', 'view_img', 'view_thumb', 'view_detect_img', 'view_detect_thumb', 'view_log_img']
def check_green(s: str): def check_green(s: str):