++vpn network

uncloud_net/views.py

@@ -23,17 +23,17 @@ from .forms       import *
 #     queryset = VPNNetworkReservation.objects.all()
 
 
-# class VPNNetworkViewSet(viewsets.ModelViewSet):
-#     serializer_class = VPNNetworkSerializer
-#     permission_classes = [permissions.IsAuthenticated]
+class WireGuardVPNViewSet(viewsets.ModelViewSet):
+    serializer_class = WireGuardVPNSerializer
+    permission_classes = [permissions.IsAuthenticated]
 
-#     def get_queryset(self):
-#         if self.request.user.is_superuser:
-#             obj = VPNNetwork.objects.all()
-#         else:
-#             obj = VPNNetwork.objects.filter(owner=self.request.user)
+    def get_queryset(self):
+        if self.request.user.is_superuser:
+            obj = WireGuardVPN.objects.all()
+        else:
+            obj = WireGuardVPN.objects.filter(owner=self.request.user)
 
-#         return obj
+        return obj
 
 
 class WireGuardVPNCreateView(LoginRequiredMixin, SuccessMessageMixin, CreateView):
@@ -49,7 +49,14 @@ class WireGuardVPNCreateView(LoginRequiredMixin, SuccessMessageMixin, CreateView
         return self.success_message % dict(cleaned_data,
                                            the_prefix = self.object.prefix)
 
-    def get_context_data(self, **kwargs):
-        context = super().get_context_data(**kwargs)
-        context['available_sizes'] = 2
-        return context
+    # def get_context_data(self, **kwargs):
+    #     context = super().get_context_data(**kwargs)
+    #     context['available_sizes'] = 2
+    #     return context
+
+    # def post(request, *args, **kwargs):
+    #     print(request)
+    #     print(*args)
+    #     print(*kwargs)
+
+    # def post(self, request, *args, **kwargs):