1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
|
##############################################################################
# Copyright (c) 2016 Max Breitenfeldt and others.
# Copyright (c) 2018 Parker Berberian, Sawyer Bergeron, and others.
#
# All rights reserved. This program and the accompanying materials
# are made available under the terms of the Apache License, Version 2.0
# which accompanies this distribution, and is available at
# http://www.apache.org/licenses/LICENSE-2.0
##############################################################################
from django.contrib import messages
from django.shortcuts import get_object_or_404
from django.http import JsonResponse, HttpResponse
from django.utils import timezone
from django.views import View
from django.views.generic import TemplateView
from django.shortcuts import redirect, render
from django.db.models import Q
from resource_inventory.models import ResourceBundle, HostProfile, Image, Host
from resource_inventory.resource_manager import ResourceManager
from account.models import Lab
from booking.models import Booking
from booking.stats import StatisticsManager
from booking.forms import HostReImageForm
from api.models import HostHardwareRelation, JobStatus
from workflow.views import login
from booking.forms import QuickBookingForm
from booking.quick_deployer import create_from_form, drop_filter
def quick_create_clear_fields(request):
request.session['quick_create_forminfo'] = None
def quick_create(request):
if not request.user.is_authenticated:
return login(request)
if request.method == 'GET':
context = {}
r_manager = ResourceManager.getInstance()
profiles = {}
for lab in Lab.objects.all():
profiles[str(lab)] = r_manager.getAvailableHostTypes(lab)
context['lab_profile_map'] = profiles
context['form'] = QuickBookingForm(initial={}, chosen_users=[], default_user=request.user.username, user=request.user)
context.update(drop_filter(request.user))
return render(request, 'booking/quick_deploy.html', context)
if request.method == 'POST':
form = QuickBookingForm(request.POST, user=request.user)
context = {}
context['lab_profile_map'] = {}
context['form'] = form
if form.is_valid():
try:
create_from_form(form, request)
except Exception as e:
messages.error(request, "Whoops, looks like an error occurred. "
"Let the admins know that you got the following message: " + str(e))
return render(request, 'workflow/exit_redirect.html', context)
messages.success(request, "We've processed your request. "
"Check Account->My Bookings for the status of your new booking")
return render(request, 'workflow/exit_redirect.html', context)
else:
messages.error(request, "Looks like the form didn't validate. Check that you entered everything correctly")
return render(request, 'booking/quick_deploy.html', context)
class BookingView(TemplateView):
template_name = "booking/booking_detail.html"
def get_context_data(self, **kwargs):
booking = get_object_or_404(Booking, id=self.kwargs['booking_id'])
title = 'Booking Details'
context = super(BookingView, self).get_context_data(**kwargs)
context.update({'title': title, 'booking': booking})
return context
class BookingDeleteView(TemplateView):
template_name = "booking/booking_delete.html"
def get_context_data(self, **kwargs):
booking = get_object_or_404(Booking, id=self.kwargs['booking_id'])
title = 'Delete Booking'
context = super(BookingDeleteView, self).get_context_data(**kwargs)
context.update({'title': title, 'booking': booking})
return context
def bookingDelete(request, booking_id):
booking = get_object_or_404(Booking, id=booking_id)
booking.delete()
messages.add_message(request, messages.SUCCESS, 'Booking deleted')
return redirect('../../../../')
class BookingListView(TemplateView):
template_name = "booking/booking_list.html"
def get_context_data(self, **kwargs):
bookings = Booking.objects.filter(end__gte=timezone.now())
title = 'Search Booking'
context = super(BookingListView, self).get_context_data(**kwargs)
context.update({'title': title, 'bookings': bookings})
return context
class ResourceBookingsJSON(View):
def get(self, request, *args, **kwargs):
resource = get_object_or_404(ResourceBundle, id=self.kwargs['resource_id'])
bookings = resource.booking_set.get_queryset().values(
'id',
'start',
'end',
'purpose',
'jira_issue_status',
'config_bundle__name'
)
return JsonResponse({'bookings': list(bookings)})
def build_image_mapping(lab, user):
mapping = {}
for profile in HostProfile.objects.filter(labs=lab):
images = Image.objects.filter(
from_lab=lab,
host_type=profile
).filter(
Q(public=True) | Q(owner=user)
)
mapping[profile.name] = [{"name": image.name, "value": image.id} for image in images]
return mapping
def booking_detail_view(request, booking_id):
user = None
if request.user.is_authenticated:
user = request.user
else:
return render(request, "dashboard/login.html", {'title': 'Authentication Required'})
booking = get_object_or_404(Booking, id=booking_id)
allowed_users = set(list(booking.collaborators.all()))
allowed_users.add(booking.owner)
if user not in allowed_users:
return render(request, "dashboard/login.html", {'title': 'This page is private'})
context = {
'title': 'Booking Details',
'booking': booking,
'pdf': booking.pdf,
'user_id': user.id,
'image_mapping': build_image_mapping(booking.lab, user)
}
return render(
request,
"booking/booking_detail.html",
context
)
def booking_modify_image(request, booking_id):
form = HostReImageForm(request.POST)
if form.is_valid():
booking = Booking.objects.get(id=booking_id)
if request.user != booking.owner:
return HttpResponse("unauthorized")
if timezone.now() > booking.end:
return HttpResponse("unauthorized")
new_image = Image.objects.get(id=form.cleaned_data['image_id'])
host = Host.objects.get(id=form.cleaned_data['host_id'])
relation = HostHardwareRelation.objects.get(host=host, job__booking=booking)
config = relation.config
config.set_image(new_image.lab_id)
config.save()
relation.status = JobStatus.NEW
relation.save()
return HttpResponse(new_image.name)
return HttpResponse("error")
def booking_stats_view(request):
return render(
request,
"booking/stats.html",
context={"data": StatisticsManager.getContinuousBookingTimeSeries(), "title": "Booking Statistics"}
)
def booking_stats_json(request):
try:
span = int(request.GET.get("days", 14))
except Exception:
span = 14
return JsonResponse(StatisticsManager.getContinuousBookingTimeSeries(span), safe=False)
|
