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