mirror of
https://github.com/sartography/cr-connect-workflow.git
synced 2025-02-22 12:48:25 +00:00
92 lines
3.1 KiB
Python
92 lines
3.1 KiB
Python
import uuid
|
|
|
|
from crc import session
|
|
from crc.api.common import ApiError, ApiErrorSchema
|
|
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(session.query(WorkflowSpecModel).all())
|
|
|
|
|
|
def add_workflow_specification(body):
|
|
new_spec = WorkflowSpecModelSchema().load(body, session=session)
|
|
session.add(new_spec)
|
|
session.commit()
|
|
return WorkflowSpecModelSchema().dump(new_spec)
|
|
|
|
|
|
def update_workflow_specification(spec_id, body):
|
|
if spec_id is None:
|
|
error = ApiError('unknown_spec', 'Please provide a valid Workflow Specification ID.')
|
|
return ApiErrorSchema.dump(error), 404
|
|
|
|
spec: WorkflowSpecModel = session.query(WorkflowSpecModel).filter_by(id=spec_id).first()
|
|
|
|
if spec is None:
|
|
error = ApiError('unknown_spec', 'The Workflow Specification "' + spec_id + '" is not recognized.')
|
|
return ApiErrorSchema.dump(error), 404
|
|
|
|
spec = WorkflowSpecModelSchema().load(body, session=session)
|
|
session.add(spec)
|
|
session.commit()
|
|
return WorkflowSpecModelSchema().dump(spec)
|
|
|
|
|
|
def delete_workflow_specification(spec_id):
|
|
if spec_id is None:
|
|
error = ApiError('unknown_spec', 'Please provide a valid Workflow Specification ID.')
|
|
return ApiErrorSchema.dump(error), 404
|
|
|
|
spec: WorkflowSpecModel = session.query(WorkflowSpecModel).filter_by(id=spec_id).first()
|
|
|
|
if spec is None:
|
|
error = ApiError('unknown_spec', 'The Workflow Specification "' + spec_id + '" is not recognized.')
|
|
return ApiErrorSchema.dump(error), 404
|
|
|
|
session.query(WorkflowSpecModel).filter_by(id=spec_id).delete()
|
|
|
|
# TODO: Delete everything else with the same workflow_spec_id?
|
|
session.commit()
|
|
|
|
|
|
def get_workflow(workflow_id):
|
|
schema = WorkflowModelSchema()
|
|
workflow = session.query(WorkflowModel).filter_by(id=workflow_id).first()
|
|
return schema.dump(workflow)
|
|
|
|
|
|
def delete(workflow_id):
|
|
session.query(WorkflowModel).filter_by(id=workflow_id).delete()
|
|
session.commit()
|
|
|
|
|
|
def get_tasks(workflow_id):
|
|
workflow = 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 = session.query(WorkflowModel).filter_by(id=workflow_id).first()
|
|
return workflow.bpmn_workflow().get_task(task_id)
|
|
|
|
|
|
def update_task(workflow_id, task_id, body):
|
|
workflow = session.query(WorkflowModel).filter_by(id=workflow_id).first()
|
|
processor = WorkflowProcessor(workflow.workflow_spec_id, workflow.bpmn_workflow_json)
|
|
task_id = uuid.UUID(task_id)
|
|
task = processor.bpmn_workflow.get_task(task_id)
|
|
task.data = body
|
|
processor.complete_task(task)
|
|
workflow.bpmn_workflow_json = processor.serialize()
|
|
session.add(workflow)
|
|
session.commit()
|