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

82 lines
2.6 KiB
Python
Raw Normal View History

import enum
2020-03-18 19:16:15 +00:00
import marshmallow
from marshmallow import EXCLUDE
from marshmallow_sqlalchemy import SQLAlchemyAutoSchema
from crc import db
2020-03-18 20:58:50 +00:00
2020-03-13 18:56:46 +00:00
class WorkflowSpecCategoryModel(db.Model):
__tablename__ = 'workflow_spec_category'
id = db.Column(db.Integer, primary_key=True)
name = db.Column(db.String)
display_name = db.Column(db.String)
2020-03-26 19:29:52 +00:00
display_order = db.Column(db.Integer)
2020-03-13 18:56:46 +00:00
class WorkflowSpecCategoryModelSchema(SQLAlchemyAutoSchema):
2020-03-13 18:56:46 +00:00
class Meta:
model = WorkflowSpecCategoryModel
load_instance = True
include_relationships = True
2020-03-13 18:56:46 +00:00
class WorkflowSpecModel(db.Model):
__tablename__ = 'workflow_spec'
id = db.Column(db.String, primary_key=True)
2020-01-21 20:22:44 +00:00
name = db.Column(db.String)
display_name = db.Column(db.String)
2020-04-09 18:25:14 +00:00
display_order = db.Column(db.Integer, nullable=True)
description = db.Column(db.Text)
category_id = db.Column(db.Integer, db.ForeignKey('workflow_spec_category.id'), nullable=True)
category = db.relationship("WorkflowSpecCategoryModel")
is_master_spec = db.Column(db.Boolean, default=False)
2020-02-18 15:14:03 +00:00
class WorkflowSpecModelSchema(SQLAlchemyAutoSchema):
class Meta:
model = WorkflowSpecModel
load_instance = True
include_relationships = True
2020-03-13 18:56:46 +00:00
include_fk = True # Includes foreign keys
unknown = EXCLUDE
category = marshmallow.fields.Nested(WorkflowSpecCategoryModelSchema, dump_only=True)
2020-03-18 19:16:15 +00:00
class WorkflowState(enum.Enum):
hidden = "hidden"
disabled = "disabled"
required = "required"
optional = "optional"
@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"
class WorkflowModel(db.Model):
__tablename__ = 'workflow'
id = db.Column(db.Integer, primary_key=True)
2020-01-23 15:53:59 +00:00
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'))
workflow_spec = db.relationship("WorkflowSpecModel")
spec_version = db.Column(db.String)
total_tasks = db.Column(db.Integer, default=0)
completed_tasks = db.Column(db.Integer, default=0)
# task_history = db.Column(db.ARRAY(db.String), default=[]) # The history stack of user completed tasks.
last_updated = db.Column(db.DateTime)