dashboard/src/workflow/workflow_manager.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 django.http import JsonResponse
  12
  13 from booking.models import Booking
  14 from workflow.workflow_factory import WorkflowFactory
  15 from workflow.models import Repository
  16 from resource_inventory.models import (
  17     GenericResourceBundle,
  18     ConfigBundle,
  19     HostConfiguration,
  20     OPNFVConfig
  21 )
  22
  23 import logging
  24 logger = logging.getLogger(__name__)
  25
  26
  27 class SessionManager():
  28     def active_workflow(self):
  29         return self.workflows[-1]
  30
  31     def __init__(self, request=None):
  32         self.workflows = []
  33
  34         self.owner = request.user
  35
  36         self.factory = WorkflowFactory()
  37
  38     def set_step_statuses(self, superclass_type, desired_enabled=True):
  39         workflow = self.active_workflow()
  40         steps = workflow.steps
  41         for step in steps:
  42             if isinstance(step, superclass_type):
  43                 if desired_enabled:
  44                     step.enable()
  45                 else:
  46                     step.disable()
  47
  48     def add_workflow(self, workflow_type=None, target_id=None, **kwargs):
  49         if target_id is not None:
  50             self.prefill_repo(target_id, workflow_type)
  51
  52         repo = Repository()
  53         if(len(self.workflows) >= 1):
  54             defaults = self.workflows[-1].repository.get_child_defaults()
  55             repo.set_defaults(defaults)
  56             repo.el[repo.HAS_RESULT] = False
  57         repo.el[repo.SESSION_USER] = self.owner
  58         repo.el[repo.SESSION_MANAGER] = self
  59         self.workflows.append(
  60             self.factory.create_workflow(
  61                 workflow_type=workflow_type,
  62                 repo=repo
  63             )
  64         )
  65
  66     def pop_workflow(self):
  67         multiple_wfs = len(self.workflows) > 1
  68         if multiple_wfs:
  69             if self.workflows[-1].repository.el[Repository.RESULT]:  # move result
  70                 key = self.workflows[-1].repository.el[Repository.RESULT_KEY]
  71                 result = self.workflows[-1].repository.el[Repository.RESULT]
  72                 self.workflows[-2].repository.el[key] = result
  73             self.workflows.pop()
  74         current_repo = self.workflows[-1].repository
  75         return (multiple_wfs, current_repo.el[current_repo.RESULT])
  76
  77     def status(self, request):
  78         try:
  79             meta_json = []
  80             for step in self.active_workflow().steps:
  81                 meta_json.append(step.to_json())
  82             responsejson = {}
  83             responsejson["steps"] = meta_json
  84             responsejson["active"] = self.active_workflow().repository.el['active_step']
  85             responsejson["workflow_count"] = len(self.workflows)
  86             return JsonResponse(responsejson, safe=False)
  87         except Exception:
  88             pass
  89
  90     def render(self, request, **kwargs):
  91         # filter out when a step needs to handle post/form data
  92         # if 'workflow' in post data, this post request was meant for me, not step
  93         if request.method == 'POST' and request.POST.get('workflow', None) is None:
  94             return self.active_workflow().steps[self.active_workflow().active_index].post_render(request)
  95         return self.active_workflow().steps[self.active_workflow().active_index].render(request)
  96
  97     def post_render(self, request):
  98         return self.active_workflow().steps[self.active_workflow().active_index].post_render(request)
  99
 100     def go_next(self, **kwargs):
 101         next_step = self.active_workflow().active_index + 1
 102         if next_step >= len(self.active_workflow().steps):
 103             raise Exception("Out of bounds request for step")
 104         while not self.active_workflow().steps[next_step].enabled:
 105             next_step += 1
 106         self.active_workflow().repository.el['active_step'] = next_step
 107         self.active_workflow().active_index = next_step
 108
 109     def go_prev(self, **kwargs):
 110         prev_step = self.active_workflow().active_index - 1
 111         if prev_step < 0:
 112             raise Exception("Out of bounds request for step")
 113         while not self.active_workflow().steps[prev_step].enabled:
 114             prev_step -= 1
 115         self.active_workflow().repository.el['active_step'] = prev_step
 116         self.active_workflow().active_index = prev_step
 117
 118     def prefill_repo(self, target_id, workflow_type):
 119         self.repository.el[self.repository.EDIT] = True
 120         edit_object = None
 121         if workflow_type == 0:
 122             edit_object = Booking.objects.get(pk=target_id)
 123             self.prefill_booking(edit_object)
 124         elif workflow_type == 1:
 125             edit_object = GenericResourceBundle.objects.get(pk=target_id)
 126             self.prefill_resource(edit_object)
 127         elif workflow_type == 2:
 128             edit_object = ConfigBundle.objects.get(pk=target_id)
 129             self.prefill_config(edit_object)
 130
 131     def prefill_booking(self, booking):
 132         models = self.make_booking_models(booking)
 133         confirmation = self.make_booking_confirm(booking)
 134         self.active_workflow().repository.el[self.active_workflow().repository.BOOKING_MODELS] = models
 135         self.active_workflow().repository.el[self.active_workflow().repository.CONFIRMATION] = confirmation
 136         self.active_workflow().repository.el[self.active_workflow().repository.GRESOURCE_BUNDLE_MODELS] = self.make_grb_models(booking.resource.template)
 137         self.active_workflow().repository.el[self.active_workflow().repository.SELECTED_GRESOURCE_BUNDLE] = self.make_grb_models(booking.resource.template)['bundle']
 138         self.active_workflow().repository.el[self.active_workflow().repository.CONFIG_MODELS] = self.make_config_models(booking.config_bundle)
 139
 140     def prefill_resource(self, resource):
 141         models = self.make_grb_models(resource)
 142         confirm = self.make_grb_confirm(resource)
 143         self.active_workflow().repository.el[self.active_workflow().repository.GRESOURCE_BUNDLE_MODELS] = models
 144         self.active_workflow().repository.el[self.active_workflow().repository.CONFIRMATION] = confirm
 145
 146     def prefill_config(self, config):
 147         models = self.make_config_models(config)
 148         confirm = self.make_config_confirm(config)
 149         self.active_workflow().repository.el[self.active_workflow().repository.CONFIG_MODELS] = models
 150         self.active_workflow().repository.el[self.active_workflow().repository.CONFIRMATION] = confirm
 151         grb_models = self.make_grb_models(config.bundle)
 152         self.active_workflow().repository.el[self.active_workflow().repository.GRESOURCE_BUNDLE_MODELS] = grb_models
 153
 154     def make_grb_models(self, resource):
 155         models = self.active_workflow().repository.el.get(self.active_workflow().repository.GRESOURCE_BUNDLE_MODELS, {})
 156         models['hosts'] = []
 157         models['bundle'] = resource
 158         models['interfaces'] = {}
 159         models['vlans'] = {}
 160         for host in resource.getHosts():
 161             models['hosts'].append(host)
 162             models['interfaces'][host.resource.name] = []
 163             models['vlans'][host.resource.name] = {}
 164             for interface in host.generic_interfaces.all():
 165                 models['interfaces'][host.resource.name].append(interface)
 166                 models['vlans'][host.resource.name][interface.profile.name] = []
 167                 for vlan in interface.vlans.all():
 168                     models['vlans'][host.resource.name][interface.profile.name].append(vlan)
 169         return models
 170
 171     def make_grb_confirm(self, resource):
 172         confirm = self.active_workflow().repository.el.get(self.active_workflow().repository.CONFIRMATION, {})
 173         confirm['resource'] = {}
 174         confirm['resource']['hosts'] = []
 175         confirm['resource']['lab'] = resource.lab.lab_user.username
 176         for host in resource.getHosts():
 177             confirm['resource']['hosts'].append({"name": host.resource.name, "profile": host.profile.name})
 178         return confirm
 179
 180     def make_config_models(self, config):
 181         models = self.active_workflow().repository.el.get(self.active_workflow().repository.CONFIG_MODELS, {})
 182         models['bundle'] = config
 183         models['host_configs'] = []
 184         for host_conf in HostConfiguration.objects.filter(bundle=config):
 185             models['host_configs'].append(host_conf)
 186         models['opnfv'] = OPNFVConfig.objects.filter(bundle=config).last()
 187         return models
 188
 189     def make_config_confirm(self, config):
 190         confirm = self.active_workflow().repository.el.get(self.active_workflow().repository.CONFIRMATION, {})
 191         confirm['configuration'] = {}
 192         confirm['configuration']['hosts'] = []
 193         confirm['configuration']['name'] = config.name
 194         confirm['configuration']['description'] = config.description
 195         opnfv = OPNFVConfig.objects.filter(bundle=config).last()
 196         confirm['configuration']['installer'] = opnfv.installer.name
 197         confirm['configuration']['scenario'] = opnfv.scenario.name
 198         for host_conf in HostConfiguration.objects.filter(bundle=config):
 199             h = {"name": host_conf.host.resource.name, "image": host_conf.image.name, "role": host_conf.opnfvRole.name}
 200             confirm['configuration']['hosts'].append(h)
 201         return confirm
 202
 203     def make_booking_models(self, booking):
 204         models = self.active_workflow().repository.el.get(self.active_workflow().repository.BOOKING_MODELS, {})
 205         models['booking'] = booking
 206         models['collaborators'] = []
 207         for user in booking.collaborators.all():
 208             models['collaborators'].append(user)
 209         return models
 210
 211     def make_booking_confirm(self, booking):
 212         confirm = self.active_workflow().repository.el.get(self.active_workflow().repository.CONFIRMATION, {})
 213         confirm['booking'] = {}
 214         confirm['booking']['length'] = (booking.end - booking.start).days
 215         confirm['booking']['project'] = booking.project
 216         confirm['booking']['purpose'] = booking.purpose
 217         confirm['booking']['resource name'] = booking.resource.template.name
 218         confirm['booking']['configuration name'] = booking.config_bundle.name
 219         confirm['booking']['collaborators'] = []
 220         for user in booking.collaborators.all():
 221             confirm['booking']['collaborators'].append(user.username)
 222         return confirm
 223
 224
 225 class ManagerTracker():
 226     instance = None
 227
 228     managers = {}
 229
 230     def __init__(self):
 231         pass
 232
 233     @staticmethod
 234     def getInstance():
 235         if ManagerTracker.instance is None:
 236             ManagerTracker.instance = ManagerTracker()
 237         return ManagerTracker.instance