LIL-308. Only authors & admins in author balance req

8 years ago · 75954748c2
parent 3fcde04387
commit 75954748c2
1 changed files with 3 additions and 1 deletions
--- a/api/v1/views.py
+++ b/api/v1/views.py
@ -55,7 +55,9 @@ User = get_user_model()


 class AuthorBalanceViewSet(ExtendedModelViewSet):
-    queryset = AuthorBalance.objects.all()
+    queryset = AuthorBalance.objects.filter(
+        author__role__in=[User.AUTHOR_ROLE, User.ADMIN_ROLE],
+    )
    serializer_class = AuthorBalanceCreateSerializer
    serializer_class_map = {
        'list': AuthorBalanceSerializer,