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

fixed bad view

Browse Source
This commit is contained in:
pepper 2021-01-19 22:24:05 -05:00
parent c498286dbe
commit 7e3ed01e38
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
migratorapi/api/views.py
View File

@ -94,10 +94,9 @@ class MigrationsBooked(APIView):
Returns list of booked migrations Returns list of booked migrations
''' '''
def get(self, request, format=None): def get(self, request, format=None):
days = checkDays(request.GET.get('days', ''))
limit = checkLimit(request.GET.get('limit', '')) limit = checkLimit(request.GET.get('limit', ''))
migrations = MigrationSerializer(Migration.objects.filter( migrations = MigrationSerializer(Migration.objects.filter(
migration_status="Booked", booked_date=datetime.datetime.now().date() + datetime.timedelta(days))[:limit], many=True) migration_status="Booked", booked_date=datetime.datetime.now().date())[:limit], many=True)
return Response(migrations.data, status=status.HTTP_200_OK) return Response(migrations.data, status=status.HTTP_200_OK)
class MigrationsWaitingTerm(APIView): class MigrationsWaitingTerm(APIView):
@ -126,9 +125,10 @@ class PendingMigrations(APIView):
are booked for today (maybe before timezone.now)) are booked for today (maybe before timezone.now))
''' '''
def get(self, request, format=None): def get(self, request, format=None):
days = checkDays(request.GET.get('days', ''))
limit = checkLimit(request.GET.get('limit', '')) limit = checkLimit(request.GET.get('limit', ''))
migrations = Migration.objects.filter( migrations = Migration.objects.filter(
migration_status="Booked", booked_date=datetime.datetime.now().date() migration_status="Booked", booked_date=datetime.datetime.now().date() + datetime.timedelta(days)
)[:limit] )[:limit]
serializer = MigrationSerializer(migrations, many=True) serializer = MigrationSerializer(migrations, many=True)
return Response(serializer.data, status=status.HTTP_200_OK) return Response(serializer.data, status=status.HTTP_200_OK)