diff options
Diffstat (limited to 'src/workflow/booking_workflow.py')
| -rw-r--r-- | src/workflow/booking_workflow.py | 295 |
1 files changed, 295 insertions, 0 deletions
diff --git a/src/workflow/booking_workflow.py b/src/workflow/booking_workflow.py new file mode 100644 index 0000000..52fe36b --- /dev/null +++ b/src/workflow/booking_workflow.py @@ -0,0 +1,295 @@ +############################################################################## +# Copyright (c) 2018 Sawyer Bergeron, Parker Berberian, and others. +# +# All rights reserved. This program and the accompanying materials +# are made available under the terms of the Apache License, Version 2.0 +# which accompanies this distribution, and is available at +# http://www.apache.org/licenses/LICENSE-2.0 +############################################################################## + +from django.contrib import messages +from django.shortcuts import render +from django.contrib.auth.models import User +from django.utils import timezone + +import json +from datetime import timedelta + +from account.models import UserProfile +from booking.models import Booking +from workflow.models import WorkflowStep +from workflow.forms import ResourceSelectorForm, SWConfigSelectorForm, BookingMetaForm, ConfirmationForm +from resource_inventory.models import GenericResourceBundle, ResourceBundle, ConfigBundle + +class Resource_Select(WorkflowStep): + template = 'booking/steps/resource_select.html' + title = "Select Resource" + description = "Select a resource template to use for your deployment" + short_title = "pod select" + + def __init__(self, *args, **kwargs): + super(Resource_Select, self).__init__(*args, **kwargs) + self.repo_key = self.repo.SELECTED_GRESOURCE_BUNDLE + self.repo_check_key = False + self.confirm_key = "booking" + + def get_default_entry(self): + return None + + def get_context(self): + context = super(Resource_Select, self).get_context() + default = [] + chosen_bundle = None + default_bundle = self.get_default_entry() + if default_bundle: + context['disabled'] = True + chosen_bundle = default_bundle + if chosen_bundle.id: + default.append(chosen_bundle.id) + else: + default.append("repo bundle") + else: + chosen_bundle = self.repo_get(self.repo_key, False) + if chosen_bundle: + if chosen_bundle.id: + default.append(chosen_bundle.id) + else: + default.append("repo bundle") + + bundle = default_bundle + if not bundle: + bundle = chosen_bundle + edit = self.repo_get(self.repo.EDIT, False) + user = self.repo_get(self.repo.SESSION_USER) + context['form'] = ResourceSelectorForm( + data={"user": user}, + chosen_resource=default, + bundle=bundle, + edit=edit + ) + return context + + def post_render(self, request): + form = ResourceSelectorForm(request.POST) + context = self.get_context() + if form.is_valid(): + data = form.cleaned_data['generic_resource_bundle'] + irint(str(data['user'])) + data = data[2:-2] + if not data: + self.metastep.set_invalid("Please select a valid bundle") + return render(request, self.template, context) + selected_bundle = json.loads(data) + selected_id = selected_bundle[0]['id'] + gresource_bundle = None + try: + selected_id = int(selected_id) + gresource_bundle = GenericResourceBundle.objects.get(id=selected_id) + except ValueError: + # we want the bundle in the repo + gresource_bundle = self.repo_get(self.repo.GRESOURCE_BUNDLE_MODELS,{}).get("bundle", GenericResourceBundle()) + self.repo_put( + self.repo_key, + gresource_bundle + ) + confirm = self.repo_get(self.repo.CONFIRMATION) + if self.confirm_key not in confirm: + confirm[self.confirm_key] = {} + confirm[self.confirm_key]["resource name"] = gresource_bundle.name + self.repo_put(self.repo.CONFIRMATION, confirm) + messages.add_message(request, messages.SUCCESS, 'Form Validated Successfully', fail_silently=True) + self.metastep.set_valid("Step Completed") + return render(request, self.template, context) + else: + messages.add_message(request, messages.ERROR, "Form Didn't Validate", fail_silently=True) + self.metastep.set_invalid("Please complete the fields highlighted in red to continue") + return render(request, self.template, context) + +class Booking_Resource_Select(Resource_Select): + + def __init__(self, *args, **kwargs): + super(Booking_Resource_Select, self).__init__(*args, **kwargs) + self.repo_key = self.repo.BOOKING_SELECTED_GRB + self.confirm_key = "booking" + + def get_default_entry(self): + default = self.repo_get(self.repo.GRESOURCE_BUNDLE_MODELS, {}).get("bundle") + mine = self.repo_get(self.repo_key) + if mine: + return None + try: + config_bundle = self.repo_get(self.repo.BOOKING_MODELS)['booking'].config_bundle + if default: + return default # select created grb, even if preselected config bundle + return config_bundle.bundle + except: + pass + return default + + def get_context(self): + context = super(Booking_Resource_Select, self).get_context() + return context + + def post_render(self, request): + response = super(Booking_Resource_Select, self).post_render(request) + models = self.repo_get(self.repo.BOOKING_MODELS, {}) + if "booking" not in models: + models['booking'] = Booking() + booking = models['booking'] + resource = self.repo_get(self.repo_key, False) + if resource: + try: + booking.resource.template = resource + except: + booking.resource = ResourceBundle(template=resource) + models['booking'] = booking + self.repo_put(self.repo.BOOKING_MODELS, models) + return response + +class SWConfig_Select(WorkflowStep): + template = 'booking/steps/swconfig_select.html' + title = "Select Software Configuration" + description = "Choose the software and related configurations you want to have used for your deployment" + short_title = "pod config" + + def post_render(self, request): + form = SWConfigSelectorForm(request.POST) + if form.is_valid(): + + bundle_json = form.cleaned_data['software_bundle'] + bundle_json = bundle_json[2:-2] # Stupid django string bug + if not bundle_json: + self.metastep.set_invalid("Please select a valid config") + return self.render(request) + bundle_json = json.loads(bundle_json) + bundle = None + try: + id = int(bundle_json[0]['id']) + bundle = ConfigBundle.objects.get(id=id) + except ValueError: + bundle = self.repo_get(self.repo.CONFIG_MODELS).get("bundle") + + models = self.repo_get(self.repo.BOOKING_MODELS, {}) + if "booking" not in models: + models['booking'] = Booking() + models['booking'].config_bundle = bundle + self.repo_put(self.repo.BOOKING_MODELS, models) + confirm = self.repo_get(self.repo.CONFIRMATION) + if "booking" not in confirm: + confirm['booking'] = {} + confirm['booking']["configuration name"] = bundle.name + self.repo_put(self.repo.CONFIRMATION, confirm) + self.metastep.set_valid("Step Completed") + messages.add_message(request, messages.SUCCESS, 'Form Validated Successfully', fail_silently=True) + else: + self.metastep.set_invalid("Please select or create a valid config") + messages.add_message(request, messages.ERROR, "Form Didn't Validate", fail_silently=True) + + return self.render(request) + + + def get_context(self): + context = super(SWConfig_Select, self).get_context() + default = [] + bundle = None + chosen_bundle = None + created_bundle = self.repo_get(self.repo.CONFIG_MODELS, {}).get("bundle", False) + booking = self.repo_get(self.repo.BOOKING_MODELS, {}).get("booking", False) + try: + chosen_bundle = booking.config_bundle + default.append(chosen_bundle.id) + bundle=chosen_bundle + except: + if created_bundle: + default.append("repo bundle") + bundle = created_bundle + context['disabled'] = True + edit = self.repo_get(self.repo.EDIT, False) + grb = self.repo_get(self.repo.BOOKING_SELECTED_GRB) + context['form'] = SWConfigSelectorForm(chosen_software=default, bundle=bundle, edit=edit, resource=grb) + return context + +class Booking_Meta(WorkflowStep): + template = 'booking/steps/booking_meta.html' + title = "Extra Info" + description = "Tell us how long you want your booking, what it is for, and who else should have access to it" + short_title = "booking info" + + def get_context(self): + context = super(Booking_Meta, self).get_context() + initial = {} + default = [] + try: + models = self.repo_get(self.repo.BOOKING_MODELS, {}) + booking = models.get("booking") + if booking: + initial['purpose'] = booking.purpose + initial['project'] = booking.project + initial['length'] = (booking.end - booking.start).days + info = self.repo_get(self.repo.BOOKING_INFO_FILE, False) + if info: + initial['info_file'] = info + users = models.get("collaborators", []) + for user in users: + default.append(user.id) + except Exception as e: + pass + + default_user = self.repo_get(self.repo.SESSION_USER) + if default_user is None: + # TODO: error + default_user = "you" + else: + default_user = default_user.username + + context['form'] = BookingMetaForm(initial=initial, chosen_users=default, default_user=default_user) + return context + + def post_render(self, request): + form = BookingMetaForm(data=request.POST) + context = self.get_context() + + forms = self.repo_get(self.repo.BOOKING_FORMS, {}) + + forms["meta_form"] = form + self.repo_put(self.repo.BOOKING_FORMS, forms) + + if form.is_valid(): + models = self.repo_get(self.repo.BOOKING_MODELS, {}) + if "booking" not in models: + models['booking'] = Booking() + models['collaborators'] = [] + confirm = self.repo_get(self.repo.CONFIRMATION) + if "booking" not in confirm: + confirm['booking'] = {} + + models['booking'].start = timezone.now() + models['booking'].end = timezone.now() + timedelta(days=int(form.cleaned_data['length'])) + models['booking'].purpose = form.cleaned_data['purpose'] + models['booking'].project = form.cleaned_data['project'] + for key in ['length', 'project', 'purpose']: + confirm['booking'][key] = form.cleaned_data[key] + + user_data = form.cleaned_data['users'] + confirm['booking']['collaborators'] = [] + user_data = user_data[2:-2] #fixes malformed string from querydict + if user_data: + form_users = json.loads(user_data) + for user_json in form_users: + user = User.objects.get(pk=user_json['id']) + models['collaborators'].append(user) + confirm['booking']['collaborators'].append(user.username) + + info_file = form.cleaned_data.get("info_file", False) + if info_file: + self.repo_put(self.repo.BOOKING_INFO_FILE, info_file) + + self.repo_put(self.repo.BOOKING_MODELS, models) + self.repo_put(self.repo.CONFIRMATION, confirm) + messages.add_message(request, messages.SUCCESS, 'Form Validated', fail_silently=True) + self.metastep.set_valid("Step Completed") + else: + messages.add_message(request, messages.ERROR, "Form didn't validate", fail_silently=True) + self.metastep.set_invalid("Please complete the fields highlighted in red to continue") + context['form'] = form # TODO: store this form + return render(request, self.template, context) |