from rest_framework import viewsets, generics, permissions
from .models import VM
from .serializers import VMSerializer, OpenNebulaVMSerializer


#class VMList(generics.ListAPIView):
#    queryset = VM.objects.all()
#    serializer_class = VMSerializer


class RawVMViewSet(viewsets.ModelViewSet):
#    lookup_field = 'vmid'
    queryset = VM.objects.all()
    serializer_class = VMSerializer
    permission_classes = [permissions.IsAuthenticated]


class VMViewSet(viewsets.ModelViewSet):
    queryset = VM.objects.all()
    serializer_class = OpenNebulaVMSerializer

    permission_classes = [permissions.IsAuthenticated]