Update perm api

This commit is contained in:
ibuler
2016-11-10 00:18:57 +08:00
parent 47090eb0f7
commit dde9ffb2ae
4 changed files with 25 additions and 29 deletions

View File

@@ -3,17 +3,26 @@
from rest_framework.views import APIView, Response
from rest_framework.generics import ListCreateAPIView
from rest_framework import viewsets
from users.backends import IsValidUser, IsSuperUser
from .utils import get_user_granted_assets, get_user_granted_asset_groups
from .models import AssetPermission
from . import serializers
class AssetPermissionListCreateApi(ListCreateAPIView):
class AssetPermissionViewSet(viewsets.ModelViewSet):
queryset = AssetPermission.objects.all()
serializer_class = serializers.AssetPermissionSerializer
permission_classes = (IsSuperUser,)
def get_queryset(self):
queryset = super(AssetPermissionViewSet, self).get_queryset()
user_id = self.request.query_params.get('user', '')
if user_id:
queryset = queryset.filter(users__id=user_id)
return queryset
class UserAssetsApi(APIView):
permission_classes = (IsValidUser,)