2 files changed, 126 insertions, 1 deletions

diff --git a/src/dashboard/admin_utils.py b/src/dashboard/admin_utils.py
new file mode 100644
index 0000000..367c613
--- /dev/null
+++ b/src/dashboard/admin_utils.py
@@ -0,0 +1,111 @@
+from resource_inventory.models import (
+    ResourceTemplate,
+    Image,
+    Server,
+    ResourceBundle,
+)
+
+from django.contrib.auth.models import User
+
+from account.models import Lab
+
+from resource_inventory.resource_manager import ResourceManager
+from resource_inventory.pdf_templater import PDFTemplater
+
+from booking.quick_deployer import update_template
+
+from datetime import timedelta
+
+from django.utils import timezone
+
+from booking.models import Booking
+from notifier.manager import NotificationHandler
+from api.models import JobFactory
+
+
+"""
+creates a quick booking using the given host
+"""
+
+
+def book_host(owner_username, host_labid, lab_username, hostname, image_id, template_name, length_days=21, collaborator_usernames=[], purpose="internal", project="LaaS"):
+    lab = Lab.objects.get(lab_user__username=lab_username)
+    host = Server.objects.filter(lab=lab).get(labid=host_labid)
+    if host.booked:
+        print("Can't book host, already marked as booked")
+        return
+    else:
+        host.booked = True
+        host.save()
+
+    template = ResourceTemplate.objects.filter(public=True).get(name=template_name)
+    image = Image.objects.get(id=image_id)
+
+    owner = User.objects.get(username=owner_username)
+
+    new_template = update_template(template, image, hostname, owner)
+
+    rmanager = ResourceManager.getInstance()
+
+    vlan_map = rmanager.get_vlans(new_template)
+
+    # only a single host so can reuse var for iter here
+    resource_bundle = ResourceBundle.objects.create(template=new_template)
+    res_configs = new_template.getConfigs()
+
+    for config in res_configs:
+        try:
+            host.bundle = resource_bundle
+            host.config = config
+            rmanager.configureNetworking(resource_bundle, host, vlan_map)
+            host.save()
+        except Exception:
+            host.booked = False
+            host.save()
+            print("Failed to book host due to error configuring it")
+            return
+
+    new_template.save()
+
+    booking = Booking.objects.create(
+        purpose=purpose,
+        project=project,
+        lab=lab,
+        owner=owner,
+        start=timezone.now(),
+        end=timezone.now() + timedelta(days=int(length_days)),
+        resource=resource_bundle,
+        opnfv_config=None
+    )
+
+    booking.pdf = PDFTemplater.makePDF(booking)
+
+    booking.save()
+
+    for collaborator_username in collaborator_usernames:
+        try:
+            user = User.objects.get(username=collaborator_username)
+            booking.collaborators.add(user)
+        except Exception:
+            print("couldn't add user with username ", collaborator_username)
+
+    booking.save()
+
+    JobFactory.makeCompleteJob(booking)
+    NotificationHandler.notify_new_booking(booking)
+
+
+def mark_working(host_labid, lab_username, working=True):
+    lab = Lab.objects.get(lab_user__username=lab_username)
+    server = Server.objects.filter(lab=lab).get(labid=host_labid)
+    print("changing server working status from ", server.working, "to", working)
+    server.working = working
+    server.save()
+
+
+def mark_booked(host_labid, lab_username, booked=True):
+    lab = Lab.objects.get(lab_user__username=lab_username)
+    server = Server.objects.filter(lab=lab).get(labid=host_labid)
+    print("changing server booked status from ", server.booked, "to", booked)
+    server.booked = booked
+    server.save()
diff --git a/src/dashboard/views.py b/src/dashboard/views.py
index 2ace2d4..f9a908c 100644
--- a/src/dashboard/views.py
+++ b/src/dashboard/views.py
@@ -12,8 +12,12 @@
 from django.shortcuts import get_object_or_404
 from django.views.generic import TemplateView
 from django.shortcuts import render
+from django.db.models import Q
+from datetime import datetime
+import pytz
 
 from account.models import Lab
+from booking.models import Booking
 
 from resource_inventory.models import Image, ResourceProfile, ResourceQuery
 from workflow.workflow_manager import ManagerTracker
@@ -65,12 +69,22 @@ def host_profile_detail_view(request):
 
 def landing_view(request):
     manager = ManagerTracker.managers.get(request.session.get('manager_session'))
+    user = request.user
+    if not user.is_anonymous:
+        bookings = Booking.objects.filter(
+            Q(owner=user) | Q(collaborators=user),
+            end__gte=datetime.now(pytz.utc)
+        )
+    else:
+        bookings = None
+
     return render(
         request,
         'dashboard/landing.html',
         {
             'manager': manager is not None,
-            'title': "Welcome to the Lab as a Service Dashboard"
+            'title': "Welcome to the Lab as a Service Dashboard",
+            'bookings': bookings
         }
     )