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 django.http import JsonResponse
15 from booking.models import Booking
16 from workflow.workflow_factory import WorkflowFactory, MetaStep, MetaRelation
17 from workflow.models import Repository, Confirmation_Step
18 from resource_inventory.models import (
19 GenericResourceBundle,
26 logger = logging.getLogger(__name__)
29 class SessionManager():
31 def __init__(self, request=None):
32 self.repository = Repository()
33 self.repository.el[self.repository.SESSION_USER] = request.user
34 self.repository.el['active_step'] = 0
36 self.factory = WorkflowFactory()
37 c_step = WorkflowFactory().make_step(Confirmation_Step, self.repository)
38 self.steps.append(c_step)
39 metaconfirm = MetaStep()
41 metaconfirm.short_title = "confirm"
42 self.repository.el['steps'] = 1
43 self.metaworkflow = None
44 self.metaworkflows = []
45 self.metarelations = []
46 self.relationreverselookup = {}
47 self.initialized = False
49 self.step_meta = [metaconfirm]
50 self.relation_depth = 0
52 def add_workflow(self, workflow_type=None, target_id=None, **kwargs):
53 if target_id is not None:
54 self.prefill_repo(target_id, workflow_type)
55 factory_steps, meta_info = self.factory.conjure(workflow_type=workflow_type, repo=self.repository)
56 offset = len(meta_info)
57 for relation in self.metarelations:
58 if relation.depth > self.relation_depth:
59 self.relation_depth = relation.depth
60 if relation.parent >= self.repository.el['active_step']:
61 relation.parent += offset
62 for i in range(0, len(relation.children)):
63 if relation.children[i] >= self.repository.el['active_step']:
64 relation.children[i] += offset
65 self.step_meta[self.active_index:self.active_index] = meta_info
66 self.steps[self.active_index:self.active_index] = factory_steps
69 relation = MetaRelation()
70 relation.parent = self.repository.el['active_step'] + offset
71 relation.depth = self.relationreverselookup[self.step_meta[relation.parent]].depth + 1
72 if relation.depth > self.relation_depth:
73 self.relation_depth = relation.depth
74 for i in range(self.repository.el['active_step'], offset + self.repository.el['active_step']):
75 relation.children.append(i)
76 self.relationreverselookup[self.step_meta[i]] = relation
77 relation.color = "#%06x" % random.randint(0, 0xFFFFFF)
78 self.metarelations.append(relation)
80 relation = MetaRelation()
82 relation.parent = 500000000000
83 for i in range(0, len(self.step_meta)):
84 relation.children.append(i)
85 self.relationreverselookup[self.step_meta[i]] = relation
86 self.metarelations.append(relation)
87 self.initialized = True
89 def status(self, request):
92 for step in self.step_meta:
93 steps.append(step.to_json())
97 responsejson["steps"] = steps
98 responsejson["active"] = self.repository.el['active_step']
99 responsejson["relations"] = []
101 for relation in self.metarelations:
102 responsejson["relations"].append(relation.to_json())
103 children[relation.parent] = i
104 for child in relation.children:
107 responsejson['max_depth'] = self.relation_depth
108 responsejson['parents'] = parents
109 responsejson['children'] = children
110 return JsonResponse(responsejson, safe=False)
114 def render(self, request, **kwargs):
115 # filter out when a step needs to handle post/form data
116 # if 'workflow' in post data, this post request was meant for me, not step
117 if request.method == 'POST' and request.POST.get('workflow', None) is None:
118 return self.steps[self.active_index].post_render(request)
119 return self.steps[self.active_index].render(request)
121 def post_render(self, request):
122 return self.steps[self.active_index].post_render(request)
124 def goto(self, num, **kwargs):
125 self.repository.el['active_step'] = int(num)
126 self.active_index = int(num)
127 # TODO: change to include some checking
129 def prefill_repo(self, target_id, workflow_type):
130 self.repository.el[self.repository.EDIT] = True
132 if workflow_type == 0:
133 edit_object = Booking.objects.get(pk=target_id)
134 self.prefill_booking(edit_object)
135 elif workflow_type == 1:
136 edit_object = GenericResourceBundle.objects.get(pk=target_id)
137 self.prefill_resource(edit_object)
138 elif workflow_type == 2:
139 edit_object = ConfigBundle.objects.get(pk=target_id)
140 self.prefill_config(edit_object)
142 def prefill_booking(self, booking):
143 models = self.make_booking_models(booking)
144 confirmation = self.make_booking_confirm(booking)
145 self.repository.el[self.repository.BOOKING_MODELS] = models
146 self.repository.el[self.repository.CONFIRMATION] = confirmation
147 self.repository.el[self.repository.GRESOURCE_BUNDLE_MODELS] = self.make_grb_models(booking.resource.template)
148 self.repository.el[self.repository.BOOKING_SELECTED_GRB] = self.make_grb_models(booking.resource.template)['bundle']
149 self.repository.el[self.repository.CONFIG_MODELS] = self.make_config_models(booking.config_bundle)
151 def prefill_resource(self, resource):
152 models = self.make_grb_models(resource)
153 confirm = self.make_grb_confirm(resource)
154 self.repository.el[self.repository.GRESOURCE_BUNDLE_MODELS] = models
155 self.repository.el[self.repository.CONFIRMATION] = confirm
157 def prefill_config(self, config):
158 models = self.make_config_models(config)
159 confirm = self.make_config_confirm(config)
160 self.repository.el[self.repository.CONFIG_MODELS] = models
161 self.repository.el[self.repository.CONFIRMATION] = confirm
162 grb_models = self.make_grb_models(config.bundle)
163 self.repository.el[self.repository.GRESOURCE_BUNDLE_MODELS] = grb_models
164 self.repository.el[self.repository.SWCONF_SELECTED_GRB] = config.bundle
166 def make_grb_models(self, resource):
167 models = self.repository.el.get(self.repository.GRESOURCE_BUNDLE_MODELS, {})
169 models['bundle'] = resource
170 models['interfaces'] = {}
172 for host in resource.getHosts():
173 models['hosts'].append(host)
174 models['interfaces'][host.resource.name] = []
175 models['vlans'][host.resource.name] = {}
176 for interface in host.generic_interfaces.all():
177 models['interfaces'][host.resource.name].append(interface)
178 models['vlans'][host.resource.name][interface.profile.name] = []
179 for vlan in interface.vlans.all():
180 models['vlans'][host.resource.name][interface.profile.name].append(vlan)
183 def make_grb_confirm(self, resource):
184 confirm = self.repository.el.get(self.repository.CONFIRMATION, {})
185 confirm['resource'] = {}
186 confirm['resource']['hosts'] = []
187 confirm['resource']['lab'] = resource.lab.lab_user.username
188 for host in resource.getHosts():
189 confirm['resource']['hosts'].append({"name": host.resource.name, "profile": host.profile.name})
192 def make_config_models(self, config):
193 models = self.repository.el.get(self.repository.CONFIG_MODELS, {})
194 models['bundle'] = config
195 models['host_configs'] = []
196 for host_conf in HostConfiguration.objects.filter(bundle=config):
197 models['host_configs'].append(host_conf)
198 models['opnfv'] = OPNFVConfig.objects.filter(bundle=config).last()
201 def make_config_confirm(self, config):
202 confirm = self.repository.el.get(self.repository.CONFIRMATION, {})
203 confirm['configuration'] = {}
204 confirm['configuration']['hosts'] = []
205 confirm['configuration']['name'] = config.name
206 confirm['configuration']['description'] = config.description
207 opnfv = OPNFVConfig.objects.filter(bundle=config).last()
208 confirm['configuration']['installer'] = opnfv.installer.name
209 confirm['configuration']['scenario'] = opnfv.scenario.name
210 for host_conf in HostConfiguration.objects.filter(bundle=config):
211 h = {"name": host_conf.host.resource.name, "image": host_conf.image.name, "role": host_conf.opnfvRole.name}
212 confirm['configuration']['hosts'].append(h)
215 def make_booking_models(self, booking):
216 models = self.repository.el.get(self.repository.BOOKING_MODELS, {})
217 models['booking'] = booking
218 models['collaborators'] = []
219 for user in booking.collaborators.all():
220 models['collaborators'].append(user)
223 def make_booking_confirm(self, booking):
224 confirm = self.repository.el.get(self.repository.CONFIRMATION, {})
225 confirm['booking'] = {}
226 confirm['booking']['length'] = (booking.end - booking.start).days
227 confirm['booking']['project'] = booking.project
228 confirm['booking']['purpose'] = booking.purpose
229 confirm['booking']['resource name'] = booking.resource.template.name
230 confirm['booking']['configuration name'] = booking.config_bundle.name
231 confirm['booking']['collaborators'] = []
232 for user in booking.collaborators.all():
233 confirm['booking']['collaborators'].append(user.username)
237 class ManagerTracker():
247 if ManagerTracker.instance is None:
248 ManagerTracker.instance = ManagerTracker()
249 return ManagerTracker.instance