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     SearchableSelectMultipleWidget,
  14     MultipleSelectFilterField,
  15     MultipleSelectFilterWidget,
  16     FormUtils)
  17 from account.models import UserProfile
  18 from resource_inventory.models import Image, Installer, Scenario
  19
  20
  21 class QuickBookingForm(forms.Form):
  22     purpose = forms.CharField(max_length=1000)
  23     project = forms.CharField(max_length=400)
  24     hostname = forms.CharField(max_length=400)
  25
  26     installer = forms.ModelChoiceField(queryset=Installer.objects.all(), required=False)
  27     scenario = forms.ModelChoiceField(queryset=Scenario.objects.all(), required=False)
  28
  29     def __init__(self, data=None, user=None, *args, **kwargs):
  30         chosen_users = []
  31         if "default_user" in kwargs:
  32             default_user = kwargs.pop("default_user")
  33         else:
  34             default_user = "you"
  35         self.default_user = default_user
  36         if "chosen_users" in kwargs:
  37             chosen_users = kwargs.pop("chosen_users")
  38         elif data and "users" in data:
  39             chosen_users = data.getlist("users")
  40
  41         super(QuickBookingForm, self).__init__(data=data, **kwargs)
  42
  43         self.fields["image"] = forms.ModelChoiceField(
  44             queryset=Image.objects.difference(
  45                 Image.objects.filter(public=False).difference(Image.objects.filter(owner=user))
  46             )
  47         )
  48
  49         self.fields['users'] = forms.CharField(
  50             widget=SearchableSelectMultipleWidget(
  51                 attrs=self.build_search_widget_attrs(chosen_users, default_user=default_user)
  52             ),
  53             required=False
  54         )
  55         attrs = FormUtils.getLabData(0)
  56         attrs['selection_data'] = 'false'
  57         self.fields['filter_field'] = MultipleSelectFilterField(widget=MultipleSelectFilterWidget(attrs=attrs))
  58         self.fields['length'] = forms.IntegerField(
  59             widget=NumberInput(
  60                 attrs={
  61                     "type": "range",
  62                     'min': "1",
  63                     "max": "21",
  64                     "value": "1"
  65                 }
  66             )
  67         )
  68
  69     def build_user_list(self):
  70         """
  71         returns a mapping of UserProfile ids to displayable objects expected by
  72         searchable multiple select widget
  73         """
  74         try:
  75             users = {}
  76             d_qset = UserProfile.objects.select_related('user').all().exclude(user__username=self.default_user)
  77             for userprofile in d_qset:
  78                 user = {
  79                     'id': userprofile.user.id,
  80                     'expanded_name': userprofile.full_name,
  81                     'small_name': userprofile.user.username,
  82                     'string': userprofile.email_addr
  83                 }
  84
  85                 users[userprofile.user.id] = user
  86
  87             return users
  88         except Exception:
  89             pass
  90
  91     def build_search_widget_attrs(self, chosen_users, default_user="you"):
  92
  93         attrs = {
  94             'set': self.build_user_list(),
  95             'show_from_noentry': "false",
  96             'show_x_results': 10,
  97             'scrollable': "false",
  98             'selectable_limit': -1,
  99             'name': "users",
 100             'placeholder': "username",
 101             'initial': chosen_users,
 102             'edit': False
 103         }
 104         return attrs
 105
 106
 107 class HostReImageForm(forms.Form):
 108
 109     image_id = forms.IntegerField()
 110     host_id = forms.IntegerField()