The_Bois/MigratorProject
Archived
3
2
Fork 0
Code Issues 1 Pull Requests Releases Wiki Activity

Added /all/ and modified limits

Browse Source
This commit is contained in:
Benjamyn 2020-11-04 12:18:20 +11:00
parent b6b4e03890
commit 08cbbb7ba7
1 changed files with 28 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

36
migratorapi/api/views.py
View File

@ -14,12 +14,19 @@ from .serializers import MigrationSerializer, MigrationOverView
from .models import Migration
def checkLimit(limit):
if limit == "":
limit = None
else:
limit = int(limit)
return limit
class MigrationList(APIView):
'''
Returns a list of migrations with simpleData
'''
def get(self, request, format=None):
migrations = Migration.objects.all()
limit = checkLimit(request.GET.get('limit', ''))
migrations = Migration.objects.all()[:limit]
serializer = MigrationOverView(migrations, many=True)
return Response(serializer.data)
@ -59,8 +66,9 @@ class MigrationsBooked(APIView):
Returns list of booked migrations
'''
def get(self, request, format=None):
limit = checkLimit(request.GET.get('limit', ''))
migrations = MigrationOverView(Migration.objects.filter(
migration_status="Booked",), many=True)
migration_status="Booked",)[:limit], many=True)
return Response(migrations.data, status=status.HTTP_200_OK)
class MigrationsWaitingTerm(APIView):
@ -68,8 +76,9 @@ class MigrationsWaitingTerm(APIView):
Returns list of booked migrations
'''
def get(self, request, format=None):
limit = checkLimit(request.GET.get('limit', ''))
migrations = MigrationOverView(Migration.objects.filter(
migration_status="Waiting Termination",), many=True)
migration_status="Waiting Termination",)[:limit], many=True)
return Response(migrations.data, status=status.HTTP_200_OK)
class MigrationsCompleted(APIView):
@ -77,8 +86,9 @@ class MigrationsCompleted(APIView):
Returns list of booked migrations
'''
def get(self, request, format=None):
limit = checkLimit(request.GET.get('limit', ''))
migrations = MigrationOverView(Migration.objects.filter(
migration_status="Completed",), many=True)
migration_status="Completed",)[:limit], many=True)
return Response(migrations.data, status=status.HTTP_200_OK)
class PendingMigrations(APIView):
@ -86,10 +96,11 @@ class PendingMigrations(APIView):
Returns list of 'pending' migrations (migrations that are booked for today (maybe before timezone.now))
'''
def get(self, request, format=None):
limit = checkLimit(request.GET.get('limit', ''))
migrations = Migration.objects.filter(
migration_status="Booked", booked_time__lte=timezone.now() + datetime.timedelta(1),
booked_time__gte=(timezone.now() + datetime.timedelta(-1))
)
)[:limit]
serializer = MigrationOverView(migrations, many=True)
return Response(serializer.data, status=status.HTTP_200_OK)
@ -98,10 +109,11 @@ class MissedMigrations(APIView):
Returns a list of missed migrations
'''
def get(self, request, format=None):
limit = checkLimit(request.GET.get('limit', ''))
migrations = Migration.objects.filter(
booked_time__lte=timezone.now() + datetime.timedelta(-1),
migration_status="Booked"
)
)[:limit]
serializer = MigrationOverView(migrations, many=True)
return Response(serializer.data)
@ -110,8 +122,16 @@ class PendingTerm(APIView):
Returns a list of accounts pending termination (might add an arg so we can specify how many days)
'''
def get(self, request, format=None):
limit = checkLimit(request.GET.get('limit', ''))
migrations = Migration.objects.filter(
migration_status="Waiting Termination", term_date__lte=timezone.now() + datetime.timedelta(1)
)
)[:limit]
serializer = MigrationOverView(migrations, many=True)
return Response(serializer.data)
return Response(serializer.data)
class MigrationListAll(APIView):
def get(self, request, format=None):
limit = checkLimit(request.GET.get('limit', ''))
migrations = Migration.objects.all()[:limit]
serializer = MigrationSerializer(migrations, many=True)
return Response(serializer.data)