aboutsummaryrefslogtreecommitdiffstats
path: root/src/booking/views.py
diff options
context:
space:
mode:
Diffstat (limited to 'src/booking/views.py')
-rw-r--r--src/booking/views.py138
1 files changed, 84 insertions, 54 deletions
diff --git a/src/booking/views.py b/src/booking/views.py
index a52cfe2..c139b4c 100644
--- a/src/booking/views.py
+++ b/src/booking/views.py
@@ -1,5 +1,6 @@
##############################################################################
# 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
@@ -7,39 +8,38 @@
# http://www.apache.org/licenses/LICENSE-2.0
##############################################################################
-from django.conf import settings
from django.contrib import messages
-from django.contrib.auth.mixins import LoginRequiredMixin
-from django.http import JsonResponse
from django.shortcuts import get_object_or_404
+from django.http import JsonResponse
from django.urls import reverse
from django.utils import timezone
from django.views import View
from django.views.generic import FormView
from django.views.generic import TemplateView
-from jira import JIRAError
-from django.shortcuts import redirect
+from django.shortcuts import redirect, render
+import json
-from account.jira_util import get_jira
from booking.forms import BookingForm, BookingEditForm
-from booking.models import Booking
-from dashboard.models import Resource
-
-def create_jira_ticket(user, booking):
- jira = get_jira(user)
- issue_dict = {
- 'project': 'PHAROS',
- 'summary': str(booking.resource) + ': Access Request',
- 'description': booking.purpose,
- 'issuetype': {'name': 'Task'},
- 'components': [{'name': 'POD Access Request'}],
- 'assignee': {'name': booking.resource.owner.username}
- }
- issue = jira.create_issue(fields=issue_dict)
- jira.add_attachment(issue, user.userprofile.pgp_public_key)
- jira.add_attachment(issue, user.userprofile.ssh_public_key)
- booking.jira_issue_id = issue.id
- booking.save()
+from resource_inventory.models import ResourceBundle
+from resource_inventory.resource_manager import ResourceManager
+from booking.models import Booking, Installer, Opsys
+from booking.stats import StatisticsManager
+
+
+def drop_filter(context):
+ installer_filter = {}
+ for os in Opsys.objects.all():
+ installer_filter[os.id] = []
+ for installer in os.sup_installers.all():
+ installer_filter[os.id].append(installer.id)
+
+ scenario_filter = {}
+ for installer in Installer.objects.all():
+ scenario_filter[installer.id] = []
+ for scenario in installer.sup_scenarios.all():
+ scenario_filter[installer.id].append(scenario.id)
+
+ context.update({'installer_filter': json.dumps(installer_filter), 'scenario_filter': json.dumps(scenario_filter)})
class BookingFormView(FormView):
@@ -47,14 +47,16 @@ class BookingFormView(FormView):
form_class = BookingForm
def dispatch(self, request, *args, **kwargs):
- self.resource = get_object_or_404(Resource, id=self.kwargs['resource_id'])
+ self.resource = get_object_or_404(ResourceBundle, id=self.kwargs['resource_id'])
return super(BookingFormView, self).dispatch(request, *args, **kwargs)
def get_context_data(self, **kwargs):
- title = 'Booking: ' + self.resource.name
+ title = 'Booking: ' + str(self.resource.id)
context = super(BookingFormView, self).get_context_data(**kwargs)
context.update({'title': title, 'resource': self.resource})
- #raise PermissionDenied('check')
+
+ drop_filter(context)
+
return context
def get_success_url(self):
@@ -75,24 +77,16 @@ class BookingFormView(FormView):
booking = Booking(start=form.cleaned_data['start'],
end=form.cleaned_data['end'],
purpose=form.cleaned_data['purpose'],
- opsys=form.cleaned_data['opsys'],
installer=form.cleaned_data['installer'],
scenario=form.cleaned_data['scenario'],
- resource=self.resource, user=user)
+ resource=self.resource,
+ owner=user
+ )
try:
booking.save()
except ValueError as err:
messages.add_message(self.request, messages.ERROR, err)
return super(BookingFormView, self).form_invalid(form)
- try:
- if settings.CREATE_JIRA_TICKET:
- create_jira_ticket(user, booking)
- except JIRAError:
- messages.add_message(self.request, messages.ERROR, 'Failed to create Jira Ticket. '
- 'Please check your Jira '
- 'permissions.')
- booking.delete()
- return super(BookingFormView, self).form_invalid(form)
messages.add_message(self.request, messages.SUCCESS, 'Booking saved')
return super(BookingFormView, self).form_valid(form)
@@ -105,7 +99,7 @@ class BookingEditFormView(FormView):
return True
def dispatch(self, request, *args, **kwargs):
- self.resource = get_object_or_404(Resource, id=self.kwargs['resource_id'])
+ self.resource = get_object_or_404(ResourceBundle, id=self.kwargs['resource_id'])
self.original_booking = get_object_or_404(Booking, id=self.kwargs['booking_id'])
return super(BookingEditFormView, self).dispatch(request, *args, **kwargs)
@@ -113,6 +107,9 @@ class BookingEditFormView(FormView):
title = 'Editing Booking on: ' + self.resource.name
context = super(BookingEditFormView, self).get_context_data(**kwargs)
context.update({'title': title, 'resource': self.resource, 'booking': self.original_booking})
+
+ drop_filter(context)
+
return context
def get_form_kwargs(self):
@@ -120,14 +117,6 @@ class BookingEditFormView(FormView):
kwargs['purpose'] = self.original_booking.purpose
kwargs['start'] = self.original_booking.start
kwargs['end'] = self.original_booking.end
- try:
- kwargs['installer'] = self.original_booking.installer
- except AttributeError:
- pass
- try:
- kwargs['scenario'] = self.original_booking.scenario
- except AttributeError:
- pass
return kwargs
def get_success_url(self):
@@ -145,7 +134,7 @@ class BookingEditFormView(FormView):
'You are not the owner of this booking.')
return super(BookingEditFormView, self).form_invalid(form)
- #Do Conflict Checks
+ # Do Conflict Checks
if self.original_booking.end != form.cleaned_data['end']:
if form.cleaned_data['end'] - self.original_booking.end > timezone.timedelta(days=7):
messages.add_message(self.request, messages.ERROR,
@@ -176,13 +165,12 @@ class BookingEditFormView(FormView):
messages.add_message(self.request, messages.ERROR, err)
return super(BookingEditFormView, self).form_invalid(form)
- user = self.request.user
return super(BookingEditFormView, self).form_valid(form)
+
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'
@@ -190,6 +178,7 @@ class BookingView(TemplateView):
context.update({'title': title, 'booking': booking})
return context
+
class BookingDeleteView(TemplateView):
template_name = "booking/booking_delete.html"
@@ -200,12 +189,14 @@ class BookingDeleteView(TemplateView):
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"
@@ -219,8 +210,47 @@ class BookingListView(TemplateView):
class ResourceBookingsJSON(View):
def get(self, request, *args, **kwargs):
- resource = get_object_or_404(Resource, id=self.kwargs['resource_id'])
- bookings = resource.booking_set.get_queryset().values('id', 'start', 'end', 'purpose',
- 'jira_issue_status', 'opsys__name',
- 'installer__name', 'scenario__name')
+ 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 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)
+ return render(request, "booking/booking_detail.html", {
+ 'title': 'Booking Details',
+ 'booking': booking,
+ 'pdf': ResourceManager().makePDF(booking.resource),
+ 'user_id': user.id})
+
+
+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:
+ span = 14
+ return JsonResponse(StatisticsManager.getContinuousBookingTimeSeries(span), safe=False)