forked from uncloud/uncloud
29 lines
995 B
Python
29 lines
995 B
Python
from django.shortcuts import render
|
|
|
|
from django.contrib.auth.models import User
|
|
from django.shortcuts import get_object_or_404
|
|
|
|
from rest_framework import viewsets, permissions
|
|
from rest_framework.response import Response
|
|
|
|
from .models import VMHost, VMProduct
|
|
from .serializers import VMHostSerializer, VMProductSerializer
|
|
|
|
class VMHostViewSet(viewsets.ModelViewSet):
|
|
serializer_class = VMHostSerializer
|
|
queryset = VMHost.objects.all()
|
|
permission_classes = [permissions.IsAdminUser]
|
|
|
|
class VMProductViewSet(viewsets.ModelViewSet):
|
|
permission_classes = [permissions.IsAuthenticated]
|
|
serializer_class = VMProductSerializer
|
|
|
|
def get_queryset(self):
|
|
return VMProduct.objects.filter(owner=self.request.user)
|
|
|
|
def create(self, request):
|
|
serializer = VMProductSerializer(data=request.data, context={'request': request})
|
|
serializer.is_valid(raise_exception=True)
|
|
serializer.save(owner=request.user)
|
|
|
|
return Response(serializer.data)
|