diff --git a/core/apps/counterparty/serializers/counterparty_folder.py b/core/apps/counterparty/serializers/counterparty_folder.py index 6e7ef70..c1d2d4d 100644 --- a/core/apps/counterparty/serializers/counterparty_folder.py +++ b/core/apps/counterparty/serializers/counterparty_folder.py @@ -6,12 +6,17 @@ from core.apps.counterparty.models import CounterpartyFolder class CounterpartyFolderListSerializer(serializers.ModelSerializer): + counterparty_count = serializers.SerializerMethodField(method_name='get_counterparty_count') + class Meta: model = CounterpartyFolder fields = [ - 'id', 'name' + 'id', 'name', 'counterparty_count' ] + def get_counterparty_count(self, obj): + return obj.counterparties.count() + class CounterpartyFolderCreateSerializer(serializers.Serializer): name = serializers.CharField() diff --git a/core/apps/counterparty/views/counterparty_folder.py b/core/apps/counterparty/views/counterparty_folder.py index 9341592..e74b6a1 100644 --- a/core/apps/counterparty/views/counterparty_folder.py +++ b/core/apps/counterparty/views/counterparty_folder.py @@ -11,6 +11,7 @@ class CounterpartyFolderListApiView(generics.GenericAPIView): queryset = CounterpartyFolder.objects.all() permission_classes = [HasRolePermission] required_permissions = [] + pagination_class = None def get(self, request): folders = self.get_queryset()