src/booking/forms.py

   1 ##############################################################################
   2 # Copyright (c) 2018 Parker Berberian, Sawyer Bergeron, 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 import django.forms as forms
  10 from django.forms.widgets import NumberInput
  11
  12 from workflow.forms import (
  13     MultipleSelectFilterField,
  14     MultipleSelectFilterWidget)
  15 from account.models import UserProfile
  16 from resource_inventory.models import Image, Installer, Scenario
  17 from workflow.forms import SearchableSelectMultipleField
  18 from booking.lib import get_user_items, get_user_field_opts
  19
  20
  21 class QuickBookingForm(forms.Form):
  22     # Django Form class for Express Booking
  23     purpose = forms.CharField(max_length=1000)
  24     project = forms.CharField(max_length=400)
  25     hostname = forms.CharField(required=False, max_length=400)
  26     global_cloud_config = forms.CharField(widget=forms.Textarea, required=False)
  27
  28     installer = forms.ModelChoiceField(queryset=Installer.objects.all(), required=False)
  29     scenario = forms.ModelChoiceField(queryset=Scenario.objects.all(), required=False)
  30
  31     def __init__(self, data=None, user=None, lab_data=None, *args, **kwargs):
  32         if "default_user" in kwargs:
  33             default_user = kwargs.pop("default_user")
  34         else:
  35             default_user = "you"
  36         self.default_user = default_user
  37
  38         super(QuickBookingForm, self).__init__(data=data, **kwargs)
  39
  40         image_help_text = 'Image can be set only for single-node bookings. For multi-node bookings set image through Design a POD.'
  41         self.fields["image"] = forms.ModelChoiceField(
  42             Image.objects.filter(public=True) | Image.objects.filter(owner=user), required=False
  43         )
  44
  45         self.fields['image'].widget.attrs.update({
  46             'class': 'has-popover',
  47             'data-content': image_help_text,
  48             'data-placement': 'bottom',
  49             'data-container': 'body'
  50         })
  51
  52         self.fields['users'] = SearchableSelectMultipleField(
  53             queryset=UserProfile.objects.filter(public_user=True).select_related('user').exclude(user=user),
  54             items=get_user_items(exclude=user),
  55             required=False,
  56             **get_user_field_opts()
  57         )
  58
  59         self.fields['length'] = forms.IntegerField(
  60             widget=NumberInput(
  61                 attrs={
  62                     "type": "range",
  63                     'min': "1",
  64                     "max": "21",
  65                     "value": "1"
  66                 }
  67             )
  68         )
  69
  70         self.fields['filter_field'] = MultipleSelectFilterField(widget=MultipleSelectFilterWidget(**lab_data))
  71
  72         hostname_help_text = 'Hostname can be set only for single-node bookings. For multi-node bookings set hostname through Design a POD.'
  73         self.fields['hostname'].widget.attrs.update({
  74             'class': 'has-popover',
  75             'data-content': hostname_help_text,
  76             'data-placement': 'top',
  77             'data-container': 'body'
  78         })
  79
  80     def build_user_list(self):
  81         """
  82         Build list of UserProfiles.
  83
  84         returns a mapping of UserProfile ids to displayable objects expected by
  85         searchable multiple select widget
  86         """
  87         try:
  88             users = {}
  89             d_qset = UserProfile.objects.select_related('user').all().exclude(user__username=self.default_user)
  90             for userprofile in d_qset:
  91                 user = {
  92                     'id': userprofile.user.id,
  93                     'expanded_name': userprofile.full_name,
  94                     'small_name': userprofile.user.username,
  95                     'string': userprofile.email_addr
  96                 }
  97
  98                 users[userprofile.user.id] = user
  99
 100             return users
 101         except Exception:
 102             pass
 103
 104     def build_search_widget_attrs(self, chosen_users, default_user="you"):
 105
 106         attrs = {
 107             'set': self.build_user_list(),
 108             'show_from_noentry': "false",
 109             'show_x_results': 10,
 110             'scrollable': "false",
 111             'selectable_limit': -1,
 112             'name': "users",
 113             'placeholder': "username",
 114             'initial': chosen_users,
 115             'edit': False
 116         }
 117         return attrs
 118
 119
 120 class HostReImageForm(forms.Form):
 121
 122     image_id = forms.IntegerField()
 123     host_id = forms.IntegerField()