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)
15 from account.models import UserProfile
16 from resource_inventory.models import Image, Installer, Scenario
17 from workflow.forms import SearchableSelectMultipleField
18 from booking.lib import get_user_items, get_user_field_opts
21 class QuickBookingForm(forms.Form):
22 purpose = forms.CharField(max_length=1000)
23 project = forms.CharField(max_length=400)
24 hostname = forms.CharField(required=False, 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, lab_data=None, *args, **kwargs):
30 if "default_user" in kwargs:
31 default_user = kwargs.pop("default_user")
34 self.default_user = default_user
36 super(QuickBookingForm, self).__init__(data=data, **kwargs)
38 image_help_text = 'Image can be set only for single-node bookings. For multi-node bookings set image through Design a POD.'
39 self.fields["image"] = forms.ModelChoiceField(
40 Image.objects.filter(public=True) | Image.objects.filter(owner=user), required=False
43 self.fields['image'].widget.attrs.update({
44 'class': 'has-popover',
45 'data-content': image_help_text,
46 'data-placement': 'bottom',
47 'data-container': 'body'
50 self.fields['users'] = SearchableSelectMultipleField(
51 queryset=UserProfile.objects.filter(public_user=True).select_related('user').exclude(user=user),
52 items=get_user_items(exclude=user),
54 **get_user_field_opts()
57 self.fields['length'] = forms.IntegerField(
68 self.fields['filter_field'] = MultipleSelectFilterField(widget=MultipleSelectFilterWidget(**lab_data))
70 hostname_help_text = 'Hostname can be set only for single-node bookings. For multi-node bookings set hostname through Design a POD.'
71 self.fields['hostname'].widget.attrs.update({
72 'class': 'has-popover',
73 'data-content': hostname_help_text,
74 'data-placement': 'top',
75 'data-container': 'body'
78 def build_user_list(self):
80 Build list of UserProfiles.
82 returns a mapping of UserProfile ids to displayable objects expected by
83 searchable multiple select widget
87 d_qset = UserProfile.objects.select_related('user').all().exclude(user__username=self.default_user)
88 for userprofile in d_qset:
90 'id': userprofile.user.id,
91 'expanded_name': userprofile.full_name,
92 'small_name': userprofile.user.username,
93 'string': userprofile.email_addr
96 users[userprofile.user.id] = user
102 def build_search_widget_attrs(self, chosen_users, default_user="you"):
105 'set': self.build_user_list(),
106 'show_from_noentry': "false",
107 'show_x_results': 10,
108 'scrollable': "false",
109 'selectable_limit': -1,
111 'placeholder': "username",
112 'initial': chosen_users,
118 class HostReImageForm(forms.Form):
120 image_id = forms.IntegerField()
121 host_id = forms.IntegerField()