user roles for csv

This commit is contained in:
alex 2020-01-30 14:42:27 +03:00
parent d6b8c6191d
commit ce919b54d0
2 changed files with 6 additions and 2 deletions

View File

@ -168,4 +168,5 @@ class UserCSVSerializer(serializers.ModelSerializer):
'created_at',
'last_seen_at',
'last_ip',
'roles',
]

View File

@ -143,12 +143,12 @@ def get_user_csv(request, id):
class UserCSVViewSet(ModelViewSet):
queryset = User.objects.all()
http_method_names = ['get', ]
serializer_class = serializers.UserCSVSerializer
permission_classes = (permissions.IsAdminUser,)
filter_class = filters.AccountBackOfficeFilter
filter_backends = (OrderingFilter, DjangoFilterBackend)
pagination_class = None
ordering_fields = (
'email_confirmed',
@ -159,6 +159,9 @@ class UserCSVViewSet(ModelViewSet):
'date_joined',
)
def get_queryset(self):
return User.objects.all().with_base_related().with_extend_related()
def to_csv(self, request):
queryset = self.filter_queryset(self.get_queryset())
serializer = serializers.UserCSVSerializer(queryset, many=True)
@ -177,7 +180,7 @@ class UserCSVViewSet(ModelViewSet):
# 'confirmed_at',
'last_seen_at',
'last_ip',
# 'role',
'role',
))
for item in serializer.data: