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