重构基本完成

app/model/crud/project_train_crud.py

@@ -0,0 +1,28 @@
+from sqlalchemy.orm import Session
+from sqlalchemy import asc
+
+from app.model.bussiness_model import ProjectTrain
+from app.model.schemas.project_train_schemas import ProjectTrainOut
+
+
+def add_train(train: ProjectTrain, session: Session):
+    """
+    新增训练结果
+    :param train:
+    :param session:
+    :return:
+    """
+    session.add(train)
+    session.commit()
+
+
+def get_train_list(project_id: int, session: Session):
+    """
+    根据项目id查询训练列表
+    :param project_id:
+    :param session:
+    :return:
+    """
+    query = session.query(ProjectTrain).filter_by(project_id=project_id).order_by(asc(ProjectTrain.id))
+    train_list = [ProjectTrainOut.from_orm(train) for train in query.all()]
+    return train_list