162 lines
6.3 KiB
Python
162 lines
6.3 KiB
Python
import datetime
|
|
|
|
from django.shortcuts import get_object_or_404, render
|
|
from django.http import HttpResponseRedirect, JsonResponse, Http404
|
|
from django.views import generic
|
|
from django.urls import reverse
|
|
from django.utils import timezone
|
|
from rest_framework import status, viewsets
|
|
from rest_framework.views import APIView
|
|
from rest_framework.response import Response
|
|
from rest_framework.decorators import action
|
|
from .serializers import MigrationSerializer, MigrationOverView
|
|
|
|
|
|
from .models import Migration
|
|
|
|
class MigrationList(APIView):
|
|
'''
|
|
Returns a list of migrations with simpleData
|
|
'''
|
|
def get(self, request, format=None):
|
|
migrations = Migration.objects.all()
|
|
serializer = MigrationOverView(migrations, many=True)
|
|
return Response(serializer.data)
|
|
|
|
def post(self, request, format=None):
|
|
serializer = MigrationSerializer(data=request.data)
|
|
if serializer.is_valid():
|
|
serializer.save()
|
|
return Response(serializer.data, status=status.HTTP_201_CREATED)
|
|
return Response(serializer.errors, status=status.HTTP_400_BAD_REQUEST)
|
|
|
|
class MigrationDetails(APIView):
|
|
'''
|
|
Returns full migration details
|
|
'''
|
|
def getObject(self, pk):
|
|
try:
|
|
migration = Migration.objects.get(id=pk)
|
|
return migration
|
|
except Migration.DoesNotExist:
|
|
raise Http404
|
|
|
|
def get(self, request, pk, format=None):
|
|
migration = self.getObject(pk)
|
|
serializer = MigrationSerializer(migration)
|
|
return Response(serializer.data)
|
|
|
|
def put(self, request, pk, format=None):
|
|
migration = self.getObject(pk)
|
|
serializer = MigrationSerializer(migration, data=request.data)
|
|
if serializer.is_valid():
|
|
serializer.save()
|
|
return Response(serializer.data)
|
|
return Response(serializer.errors, status=status.HTTP_400_BAD_REQUEST)
|
|
|
|
class MigrationsBooked(APIView):
|
|
'''
|
|
Returns list of booked migrations
|
|
'''
|
|
def get(self, request, format=None):
|
|
migrations = MigrationOverView(Migration.objects.filter(
|
|
migration_status="Booked",), many=True)
|
|
return Response(migrations.data, status=status.HTTP_200_OK)
|
|
|
|
class MigrationsWaitingTerm(APIView):
|
|
'''
|
|
Returns list of booked migrations
|
|
'''
|
|
def get(self, request, format=None):
|
|
migrations = MigrationOverView(Migration.objects.filter(
|
|
migration_status="Waiting Termination",), many=True)
|
|
return Response(migrations.data, status=status.HTTP_200_OK)
|
|
|
|
class MigrationsCompleted(APIView):
|
|
'''
|
|
Returns list of booked migrations
|
|
'''
|
|
def get(self, request, format=None):
|
|
migrations = MigrationOverView(Migration.objects.filter(
|
|
migration_status="Completed",), many=True)
|
|
return Response(migrations.data, status=status.HTTP_200_OK)
|
|
|
|
class PendingMigrations(APIView):
|
|
'''
|
|
Returns list of 'pending' migrations (migrations that are booked for today (maybe before timezone.now))
|
|
'''
|
|
def get(self, request, format=None):
|
|
migrations = Migration.objects.filter(
|
|
migration_status="Booked", booked_time__lte=timezone.now() + datetime.timedelta(1),
|
|
booked_time__gte=(timezone.now() + datetime.timedelta(-1))
|
|
)
|
|
serializer = MigrationOverView(migrations, many=True)
|
|
return Response(serializer.data, status=status.HTTP_200_OK)
|
|
|
|
class MissedMigrations(APIView):
|
|
'''
|
|
Returns a list of missed migrations
|
|
'''
|
|
def get(self, request, format=None):
|
|
migrations = Migration.objects.filter(
|
|
booked_time__lte=timezone.now() + datetime.timedelta(-1),
|
|
migration_status="Booked"
|
|
)
|
|
serializer = MigrationOverView(migrations, many=True)
|
|
return Response(serializer.data)
|
|
|
|
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):
|
|
migrations = Migration.objects.filter(
|
|
migration_status="Waiting Termination", term_date__lte=timezone.now() + datetime.timedelta(1)
|
|
)
|
|
serializer = MigrationOverView(migrations, many=True)
|
|
return Response(serializer.data)
|
|
|
|
# class MigrationViewSet(viewsets.ModelViewSet):
|
|
# ''' Class for defining the migration views '''
|
|
# queryset = Migration.objects.all()
|
|
# serializer_class = MigrationOverView
|
|
# # permission_classes = (AllowAny,)
|
|
|
|
# @action(detail=False, methods=['GET'])
|
|
# def upcoming(self, request):
|
|
# ''' Returns a list of the migrations due today '''
|
|
# queryset = MigrationSerializer(
|
|
# 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) '''
|
|
# queryset = MigrationSerializer(
|
|
# 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):
|
|
# ''' 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 completed(self, request):
|
|
# ''' Returns a list of the completed migrations '''
|
|
# queryset = MigrationSerializer(Migration.objects.filter(
|
|
# migration_status="Completed",), 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="Waiting Termination", term_date__lte=timezone.now()), many=True)
|
|
# return Response(queryset.data, status=status.HTTP_200_OK)
|