From d4e6a67d423aee2144000871cc698195c12c943f Mon Sep 17 00:00:00 2001
From: pepper <git@pooper.net.au>
Date: Wed, 20 Jan 2021 22:20:39 -0500
Subject: [PATCH] Added dateRange to timeslot details

---
 migratorapi/api/views.py | 40 ++++++++++++++++++++++++++++++++--------
 1 file changed, 32 insertions(+), 8 deletions(-)

diff --git a/migratorapi/api/views.py b/migratorapi/api/views.py
index 6b4fdaa..afa26af 100644
--- a/migratorapi/api/views.py
+++ b/migratorapi/api/views.py
@@ -24,7 +24,13 @@ def checkLimit(limit):
         limit = int(limit)
     return limit
 
-def checkDays(days):
+def checkValue(value):
+    if value:
+        return True
+    else:
+        return False
+
+def checkDayNumber(days):
     if days == "":
         days = 0
     else:
@@ -104,7 +110,7 @@ class MigrationsWaitingTerm(APIView):
     Returns list of booked migrations
     '''
     def get(self, request, format=None):
-        days = checkDays(request.GET.get('days', ''))
+        days = checkDayNumber(request.GET.get('days', ''))
         limit = checkLimit(request.GET.get('limit', ''))
         migrations = MigrationSerializer(Migration.objects.filter(migration_status="Waiting Termination", term_date=datetime.datetime.now().date() + datetime.timedelta(days))[:limit], many=True)
         return Response(migrations.data, status=status.HTTP_200_OK)
@@ -125,7 +131,7 @@ class PendingMigrations(APIView):
     are booked for today (maybe before timezone.now))
     '''
     def get(self, request, format=None):
-        days = checkDays(request.GET.get('days', ''))
+        days = checkDayNumber(request.GET.get('days', ''))
         limit = checkLimit(request.GET.get('limit', ''))
         migrations = Migration.objects.filter(
             migration_status="Booked", booked_date=datetime.datetime.now().date()  + datetime.timedelta(days)
@@ -166,13 +172,31 @@ class MigrationListAll(APIView):
 
 class MigrationTimeslotDetails(APIView):
     def get(self, request):
-        slotDict = {}
-        days = checkDays(request.GET.get('days', ''))
+        #TODO: FIX THIS GARBAGE I HATE IT
+        # 5 days starting day before - Return Array of arrays
+        day = checkDayNumber(request.GET.get('day', ''))
+        days = checkDayNumber(request.GET.get('noOfDays', ''))
         # timeslots = ['00:00-03:00', '03:00-06:00', '06:00-09:00', '08:00-12:00', '12:00-18:00', '18:00-00:00']
         timeslots = settings.MIG_TIMESLOTS
-        for slot in timeslots:
-            slotDict[slot] = Migration.objects.filter(migration_status="Booked", booked_date=datetime.datetime.now().date() + datetime.timedelta(days), booked_time=slot).count()
-        return Response(slotDict)
+        if checkValue(request.GET.get('values', '')):
+            slotArray = []
+            if days:
+                daySlotArray = []
+                for daynum in range(days):
+                    for slot in timeslots:
+                        slotArray.append(Migration.objects.filter(migration_status="Booked", booked_date=datetime.datetime.now().date() + datetime.timedelta(daynum -1), booked_time=slot).count())
+                        print(slotArray)
+                    daySlotArray.append(slotArray)
+                    slotArray = []
+                return Response(daySlotArray)
+            for slot in timeslots:
+                slotArray.append(Migration.objects.filter(migration_status="Booked", booked_date=datetime.datetime.now().date() + datetime.timedelta(days), booked_time=slot).count())
+            return Response(slotArray)
+        else:
+            slotDict = {}
+            for slot in timeslots:
+                slotDict[slot] = Migration.objects.filter(migration_status="Booked", booked_date=datetime.datetime.now().date() + datetime.timedelta(days), booked_time=slot).count()
+            return Response(slotDict)
 
 
 class getTimeslots(APIView):