Merge "Add Quick-Booking Workflow"

[laas.git] / src / booking / views.py

diff --git a/src/booking/views.py b/src/booking/views.py
index 9b9860f..bc1d2c9 100644 (file)
--- a/src/booking/views.py
+++ b/src/booking/views.py
@@ -11,34 +11,64 @@
 from django.contrib import messages
 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 django.shortcuts import redirect, render
-import json
 
-from resource_inventory.models import ResourceBundle
+from account.models import Lab
 from resource_inventory.resource_manager import ResourceManager
-from booking.models import Booking, Installer, Opsys
+from resource_inventory.models import ResourceBundle
+from booking.models import Booking
 from booking.stats import StatisticsManager
+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 = {}
 
-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)
+        r_manager = ResourceManager.getInstance()
+        profiles = {}
+        for lab in Lab.objects.all():
+            profiles[str(lab)] = r_manager.getAvailableHostTypes(lab)
 
-    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['lab_profile_map'] = profiles
 
-    context.update({'installer_filter': json.dumps(installer_filter), 'scenario_filter': json.dumps(scenario_filter)})
+        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):
@@ -103,27 +133,33 @@ def booking_detail_view(request, booking_id):
         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})
+    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'})
+
+    return render(
+        request,
+        "booking/booking_detail.html",
+        {
+            'title': 'Booking Details',
+            'booking': booking,
+            'pdf': booking.pdf,
+            'user_id': user.id
+        })
 
 
 def booking_stats_view(request):
     return render(
-            request,
-            "booking/stats.html",
-            context={
-                "data": StatisticsManager.getContinuousBookingTimeSeries(),
-                "title": "Booking Statistics"
-                }
-            )
+        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:
+    except Exception:
         span = 14
     return JsonResponse(StatisticsManager.getContinuousBookingTimeSeries(span), safe=False)