149 lines
4.8 KiB
Python
149 lines
4.8 KiB
Python
import enum
|
|
|
|
import marshmallow
|
|
from jinja2 import Environment, BaseLoader
|
|
from marshmallow import INCLUDE
|
|
from marshmallow_enum import EnumField
|
|
from marshmallow_sqlalchemy import ModelSchema
|
|
|
|
from crc import db, ma
|
|
|
|
|
|
class WorkflowSpecModel(db.Model):
|
|
__tablename__ = 'workflow_spec'
|
|
id = db.Column(db.String, primary_key=True)
|
|
name = db.Column(db.String)
|
|
display_name = db.Column(db.String)
|
|
description = db.Column(db.Text)
|
|
primary_process_id = db.Column(db.String)
|
|
|
|
|
|
class WorkflowSpecModelSchema(ModelSchema):
|
|
class Meta:
|
|
model = WorkflowSpecModel
|
|
|
|
|
|
class WorkflowStatus(enum.Enum):
|
|
new = "new"
|
|
user_input_required = "user_input_required"
|
|
waiting = "waiting"
|
|
complete = "complete"
|
|
|
|
|
|
class WorkflowModel(db.Model):
|
|
__tablename__ = 'workflow'
|
|
id = db.Column(db.Integer, primary_key=True)
|
|
bpmn_workflow_json = db.Column(db.JSON)
|
|
status = db.Column(db.Enum(WorkflowStatus))
|
|
study_id = db.Column(db.Integer, db.ForeignKey('study.id'))
|
|
workflow_spec_id = db.Column(db.String, db.ForeignKey('workflow_spec.id'))
|
|
|
|
|
|
class Task(object):
|
|
def __init__(self, id, name, title, type, state, form, documentation, data):
|
|
self.id = id
|
|
self.name = name
|
|
self.title = title
|
|
self.type = type
|
|
self.state = state
|
|
self.form = form
|
|
self.documentation = documentation
|
|
self.data = data
|
|
|
|
@classmethod
|
|
def from_spiff(cls, spiff_task):
|
|
try:
|
|
documentation = spiff_task.task_spec.documentation
|
|
except AttributeError:
|
|
documentation = ""
|
|
instance = cls(spiff_task.id,
|
|
spiff_task.task_spec.name,
|
|
spiff_task.task_spec.description,
|
|
spiff_task.task_spec.__class__.__name__,
|
|
spiff_task.get_state_name(),
|
|
None,
|
|
documentation,
|
|
spiff_task.data)
|
|
if hasattr(spiff_task.task_spec, "form"):
|
|
instance.form = spiff_task.task_spec.form
|
|
if documentation != "" and documentation is not None:
|
|
instance.process_documentation(documentation)
|
|
return instance
|
|
|
|
def process_documentation(self, documentation):
|
|
'''Runs markdown documentation through the Jinja2 processor to inject data
|
|
create loops, etc...'''
|
|
rtemplate = Environment(autoescape=True, loader=BaseLoader).from_string(documentation)
|
|
self.documentation = rtemplate.render(**self.data)
|
|
|
|
|
|
class OptionSchema(ma.Schema):
|
|
class Meta:
|
|
fields = ["id", "name"]
|
|
|
|
|
|
class ValidationSchema(ma.Schema):
|
|
class Meta:
|
|
fields = ["name", "config"]
|
|
|
|
|
|
class PropertiesSchema(ma.Schema):
|
|
class Meta:
|
|
fields = ["id", "value"]
|
|
|
|
|
|
class FormFieldSchema(ma.Schema):
|
|
class Meta:
|
|
fields = [
|
|
"id", "type", "label", "default_value", "options", "validation", "properties", "value"
|
|
]
|
|
|
|
default_value = marshmallow.fields.String(required=False, allow_none=True)
|
|
options = marshmallow.fields.List(marshmallow.fields.Nested(OptionSchema))
|
|
validation = marshmallow.fields.List(marshmallow.fields.Nested(ValidationSchema))
|
|
properties = marshmallow.fields.List(marshmallow.fields.Nested(PropertiesSchema))
|
|
|
|
|
|
class FormSchema(ma.Schema):
|
|
key = marshmallow.fields.String(required=True, allow_none=False)
|
|
fields = marshmallow.fields.List(marshmallow.fields.Nested(FormFieldSchema))
|
|
|
|
|
|
class TaskSchema(ma.Schema):
|
|
class Meta:
|
|
fields = ["id", "name", "title", "type", "state", "form", "documentation", "data"]
|
|
|
|
documentation = marshmallow.fields.String(required=False, allow_none=True)
|
|
form = marshmallow.fields.Nested(FormSchema, required=False, allow_none=True)
|
|
title = marshmallow.fields.String(required=False, allow_none=True)
|
|
|
|
@marshmallow.post_load
|
|
def make_task(self, data, **kwargs):
|
|
return Task(**data)
|
|
|
|
|
|
class WorkflowApi(object):
|
|
def __init__(self, id, status, user_tasks, last_task, next_task, workflow_spec_id):
|
|
self.id = id
|
|
self.status = status
|
|
self.user_tasks = user_tasks
|
|
self.last_task = last_task
|
|
self.next_task = next_task
|
|
self.workflow_spec_id = workflow_spec_id
|
|
|
|
|
|
class WorkflowApiSchema(ma.Schema):
|
|
class Meta:
|
|
model = WorkflowApi
|
|
fields = ["id", "status", "user_tasks", "last_task", "next_task", "workflow_spec_id"]
|
|
unknown = INCLUDE
|
|
|
|
status = EnumField(WorkflowStatus)
|
|
user_tasks = marshmallow.fields.List(marshmallow.fields.Nested(TaskSchema, dump_only=True))
|
|
last_task = marshmallow.fields.Nested(TaskSchema, dump_only=True)
|
|
next_task = marshmallow.fields.Nested(TaskSchema, dump_only=True, required=False)
|
|
|
|
@marshmallow.post_load
|
|
def make_workflow(self, data, **kwargs):
|
|
return WorkflowApi(**data)
|