重构基本完成

app/model/crud/project_info_crud.py

@@ -1,5 +1,5 @@
 from sqlalchemy.orm import Session
-from sqlalchemy import desc
+from sqlalchemy import desc, update
 
 from app.model.bussiness_model import ProjectInfo
 from app.model.schemas.project_info_schemas import ProjectInfoOut
@@ -41,3 +41,24 @@ def check_project_name(project_name: str, session: Session):
     else:
         return False
 
+
+def update_project_status(project_id: int, project_status: str, session: Session):
+    """
+    更新项目训练状态,如果是已完成的话，train_version自动+1
+    :param project_id:
+    :param project_status: 0-未运行，1-运行中，2-已完成,-1-执行失败
+    :param session:
+    :return:
+    """
+    if project_status == '2':
+        stmt = update(ProjectInfo).where(ProjectInfo.id == project_id).values({
+            'train_status': project_status,
+            'train_version': ProjectInfo.train_version + 1
+        })
+        session.execute(stmt)
+    else:
+        stmt = update(ProjectInfo).where(ProjectInfo.id == project_id).values({
+            'train_status': project_status
+        })
+        session.execute(stmt)
+    session.commit()