1 ##############################################################################
2 # Copyright (c) 2018 Parker Berberian, Sawyer Bergeron, and others.
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
12 from workflow.forms import (
13 SearchableSelectMultipleWidget,
14 MultipleSelectFilterField,
15 MultipleSelectFilterWidget,
17 from account.models import UserProfile
18 from resource_inventory.models import Image, Installer, Scenario
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)
26 installer = forms.ModelChoiceField(queryset=Installer.objects.all(), required=False)
27 scenario = forms.ModelChoiceField(queryset=Scenario.objects.all(), required=False)
29 def __init__(self, data=None, user=None, *args, **kwargs):
31 if "default_user" in kwargs:
32 default_user = kwargs.pop("default_user")
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")
41 super(QuickBookingForm, self).__init__(data=data, **kwargs)
43 self.fields["image"] = forms.ModelChoiceField(
44 queryset=Image.objects.difference(
45 Image.objects.filter(public=False).difference(Image.objects.filter(owner=user))
49 self.fields['users'] = forms.CharField(
50 widget=SearchableSelectMultipleWidget(
51 attrs=self.build_search_widget_attrs(chosen_users, default_user=default_user)
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(
69 def build_user_list(self):
71 returns a mapping of UserProfile ids to displayable objects expected by
72 searchable multiple select widget
76 d_qset = UserProfile.objects.select_related('user').all().exclude(user__username=self.default_user)
77 for userprofile in d_qset:
79 'id': userprofile.user.id,
80 'expanded_name': userprofile.full_name,
81 'small_name': userprofile.user.username,
82 'string': userprofile.email_addr
85 users[userprofile.user.id] = user
91 def build_search_widget_attrs(self, chosen_users, default_user="you"):
94 'set': self.build_user_list(),
95 'show_from_noentry': "false",
97 'scrollable': "false",
98 'selectable_limit': -1,
100 'placeholder': "username",
101 'initial': chosen_users,
107 class HostReImageForm(forms.Form):
109 image_id = forms.IntegerField()
110 host_id = forms.IntegerField()