388 - Paginated wallets (#389)

Author: buckyroberts

thenewboston/wallets/filters/__init__.py

@@ -0,0 +1,11 @@
+import django_filters
+
+from ..models import Wallet
+
+
+class WalletFilter(django_filters.FilterSet):
+    currency = django_filters.NumberFilter(field_name='currency')
+
+    class Meta:
+        model = Wallet
+        fields = ('currency',)

thenewboston/wallets/filters/wallet.py

@@ -1,13 +1,16 @@
 from django.conf import settings
+from django_filters.rest_framework import DjangoFilterBackend
 from rest_framework import mixins, status, viewsets
 from rest_framework.decorators import action
 from rest_framework.permissions import IsAuthenticated
 from rest_framework.response import Response
 
 from thenewboston.api.accounts import fetch_balance, wire_funds
 from thenewboston.general.constants import TRANSACTION_FEE
+from thenewboston.general.pagination import CustomPageNumberPagination
 from thenewboston.general.permissions import IsObjectOwnerOrReadOnly
 
+from ..filters.wallet import WalletFilter
 from ..models import Wallet, Wire
 from ..models.wire import WireType
 from ..serializers.block import BlockSerializer
@@ -20,6 +23,9 @@
 class WalletViewSet(
     mixins.CreateModelMixin, mixins.RetrieveModelMixin, mixins.ListModelMixin, viewsets.GenericViewSet
 ):
+    filter_backends = [DjangoFilterBackend]
+    filterset_class = WalletFilter
+    pagination_class = CustomPageNumberPagination
     permission_classes = [IsAuthenticated, IsObjectOwnerOrReadOnly]
     queryset = Wallet.objects.none()