phase in vmhost

Signed-off-by: Nico Schottelius <nico@nico-notebook.schottelius.org>

uncloud/uncloud_vm/views.py

@@ -1,24 +1,18 @@
 from django.shortcuts import render
 
-
 from django.contrib.auth.models import User
 from django.shortcuts import get_object_or_404
-from myapps.serializers import UserSerializer
-from rest_framework import viewsets
+
+from rest_framework import viewsets, permissions
 from rest_framework.response import Response
 
+
 from opennebula.models import VM as OpenNebulaVM
 
-class VMViewSet(viewsets.ViewSet):
-    def list(self, request):
-        queryset = User.objects.all()
-        serializer = UserSerializer(queryset, many=True)
-        return Response(serializer.data)
+from .models import VMHost
+from .serializers import VMHostSerializer
 
-    def retrieve(self, request, pk=None):
-        queryset = User.objects.all()
-        user = get_object_or_404(queryset, pk=pk)
-        serializer = UserSerializer(user)
-        return Response(serializer.data)
-
-    permission_classes = [permissions.IsAuthenticated]
+class VMHostViewSet(viewsets.ModelViewSet):
+    serializer_class = VMHostSerializer
+    queryset = VMHost.objects.all()
+    permission_classes = [permissions.IsAdminUser]