migratorapi/api/migrations/0010_migration_term_date.py

@@ -1,18 +0,0 @@
-# Generated by Django 3.1.2 on 2020-10-22 22:20
-
-from django.db import migrations, models
-
-
-class Migration(migrations.Migration):
-
-    dependencies = [
-        ('api', '0009_migration_is_urgent'),
-    ]
-
-    operations = [
-        migrations.AddField(
-            model_name='migration',
-            name='term_date',
-            field=models.DateField(blank=True, null=True, verbose_name='Date to be terminated'),
-        ),
-    ]

migratorapi/api/models.py

@@ -23,8 +23,6 @@ class Migration(models.Model):
     additional_domains = models.CharField(
         max_length=500, null=True, blank=True)
     migration_type = models.CharField(max_length=200)
-    term_date = models.DateField(
-        'Date to be terminated', null=True, blank=True)
     is_urgent = models.BooleanField(default=False)
 
     def print_missed_migrations(self):

migratorapi/api/views.py

@@ -21,32 +21,22 @@ class MigrationViewSet(viewsets.ModelViewSet):
     # permission_classes = (AllowAny,)
 
     @action(detail=False, methods=['GET'])
-    def upcoming(self, request):
+    def upcoming(self, request, *args, **kwargs):
         ''' Returns a list of the migrations due today '''
         queryset = MigrationSerializer(
-            Migration.objects.filter(booked_time=timezone.now(),
-                                     migration_status="Booked"), many=True)
+            Migration.objects.filter(booked_time=timezone.now(), migration_status="Booked"), many=True)
         return Response(queryset.data, status=status.HTTP_200_OK)
 
     @action(detail=False, methods=['GET'])
-    def missed(self, request):
-        ''' Returns a list of the missed migrations (Still have the status
-         booked and date is greater then today) '''
+    def missed(self, request, *args, **kwargs):
+        ''' Returns a list of the missed migrations (Still have the status booked and date is greater then today) '''
         queryset = MigrationSerializer(
-            Migration.objects.filter(booked_time__gte=timezone.now() + datetime.timedelta(1),
-                                     migration_status="Booked"), many=True)
+            Migration.objects.filter(booked_time__gte=timezone.now() + datetime.timedelta(1), migration_status="Booked"), many=True)
         return Response(queryset.data, status=status.HTTP_200_OK)
 
     @action(detail=False, methods=['GET'])
-    def booked(self, request):
+    def booked(self, request, *args, **kwargs):
         ''' Returns a list of the booked migrations '''
         queryset = MigrationSerializer(Migration.objects.filter(
             migration_status="booked",), many=True)
         return Response(queryset.data, status=status.HTTP_200_OK)
-
-    @action(detail=False, methods=['GET'])
-    def awaitterm(self, request):
-        ''' Returns migrations awaiting termination '''
-        queryset = MigrationSerializer(Migration.objects.filter(
-            migration_status="Awaiting Termination", term_date__lte=timezone.now()), many=True)
-        return Response(queryset.data, status=status.HTTP_200_OK)