mirror of
https://github.com/sartography/cr-connect-workflow.git
synced 2025-02-24 05:38:25 +00:00
123 lines
4.2 KiB
Python
123 lines
4.2 KiB
Python
import enum
|
|
|
|
import marshmallow
|
|
from marshmallow import EXCLUDE, post_load, fields, INCLUDE
|
|
from sqlalchemy import func
|
|
from sqlalchemy.orm import deferred
|
|
|
|
from crc import db, ma
|
|
|
|
|
|
class WorkflowSpecCategory(object):
|
|
def __init__(self, id, display_name, display_order=0, admin=False):
|
|
self.id = id # A unique string name, lower case, under scores (ie, 'my_category')
|
|
self.display_name = display_name
|
|
self.display_order = display_order
|
|
self.admin = admin
|
|
self.workflows = [] # For storing Workflow Metadata
|
|
self.specs = [] # For the list of specifications associated with a category
|
|
self.meta = None # For storing category metadata
|
|
|
|
def __eq__(self, other):
|
|
if not isinstance(other, WorkflowSpecCategory):
|
|
return False
|
|
if other.id == self.id:
|
|
return True
|
|
return False
|
|
|
|
|
|
class WorkflowSpecCategorySchema(ma.Schema):
|
|
class Meta:
|
|
model = WorkflowSpecCategory
|
|
fields = ["id", "display_name", "display_order", "admin"]
|
|
|
|
@post_load
|
|
def make_cat(self, data, **kwargs):
|
|
return WorkflowSpecCategory(**data)
|
|
|
|
|
|
class WorkflowSpecInfo(object):
|
|
def __init__(self, id, display_name, description, is_master_spec=False,
|
|
standalone=False, library=False, primary_file_name='', primary_process_id='',
|
|
libraries=[], category_id="", display_order=0, is_review=False):
|
|
self.id = id # Sting unique id
|
|
self.display_name = display_name
|
|
self.description = description
|
|
self.display_order = display_order
|
|
self.is_master_spec = is_master_spec
|
|
self.standalone = standalone
|
|
self.library = library
|
|
self.primary_file_name = primary_file_name
|
|
self.primary_process_id = primary_process_id
|
|
self.is_review = is_review
|
|
self.libraries = libraries
|
|
self.category_id = category_id
|
|
|
|
def __eq__(self, other):
|
|
if not isinstance(other, WorkflowSpecInfo):
|
|
return False
|
|
if other.id == self.id:
|
|
return True
|
|
return False
|
|
|
|
|
|
class WorkflowSpecInfoSchema(ma.Schema):
|
|
class Meta:
|
|
model = WorkflowSpecInfo
|
|
id = marshmallow.fields.String(required=True)
|
|
display_name = marshmallow.fields.String(required=True)
|
|
description = marshmallow.fields.String()
|
|
is_master_spec = marshmallow.fields.Boolean(required=True)
|
|
standalone = marshmallow.fields.Boolean(required=True)
|
|
library = marshmallow.fields.Boolean(required=True)
|
|
display_order = marshmallow.fields.Integer(allow_none=True)
|
|
primary_file_name = marshmallow.fields.String(allow_none=True)
|
|
primary_process_id = marshmallow.fields.String(allow_none=True)
|
|
is_review = marshmallow.fields.Boolean(allow_none=True)
|
|
category_id = marshmallow.fields.String(allow_none=True)
|
|
libraries = marshmallow.fields.List(marshmallow.fields.String(), allow_none=True)
|
|
|
|
@post_load
|
|
def make_spec(self, data, **kwargs):
|
|
return WorkflowSpecInfo(**data)
|
|
|
|
|
|
class WorkflowState(enum.Enum):
|
|
hidden = "hidden"
|
|
disabled = "disabled"
|
|
required = "required"
|
|
optional = "optional"
|
|
locked = "locked"
|
|
|
|
@classmethod
|
|
def has_value(cls, value):
|
|
return value in cls._value2member_map_
|
|
|
|
@staticmethod
|
|
def list():
|
|
return list(map(lambda c: c.value, WorkflowState))
|
|
|
|
|
|
class WorkflowStatus(enum.Enum):
|
|
not_started = "not_started"
|
|
user_input_required = "user_input_required"
|
|
waiting = "waiting"
|
|
complete = "complete"
|
|
erroring = "erroring"
|
|
|
|
|
|
class WorkflowModel(db.Model):
|
|
__tablename__ = 'workflow'
|
|
id = db.Column(db.Integer, primary_key=True)
|
|
bpmn_workflow_json = deferred(db.Column(db.JSON))
|
|
status = db.Column(db.Enum(WorkflowStatus))
|
|
study_id = db.Column(db.Integer, db.ForeignKey('study.id'))
|
|
study = db.relationship("StudyModel", backref='workflow', lazy='select')
|
|
workflow_spec_id = db.Column(db.String)
|
|
total_tasks = db.Column(db.Integer, default=0)
|
|
completed_tasks = db.Column(db.Integer, default=0)
|
|
last_updated = db.Column(db.DateTime(timezone=True), server_default=func.now())
|
|
user_id = db.Column(db.String, default=None)
|
|
state = db.Column(db.String, nullable=True)
|
|
state_message = db.Column(db.String, nullable=True)
|