cr-connect-workflow/crc/api/workflow.py

from crc import db
from crc.models.workflow import WorkflowModel, WorkflowModelSchema, WorkflowSpecModelSchema, WorkflowSpecModel, \
    Task, TaskSchema
from crc.workflow_processor import WorkflowProcessor


def all_specifications():
    schema = WorkflowSpecModelSchema(many=True)
    return schema.dump(db.session.query(WorkflowSpecModel).all())


def add_workflow_specification(body):
    workflow_spec = WorkflowSpecModelSchema().load(body, session=db.session)
    db.session.add(workflow_spec)
    db.session.commit()
    return WorkflowSpecModelSchema().dump(workflow_spec)


def get_workflow(workflow_id):
    schema = WorkflowModelSchema()
    workflow = db.session.query(WorkflowModel).filter_by(id=workflow_id).first()
    return schema.dump(workflow)


def delete(workflow_id):
    db.session.query(WorkflowModel).filter_by(id=workflow_id).delete()
    db.session.commit()


def get_tasks(workflow_id):
    workflow = db.session.query(WorkflowModel).filter_by(id=workflow_id).first()
    processor = WorkflowProcessor(workflow.workflow_spec_id, workflow.bpmn_workflow_json)
    spiff_tasks = processor.get_ready_user_tasks()
    tasks = []
    for st in spiff_tasks:
        tasks.append(Task.from_spiff(st))
    return TaskSchema(many=True).dump(tasks)


def get_task(workflow_id, task_id):
    workflow = db.session.query(WorkflowModel).filter_by(id=workflow_id).first()
    return workflow.bpmn_workflow().get_task(task_id)


def update_task(workflow_id, task_name, body):
    workflow = db.session.query(WorkflowModel).filter_by(id=workflow_id).first()
    processor = WorkflowProcessor(workflow.workflow_spec_id, workflow.bpmn_workflow_json)
    task = processor.bpmn_workflow.get_tasks_from_spec_name(task_name)[0]
    if workflow and task and body:
        print('workflow', workflow.id)
        print('task', task.id)
        print('body', body)
        processor = WorkflowProcessor(workflow.workflow_spec_id, workflow.bpmn_workflow_json)
        processor.complete_task(task)
    return body