Abstracted getallmigrations

This commit is contained in:
Benjamyn 2020-11-13 13:01:08 +11:00
parent 08cbbb7ba7
commit fc179ed2c8

View File

@ -20,13 +20,20 @@ def checkLimit(limit):
else: else:
limit = int(limit) limit = int(limit)
return limit return limit
def getAllMigrations(limit):
return Migration.objects.all()[:limit]
def getFilteredMigrations(limit, **kwargs):
pass
class MigrationList(APIView): class MigrationList(APIView):
''' '''
Returns a list of migrations with simpleData Returns a list of migrations with simpleData
''' '''
def get(self, request, format=None): def get(self, request, format=None):
limit = checkLimit(request.GET.get('limit', '')) limit = checkLimit(request.GET.get('limit', ''))
migrations = Migration.objects.all()[:limit] migrations = getAllMigrations(limit)
serializer = MigrationOverView(migrations, many=True) serializer = MigrationOverView(migrations, many=True)
return Response(serializer.data) return Response(serializer.data)
@ -132,6 +139,6 @@ class PendingTerm(APIView):
class MigrationListAll(APIView): class MigrationListAll(APIView):
def get(self, request, format=None): def get(self, request, format=None):
limit = checkLimit(request.GET.get('limit', '')) limit = checkLimit(request.GET.get('limit', ''))
migrations = Migration.objects.all()[:limit] migrations = getAllMigrations(limit)
serializer = MigrationSerializer(migrations, many=True) serializer = MigrationSerializer(migrations, many=True)
return Response(serializer.data) return Response(serializer.data)