src/booking/views.py

   1 ##############################################################################
   2 # Copyright (c) 2016 Max Breitenfeldt and others.
   3 # Copyright (c) 2018 Parker Berberian, Sawyer Bergeron, and others.
   4 #
   5 # All rights reserved. This program and the accompanying materials
   6 # are made available under the terms of the Apache License, Version 2.0
   7 # which accompanies this distribution, and is available at
   8 # http://www.apache.org/licenses/LICENSE-2.0
   9 ##############################################################################
  10
  11 from django.contrib import messages
  12 from django.shortcuts import get_object_or_404
  13 from django.http import JsonResponse
  14 from django.urls import reverse
  15 from django.utils import timezone
  16 from django.views import View
  17 from django.views.generic import FormView
  18 from django.views.generic import TemplateView
  19 from django.shortcuts import redirect, render
  20 import json
  21
  22 from resource_inventory.models import ResourceBundle
  23 from resource_inventory.resource_manager import ResourceManager
  24 from booking.models import Booking, Installer, Opsys
  25 from booking.stats import StatisticsManager
  26
  27
  28 def drop_filter(context):
  29     installer_filter = {}
  30     for os in Opsys.objects.all():
  31         installer_filter[os.id] = []
  32         for installer in os.sup_installers.all():
  33             installer_filter[os.id].append(installer.id)
  34
  35     scenario_filter = {}
  36     for installer in Installer.objects.all():
  37         scenario_filter[installer.id] = []
  38         for scenario in installer.sup_scenarios.all():
  39             scenario_filter[installer.id].append(scenario.id)
  40
  41     context.update({'installer_filter': json.dumps(installer_filter), 'scenario_filter': json.dumps(scenario_filter)})
  42
  43
  44 class BookingView(TemplateView):
  45     template_name = "booking/booking_detail.html"
  46
  47     def get_context_data(self, **kwargs):
  48         booking = get_object_or_404(Booking, id=self.kwargs['booking_id'])
  49         title = 'Booking Details'
  50         context = super(BookingView, self).get_context_data(**kwargs)
  51         context.update({'title': title, 'booking': booking})
  52         return context
  53
  54
  55 class BookingDeleteView(TemplateView):
  56     template_name = "booking/booking_delete.html"
  57
  58     def get_context_data(self, **kwargs):
  59         booking = get_object_or_404(Booking, id=self.kwargs['booking_id'])
  60         title = 'Delete Booking'
  61         context = super(BookingDeleteView, self).get_context_data(**kwargs)
  62         context.update({'title': title, 'booking': booking})
  63         return context
  64
  65
  66 def bookingDelete(request, booking_id):
  67     booking = get_object_or_404(Booking, id=booking_id)
  68     booking.delete()
  69     messages.add_message(request, messages.SUCCESS, 'Booking deleted')
  70     return redirect('../../../../')
  71
  72
  73 class BookingListView(TemplateView):
  74     template_name = "booking/booking_list.html"
  75
  76     def get_context_data(self, **kwargs):
  77         bookings = Booking.objects.filter(end__gte=timezone.now())
  78         title = 'Search Booking'
  79         context = super(BookingListView, self).get_context_data(**kwargs)
  80         context.update({'title': title, 'bookings': bookings})
  81         return context
  82
  83
  84 class ResourceBookingsJSON(View):
  85     def get(self, request, *args, **kwargs):
  86         resource = get_object_or_404(ResourceBundle, id=self.kwargs['resource_id'])
  87         bookings = resource.booking_set.get_queryset().values(
  88             'id',
  89             'start',
  90             'end',
  91             'purpose',
  92             'jira_issue_status',
  93             'config_bundle__name'
  94         )
  95         return JsonResponse({'bookings': list(bookings)})
  96
  97
  98 def booking_detail_view(request, booking_id):
  99     user = None
 100     if request.user.is_authenticated:
 101         user = request.user
 102     else:
 103         return render(request, "dashboard/login.html", {'title': 'Authentication Required'})
 104
 105     booking = get_object_or_404(Booking, id=booking_id)
 106     return render(request, "booking/booking_detail.html", {
 107         'title': 'Booking Details',
 108         'booking': booking,
 109         'pdf': ResourceManager().makePDF(booking.resource),
 110         'user_id': user.id})
 111
 112
 113 def booking_stats_view(request):
 114     return render(
 115             request,
 116             "booking/stats.html",
 117             context={
 118                 "data": StatisticsManager.getContinuousBookingTimeSeries(),
 119                 "title": "Booking Statistics"
 120                 }
 121             )
 122
 123
 124 def booking_stats_json(request):
 125     try:
 126         span = int(request.GET.get("days", 14))
 127     except:
 128         span = 14
 129     return JsonResponse(StatisticsManager.getContinuousBookingTimeSeries(span), safe=False)