aboutsummaryrefslogtreecommitdiffstats
path: root/pydis_site/apps/api/viewsets
diff options
context:
space:
mode:
authorGravatar Kunal Sharma <[email protected]>2021-01-28 03:05:53 +0530
committerGravatar Kunal Sharma <[email protected]>2021-01-28 03:05:53 +0530
commit83b5f8ebd9c17fe6a444765f8d3a8ead431fa916 (patch)
tree40c69872ff5d762d2fce73f26b2f479a9b621143 /pydis_site/apps/api/viewsets
parentupdate paginated response in list infraction (diff)
Styling fixes
Diffstat (limited to 'pydis_site/apps/api/viewsets')
-rw-r--r--pydis_site/apps/api/viewsets/bot/infraction.py17
1 files changed, 12 insertions, 5 deletions
diff --git a/pydis_site/apps/api/viewsets/bot/infraction.py b/pydis_site/apps/api/viewsets/bot/infraction.py
index 851eae34..6464c1b9 100644
--- a/pydis_site/apps/api/viewsets/bot/infraction.py
+++ b/pydis_site/apps/api/viewsets/bot/infraction.py
@@ -3,15 +3,21 @@ from django_filters.rest_framework import DjangoFilterBackend
from rest_framework.decorators import action
from rest_framework.exceptions import ValidationError
from rest_framework.filters import OrderingFilter, SearchFilter
-from rest_framework.mixins import (CreateModelMixin, DestroyModelMixin,
- ListModelMixin, RetrieveModelMixin)
+from rest_framework.mixins import (
+ CreateModelMixin,
+ DestroyModelMixin,
+ ListModelMixin,
+ RetrieveModelMixin
+)
from rest_framework.pagination import LimitOffsetPagination
from rest_framework.response import Response
from rest_framework.viewsets import GenericViewSet
from pydis_site.apps.api.models.bot.infraction import Infraction
-from pydis_site.apps.api.serializers import (ExpandedInfractionSerializer,
- InfractionSerializer)
+from pydis_site.apps.api.serializers import (
+ ExpandedInfractionSerializer,
+ InfractionSerializer
+)
class InfractionViewSet(
@@ -145,7 +151,8 @@ class InfractionViewSet(
page = self.paginate_queryset(queryset)
if page:
serializer = self.get_serializer(page, many=True)
- return Response(self.get_paginated_response(serializer.data).data.get('results'))
+ result = self.get_paginated_response(serializer.data)
+ return Response(result.data.get('results'))
serializer = InfractionSerializer(queryset, many=True)
return Response(serializer.data)