1 ##############################################################################
2 # Copyright (c) 2018 Sawyer Bergeron, Parker Berberian, 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
11 from django.db.models import Q
13 from workflow.forms import (
14 SearchableSelectMultipleWidget,
15 MultipleSelectFilterField,
16 MultipleSelectFilterWidget,
18 from account.models import UserProfile
19 from resource_inventory.models import Image, Installer, Scenario
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)
28 installer = forms.ModelChoiceField(queryset=Installer.objects.all(), required=False)
29 scenario = forms.ModelChoiceField(queryset=Scenario.objects.all(), required=False)
31 def __init__(self, data=None, *args, user=None, **kwargs):
33 if "default_user" in kwargs:
34 default_user = kwargs.pop("default_user")
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")
44 self.image = forms.ModelChoiceField(queryset=Image.objects.filter(
45 Q(public=True) | Q(owner=user)), required=False)
47 self.image = forms.ModelChoiceField(queryset=Image.objects.all(), required=False)
49 super(QuickBookingForm, self).__init__(data=data, **kwargs)
51 self.fields['users'] = forms.CharField(
52 widget=SearchableSelectMultipleWidget(
53 attrs=self.build_search_widget_attrs(chosen_users, default_user=default_user)
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(
71 def build_user_list(self):
73 returns a mapping of UserProfile ids to displayable objects expected by
74 searchable multiple select widget
78 d_qset = UserProfile.objects.select_related('user').all().exclude(user__username=self.default_user)
79 for userprofile in d_qset:
81 'id': userprofile.user.id,
82 'expanded_name': userprofile.full_name,
83 'small_name': userprofile.user.username,
84 'string': userprofile.email_addr
87 users[userprofile.user.id] = user
93 def build_search_widget_attrs(self, chosen_users, default_user="you"):
96 'set': self.build_user_list(),
97 'show_from_noentry': "false",
99 'scrollable': "false",
100 'selectable_limit': -1,
102 'placeholder': "username",
103 'initial': chosen_users,