1 files changed, 121 insertions, 0 deletions
diff --git a/src/dashboard/views.py b/src/dashboard/views.py
new file mode 100644
index 0000000..aaad7ab
--- /dev/null
+++ b/src/dashboard/views.py
@@ -0,0 +1,121 @@
+##############################################################################
+# 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.shortcuts import get_object_or_404
+from django.views.generic import TemplateView
+from django.shortcuts import render
+from django.http import HttpResponseRedirect
+
+from account.models import Lab
+
+from resource_inventory.models import Image, HostProfile
+from workflow.views import create_session
+from workflow.workflow_manager import ManagerTracker
+
+
+def lab_list_view(request):
+    labs = Lab.objects.all()
+    context = {"labs": labs}
+
+    return render(request, "dashboard/lab_list.html", context)
+
+
+def lab_detail_view(request, lab_name):
+    user = None
+    if request.user.is_authenticated:
+        user = request.user
+
+    lab = get_object_or_404(Lab, name=lab_name)
+
+    images = Image.objects.filter(from_lab=lab).filter(public=True)
+    if user:
+        images = images | Image.objects.filter(from_lab=lab).filter(owner=user)
+
+    return render(
+        request,
+        "dashboard/lab_detail.html",
+        {
+            'title': "Lab Overview",
+            'lab': lab,
+            'hostprofiles': lab.hostprofiles.all(),
+            'images': images,
+        }
+    )
+
+
+def host_profile_detail_view(request):
+
+    return render(
+        request,
+        "dashboard/host_profile_detail.html",
+        {
+            'title': "Host Types",
+        }
+    )
+
+
+def landing_view(request):
+    manager = None
+    manager_detected = False
+    if 'manager_session' in request.session:
+
+        try:
+            manager = ManagerTracker.managers[request.session['manager_session']]
+
+        except KeyError:
+            pass
+
+    if manager is not None:
+        # no manager detected, don't display continue button
+        manager_detected = True
+
+    if request.method == 'GET':
+        return render(request, 'dashboard/landing.html', {'manager': manager_detected, 'title': "Welcome to the Lab as a Service Dashboard"})
+
+    if request.method == 'POST':
+        try:
+            create = request.POST['create']
+
+            if manager is not None:
+                del manager
+
+            mgr_uuid = create_session(create, request=request,)
+            request.session['manager_session'] = mgr_uuid
+            return HttpResponseRedirect('/wf/')
+
+        except KeyError:
+            pass
+
+
+class LandingView(TemplateView):
+    template_name = "dashboard/landing.html"
+
+    def get_context_data(self, **kwargs):
+        context = super(LandingView, self).get_context_data(**kwargs)
+
+        hosts = []
+
+        for host_profile in HostProfile.objects.all():
+            name = host_profile.name
+            description = host_profile.description
+            in_labs = host_profile.labs
+
+            interfaces = host_profile.interfaceprofile
+            storage = host_profile.storageprofile
+            cpu = host_profile.cpuprofile
+            ram = host_profile.ramprofile
+
+            host = (name, description, in_labs, interfaces, storage, cpu, ram)
+            hosts.append(host)
+
+        context.update({'hosts': hosts})
+
+        return context