uncloud-mravi/uncloud_net/views.py

64 lines
2 KiB
Python
Raw Normal View History

2020-12-09 19:22:33 +00:00
from django.views.generic.edit import CreateView
from django.contrib.auth.mixins import LoginRequiredMixin
from django.contrib.messages.views import SuccessMessageMixin
2020-12-13 16:59:35 +00:00
from rest_framework.response import Response
from django.shortcuts import render
from rest_framework import viewsets, permissions
from .models import *
from .serializers import *
from .selectors import *
2020-12-13 16:59:35 +00:00
from .services import *
from .forms import *
from .tasks import *
2020-12-13 12:28:43 +00:00
class WireGuardVPNViewSet(viewsets.ModelViewSet):
serializer_class = WireGuardVPNSerializer
permission_classes = [permissions.IsAuthenticated]
2020-04-06 20:30:01 +00:00
2020-12-13 12:28:43 +00:00
def get_queryset(self):
if self.request.user.is_superuser:
obj = WireGuardVPN.objects.all()
else:
obj = WireGuardVPN.objects.filter(owner=self.request.user)
2020-12-13 12:28:43 +00:00
return obj
2020-12-13 16:59:35 +00:00
def create(self, request):
serializer = self.get_serializer(data=request.data)
serializer.is_valid(raise_exception=True)
vpn = create_wireguard_vpn(
owner=self.request.user,
public_key=serializer.validated_data['wireguard_public_key'],
network_mask=serializer.validated_data['network_mask']
)
2020-12-26 13:45:28 +00:00
2020-12-13 16:59:35 +00:00
return Response(WireGuardVPNSerializer(vpn).data)
2020-12-09 19:22:33 +00:00
class WireGuardVPNCreateView(LoginRequiredMixin, SuccessMessageMixin, CreateView):
model = WireGuardVPN
2020-12-09 19:22:33 +00:00
login_url = '/login/'
success_url = '/'
success_message = "%(network) was created successfully"
2020-12-09 19:22:33 +00:00
form_class = WireGuardVPNForm
2020-12-09 19:22:33 +00:00
def get_success_message(self, cleaned_data):
return self.success_message % dict(cleaned_data,
the_prefix = self.object.prefix)
2020-12-20 20:45:47 +00:00
class WireGuardVPNSizes(viewsets.ViewSet):
def list(self, request):
sizes = allowed_vpn_network_reservation_size()
print(sizes)
sizes = [ { 'size': size } for size in sizes ]
print(sizes)
return Response(WireGuardVPNSizesSerializer(sizes, many=True).data)