1 ##############################################################################
2 # Copyright (c) 2018 Sawyer Bergeron, Parker Berberian, and others.
4 # All rights reserved. This program and the accompanying materials
5 # are made available under the terms of the Apache License, Version 2.0
6 # which accompanies this distribution, and is available at
7 # http://www.apache.org/licenses/LICENSE-2.0
8 ##############################################################################
11 from workflow.booking_workflow import Booking_Resource_Select, SWConfig_Select, Booking_Meta
12 from workflow.resource_bundle_workflow import Define_Hardware, Define_Nets, Resource_Meta_Info
13 from workflow.sw_bundle_workflow import Config_Software, Define_Software, SWConf_Resource_Select
14 from workflow.snapshot_workflow import Select_Host_Step, Image_Meta_Step
15 from workflow.models import Confirmation_Step
20 logger = logging.getLogger(__name__)
23 class BookingMetaWorkflow(object):
29 class ResourceMetaWorkflow(object):
34 class ConfigMetaWorkflow(object):
39 class MetaStep(object):
45 def set_invalid(self, message, code=100):
47 self.message = message
49 def set_valid(self, message, code=200):
51 self.message = message
53 def __init__(self, *args, **kwargs):
54 self.short_title = "error"
58 self.id = uuid.uuid4()
62 'title': self.short_title,
63 'skip': self.skip_step,
65 'message': self.message,
69 return "metastep: " + str(self.short_title)
74 def __eq__(self, other):
75 return self.id.int == other.id.int
77 def __ne__(self, other):
78 return self.id.int != other.id.int
81 class Workflow(object):
82 def __init__(self, steps, metasteps, repository):
83 self.repository = repository
85 self.metasteps = metasteps
89 class WorkflowFactory():
91 Booking_Resource_Select,
103 SWConf_Resource_Select,
113 def conjure(self, workflow_type=None, repo=None):
121 steps = self.make_steps(workflow_types[workflow_type], repository=repo)
122 meta_steps = self.metaize(steps=steps, wf_type=workflow_type)
123 return steps, meta_steps
125 def create_workflow(self, workflow_type=None, repo=None):
126 steps, meta_steps = self.conjure(workflow_type, repo)
127 c_step = self.make_step(Confirmation_Step, repo)
128 metaconfirm = MetaStep()
129 metaconfirm.short_title = "confirm"
130 metaconfirm.index = len(steps)
132 meta_steps.append(metaconfirm)
133 return Workflow(steps, meta_steps, repo)
135 def make_steps(self, step_types, repository):
137 for step_type in step_types:
138 steps.append(self.make_step(step_type, repository))
142 def make_step(self, step_type, repository):
143 iden = step_type.description + step_type.title + step_type.template
144 return step_type(iden, repository)
146 def metaize(self, steps, wf_type):
149 meta_step = MetaStep()
150 meta_step.short_title = step.short_title
151 meta_dict.append(meta_step)
152 step.metastep = meta_step