完善相关问题,并增加推理的部分代码
This commit is contained in:
@ -26,3 +26,14 @@ def get_train_list(project_id: int, session: Session):
|
||||
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
|
||||
|
||||
|
||||
def get_train(train_id: int, session: Session):
|
||||
"""
|
||||
根据id查询训练信息
|
||||
:param train_id:
|
||||
:param session:
|
||||
:return:
|
||||
"""
|
||||
train = session.query(ProjectTrain).filter_by(id=train_id).first()
|
||||
return train
|
||||
|
Reference in New Issue
Block a user