29 lines
740 B
Python
29 lines
740 B
Python
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
|