From 75954748c20bb705ebc742aee77b80d74811692a Mon Sep 17 00:00:00 2001
From: Ivlev Denis <di-erz@yandex.ru>
Date: Fri, 16 Mar 2018 14:19:28 +0300
Subject: [PATCH] LIL-308. Only authors & admins in author balance req

---
 api/v1/views.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/api/v1/views.py b/api/v1/views.py
index 4a854b78..8aaf84b8 100644
--- 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,