diff --git a/migratorapi/api/views.py b/migratorapi/api/views.py index 5d98162..be3192d 100644 --- a/migratorapi/api/views.py +++ b/migratorapi/api/views.py @@ -94,10 +94,9 @@ class MigrationsBooked(APIView): Returns list of booked migrations ''' def get(self, request, format=None): - days = checkDays(request.GET.get('days', '')) limit = checkLimit(request.GET.get('limit', '')) 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) class MigrationsWaitingTerm(APIView): @@ -126,9 +125,10 @@ class PendingMigrations(APIView): are booked for today (maybe before timezone.now)) ''' def get(self, request, format=None): + days = checkDays(request.GET.get('days', '')) limit = checkLimit(request.GET.get('limit', '')) 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] serializer = MigrationSerializer(migrations, many=True) return Response(serializer.data, status=status.HTTP_200_OK)