src/workflow/workflow_factory.py

   1 ##############################################################################
   2 # Copyright (c) 2018 Sawyer Bergeron, Parker Berberian, and others.
   3 #
   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 ##############################################################################
   9
  10
  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.opnfv_workflow import Pick_Installer, Assign_Network_Roles, Assign_Host_Roles, OPNFV_Resource_Select, MetaInfo
  16 from workflow.models import Confirmation_Step
  17
  18 import uuid
  19
  20 import logging
  21 logger = logging.getLogger(__name__)
  22
  23
  24 class MetaStep(object):
  25
  26     UNTOUCHED = 0
  27     INVALID = 100
  28     VALID = 200
  29
  30     def set_invalid(self, message, code=100):
  31         self.valid = code
  32         self.message = message
  33
  34     def set_valid(self, message, code=200):
  35         self.valid = code
  36         self.message = message
  37
  38     def __init__(self, *args, **kwargs):
  39         self.short_title = "error"
  40         self.skip_step = 0
  41         self.valid = 0
  42         self.hidden = False
  43         self.message = ""
  44         self.id = uuid.uuid4()
  45
  46     def to_json(self):
  47         return {
  48             'title': self.short_title,
  49             'skip': self.skip_step,
  50             'valid': self.valid,
  51             'message': self.message,
  52         }
  53
  54     def __str__(self):
  55         return "metastep: " + str(self.short_title)
  56
  57     def __hash__(self):
  58         return hash(self.id)
  59
  60     def __eq__(self, other):
  61         return self.id.int == other.id.int
  62
  63     def __ne__(self, other):
  64         return self.id.int != other.id.int
  65
  66
  67 class Workflow(object):
  68     def __init__(self, steps, repository):
  69         self.repository = repository
  70         self.steps = steps
  71         self.active_index = 0
  72
  73
  74 class WorkflowFactory():
  75     booking_steps = [
  76         Booking_Resource_Select,
  77         SWConfig_Select,
  78         Booking_Meta,
  79     ]
  80
  81     resource_steps = [
  82         Define_Hardware,
  83         Define_Nets,
  84         Resource_Meta_Info,
  85     ]
  86
  87     config_steps = [
  88         SWConf_Resource_Select,
  89         Define_Software,
  90         Config_Software,
  91     ]
  92
  93     snapshot_steps = [
  94         Select_Host_Step,
  95         Image_Meta_Step,
  96     ]
  97
  98     opnfv_steps = [
  99         OPNFV_Resource_Select,
 100         Pick_Installer,
 101         Assign_Network_Roles,
 102         Assign_Host_Roles,
 103         MetaInfo
 104     ]
 105
 106     def conjure(self, workflow_type=None, repo=None):
 107         workflow_types = [
 108             self.booking_steps,
 109             self.resource_steps,
 110             self.config_steps,
 111             self.snapshot_steps,
 112             self.opnfv_steps,
 113         ]
 114
 115         steps = self.make_steps(workflow_types[workflow_type], repository=repo)
 116         return steps
 117
 118     def create_workflow(self, workflow_type=None, repo=None):
 119         steps = self.conjure(workflow_type, repo)
 120         c_step = self.make_step(Confirmation_Step, repo)
 121         steps.append(c_step)
 122         return Workflow(steps, repo)
 123
 124     def make_steps(self, step_types, repository):
 125         steps = []
 126         for step_type in step_types:
 127             steps.append(self.make_step(step_type, repository))
 128
 129         return steps
 130
 131     def make_step(self, step_type, repository):
 132         iden = step_type.description + step_type.title + step_type.template
 133         return step_type(iden, repository)