[vpn] fix urls

This commit is contained in:
Nico Schottelius 2020-04-06 22:30:01 +02:00
commit 913e992a48
4 changed files with 34 additions and 19 deletions

View file

@ -6,14 +6,20 @@ from .models import *
from .serializers import *
class VPNProductViewSet(viewsets.ModelViewSet):
serializer_class = VPNProductSerializer
class VPNPoolViewSet(viewsets.ModelViewSet):
serializer_class = VPNPoolSerializer
permission_classes = [permissions.IsAdminUser]
queryset = VPNPool.objects.all()
class VPNNetworkViewSet(viewsets.ModelViewSet):
serializer_class = VPNNetworkSerializer
permission_classes = [permissions.IsAdminUser]
def get_queryset(self):
if self.request.user.is_superuser:
obj = VPNProduct.objects.all()
obj = VPNNetwork.objects.all()
else:
obj = VPNProduct.objects.filter(owner=self.request.user)
obj = VPNNetwork.objects.filter(owner=self.request.user)
return obj