From 9edc5f504219efe5604bd0794856a32c2d4743ef Mon Sep 17 00:00:00 2001 From: benjamyn Date: Fri, 23 Oct 2020 00:03:22 -0400 Subject: [PATCH] Added completed --- migratorapi/api/models.py | 1 + migratorapi/api/views.py | 7 +++++++ 2 files changed, 8 insertions(+) diff --git a/migratorapi/api/models.py b/migratorapi/api/models.py index 2d37e09..8d00a4e 100644 --- a/migratorapi/api/models.py +++ b/migratorapi/api/models.py @@ -25,6 +25,7 @@ class Migration(models.Model): migration_type = models.CharField(max_length=200) term_date = models.DateField( 'Date to be terminated', null=True, blank=True) + migration_cmd = models.CharField(max_length=200, null=True, blank=True) is_urgent = models.BooleanField(default=False) def print_missed_migrations(self): diff --git a/migratorapi/api/views.py b/migratorapi/api/views.py index 4a09866..8a6fba1 100644 --- a/migratorapi/api/views.py +++ b/migratorapi/api/views.py @@ -44,6 +44,13 @@ class MigrationViewSet(viewsets.ModelViewSet): migration_status="booked",), many=True) return Response(queryset.data, status=status.HTTP_200_OK) + @action(detail=False, methods=['GET']) + def completed(self, request): + ''' Returns a list of the booked migrations ''' + queryset = MigrationSerializer(Migration.objects.filter( + migration_status="compelted",), many=True) + return Response(queryset.data, status=status.HTTP_200_OK) + @action(detail=False, methods=['GET']) def awaitterm(self, request): ''' Returns migrations awaiting termination '''