49 lines
1.7 KiB
Python
49 lines
1.7 KiB
Python
from connexion import NoContent
|
|
|
|
from crc import db
|
|
from crc.api.common import ApiError, ApiErrorSchema
|
|
from crc.models.study import StudySchema, StudyModel
|
|
from crc.models.workflow import WorkflowModel, WorkflowSchema, WorkflowSpecModel
|
|
from crc.workflow_processor import WorkflowProcessor
|
|
|
|
|
|
def all_studies():
|
|
# todo: Limit returned studies to a user
|
|
schema = StudySchema(many=True)
|
|
return schema.dump(db.session.query(StudyModel).all())
|
|
|
|
|
|
def get_study(study_id):
|
|
study = db.session.query(StudyModel).filter_by(id=study_id).first()
|
|
schema = StudySchema()
|
|
if study is None:
|
|
return NoContent, 404
|
|
return schema.dump(study)
|
|
|
|
|
|
def post_update_study_from_protocol_builder(study_id):
|
|
# todo: Actually get data from an external service here
|
|
return NoContent, 304
|
|
|
|
|
|
def get_study_workflows(study_id):
|
|
workflows = db.session.query(WorkflowModel).filter_by(study_id=study_id).all()
|
|
schema = WorkflowSchema(many=True)
|
|
return schema.dump(workflows)
|
|
|
|
|
|
def add_workflow_to_study(study_id, body):
|
|
workflow_spec_model = db.session.query(WorkflowSpecModel).filter_by(id=body["id"]).first()
|
|
if workflow_spec_model is None:
|
|
error = ApiError('unknown_spec', 'The specification "' + body['id'] + '" is not recognized.')
|
|
return ApiErrorSchema.dump(error), 404
|
|
|
|
processor = WorkflowProcessor.create(workflow_spec_model.id)
|
|
workflow = WorkflowModel(bpmn_workflow_json=processor.serialize(),
|
|
status=processor.get_status(),
|
|
study_id=study_id,
|
|
workflow_spec_id=workflow_spec_model.id)
|
|
db.session.add(workflow)
|
|
db.session.commit()
|
|
return WorkflowSchema().dump(workflow)
|