Improve admin utils

Change-Id: I03e80a026191e96c63c3c47d2d20b93dc944ad98
Signed-off-by: Sawyer Bergeron <sbergeron@iol.unh.edu>

1 files changed, 53 insertions, 3 deletions

diff --git a/src/dashboard/admin_utils.py b/src/dashboard/admin_utils.py
index 38e8955..fd427f5 100644
--- a/src/dashboard/admin_utils.py
+++ b/src/dashboard/admin_utils.py
@@ -24,13 +24,17 @@ from booking.models import Booking
 from notifier.manager import NotificationHandler
 from api.models import JobFactory
 
+from api.models import JobStatus
 
-"""
-creates a quick booking using the given host
-"""
+
+def print_div():
+    print("====================================================================")
 
 
 def book_host(owner_username, host_labid, lab_username, hostname, image_id, template_name, length_days=21, collaborator_usernames=[], purpose="internal", project="LaaS"):
+    """
+    creates a quick booking using the given host
+    """
     lab = Lab.objects.get(lab_user__username=lab_username)
     host = Server.objects.filter(lab=lab).get(labid=host_labid)
     if host.booked:
@@ -157,3 +161,49 @@ def map_cntt_interfaces(labid: str):
         iface.profile = new_ifprofile
 
         iface.save()
+
+
+def detect_leaked_hosts(labid="unh_iol"):
+    """
+    Use this to try to detect leaked hosts.
+    These hosts may still be in the process of unprovisioning,
+    but if they are not (or unprovisioning is frozen) then
+    these hosts are instead leaked
+    """
+    working_servers = Server.objects.filter(working=True, lab__lab_user__username=labid)
+    booked = working_servers.filter(booked=True)
+    filtered = booked
+    print_div()
+    print("In use now:")
+    for booking in Booking.objects.filter(end__gte=timezone.now()):
+        res_for_booking = booking.resource.get_resources()
+        print(res_for_booking)
+        for resource in res_for_booking:
+            filtered = filtered.exclude(id=resource.id)
+    print_div()
+    print("Possibly leaked:")
+    for host in filtered:
+        print(host)
+    print_div()
+    return filtered
+
+
+def booking_for_host(host_labid: str, labid="unh_iol"):
+    server = Server.objects.get(lab__lab_user__username=labid, labid=host_labid)
+    booking = server.bundle.booking_set.first()
+    print_div()
+    print(booking)
+    print("id:", booking.id)
+    print("owner:", booking.owner)
+    print("job (id):", booking.job, "(" + str(booking.job.id) + ")")
+    print_div()
+    return booking
+
+
+def force_release_booking(booking_id):
+    booking = Booking.objects.get(id=booking_id)
+    job = booking.job
+    tasks = job.get_tasklist()
+    for task in tasks:
+        task.status = JobStatus.DONE
+        task.save()