[refactor] cleaning up uncloud_net for Wireguardvpn

uncloud_net/views.py

@@ -9,45 +9,41 @@ from rest_framework import viewsets, permissions
 
 from .models      import *
 from .serializers import *
+from .selectors   import *
+from .forms       import *
+
+# class VPNPoolViewSet(viewsets.ModelViewSet):
+#     serializer_class = VPNPoolSerializer
+#     permission_classes = [permissions.IsAdminUser]
+#     queryset = VPNPool.objects.all()
+
+# class VPNNetworkReservationViewSet(viewsets.ModelViewSet):
+#     serializer_class = VPNNetworkReservationSerializer
+#     permission_classes = [permissions.IsAdminUser]
+#     queryset = VPNNetworkReservation.objects.all()
 
 
-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.IsAuthenticated]
 
-class VPNNetworkReservationViewSet(viewsets.ModelViewSet):
-    serializer_class = VPNNetworkReservationSerializer
-    permission_classes = [permissions.IsAdminUser]
-    queryset = VPNNetworkReservation.objects.all()
+#     def get_queryset(self):
+#         if self.request.user.is_superuser:
+#             obj = VPNNetwork.objects.all()
+#         else:
+#             obj = VPNNetwork.objects.filter(owner=self.request.user)
+
+#         return obj
 
 
-class VPNNetworkViewSet(viewsets.ModelViewSet):
-    serializer_class = VPNNetworkSerializer
-#    permission_classes = [permissions.IsAdminUser]
-    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)
-
-        return obj
-
-
-
-
-class VPNCreateView(LoginRequiredMixin, SuccessMessageMixin, CreateView):
-    model = VPNNetwork
+class WireGuardVPNCreateView(LoginRequiredMixin, SuccessMessageMixin, CreateView):
+    model = WireGuardVPN
 
     login_url = '/login/'
     success_url = '/'
-    success_message = "%(the_prefix)s/48 was created successfully"
+    success_message = "%(network) was created successfully"
 
-    gen_method = "undef"
-
-    fields = [ "wireguard_public_key" ]
+    form_class = WireGuardVPNForm
 
     def get_success_message(self, cleaned_data):
         return self.success_message % dict(cleaned_data,
@@ -55,5 +51,5 @@ class VPNCreateView(LoginRequiredMixin, SuccessMessageMixin, CreateView):
 
     def get_context_data(self, **kwargs):
         context = super().get_context_data(**kwargs)
-        context['method'] = self.gen_method
+        context['available_sizes'] = 2
         return context