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 MultipleSelectFilterField,
14 MultipleSelectFilterWidget,
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
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)
27 installer = forms.ModelChoiceField(queryset=Installer.objects.all(), required=False)
28 scenario = forms.ModelChoiceField(queryset=Scenario.objects.all(), required=False)
30 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
37 super(QuickBookingForm, self).__init__(data=data, **kwargs)
39 self.fields["image"] = forms.ModelChoiceField(
40 Image.objects.filter(public=True) | Image.objects.filter(owner=user)
43 self.fields['users'] = SearchableSelectMultipleField(
44 queryset=UserProfile.objects.select_related('user').exclude(user=user),
45 items=get_user_items(exclude=user),
47 **get_user_field_opts()
50 attrs = FormUtils.getLabData(0)
51 self.fields['filter_field'] = MultipleSelectFilterField(widget=MultipleSelectFilterWidget(**attrs))
52 self.fields['length'] = forms.IntegerField(
63 def build_user_list(self):
65 returns a mapping of UserProfile ids to displayable objects expected by
66 searchable multiple select widget
70 d_qset = UserProfile.objects.select_related('user').all().exclude(user__username=self.default_user)
71 for userprofile in d_qset:
73 'id': userprofile.user.id,
74 'expanded_name': userprofile.full_name,
75 'small_name': userprofile.user.username,
76 'string': userprofile.email_addr
79 users[userprofile.user.id] = user
85 def build_search_widget_attrs(self, chosen_users, default_user="you"):
88 'set': self.build_user_list(),
89 'show_from_noentry': "false",
91 'scrollable': "false",
92 'selectable_limit': -1,
94 'placeholder': "username",
95 'initial': chosen_users,
101 class HostReImageForm(forms.Form):
103 image_id = forms.IntegerField()
104 host_id = forms.IntegerField()