mirror of
https://github.com/retspen/webvirtcloud
synced 2025-07-31 12:41:08 +00:00
reorg imports and black format
This commit is contained in:
catborise
parent
fcd4b79431
commit
a1eab70e2d
27 changed files with 319 additions and 305 deletions
|
|
@ -1,36 +1,40 @@
|
|||
from django.shortcuts import get_object_or_404
|
||||
from appsettings.settings import app_settings
|
||||
from computes.models import Compute
|
||||
from django.shortcuts import get_object_or_404
|
||||
from rest_framework import status, viewsets
|
||||
from rest_framework.decorators import action
|
||||
from appsettings.settings import app_settings
|
||||
|
||||
from vrtManager.storage import wvmStorages, wvmStorage
|
||||
|
||||
from .serializers import StoragesSerializer, StorageSerializer, VolumeSerializer
|
||||
from rest_framework.response import Response
|
||||
from vrtManager.storage import wvmStorage, wvmStorages
|
||||
|
||||
from .serializers import StorageSerializer, StoragesSerializer, VolumeSerializer
|
||||
|
||||
|
||||
class StorageViewSet(viewsets.ViewSet):
|
||||
"""
|
||||
A viewset for listing retrieving storages.
|
||||
"""
|
||||
|
||||
|
||||
def list(self, request, compute_pk=None):
|
||||
|
||||
|
||||
compute = get_object_or_404(Compute, pk=compute_pk)
|
||||
|
||||
conn = wvmStorages(compute.hostname, compute.login, compute.password, compute.type)
|
||||
|
||||
conn = wvmStorages(
|
||||
compute.hostname, compute.login, compute.password, compute.type
|
||||
)
|
||||
queryset = conn.get_storages_info()
|
||||
|
||||
serializer = StoragesSerializer(queryset, many=True, context={'request': request})
|
||||
serializer = StoragesSerializer(
|
||||
queryset, many=True, context={"request": request}
|
||||
)
|
||||
|
||||
return Response(serializer.data)
|
||||
|
||||
|
||||
def retrieve(self, request, pk=None, compute_pk=None):
|
||||
compute = get_object_or_404(Compute, pk=compute_pk)
|
||||
|
||||
conn = wvmStorage(compute.hostname, compute.login, compute.password, compute.type, pk)
|
||||
conn = wvmStorage(
|
||||
compute.hostname, compute.login, compute.password, compute.type, pk
|
||||
)
|
||||
|
||||
infoset = {
|
||||
"state": conn.is_active(),
|
||||
|
|
@ -40,48 +44,57 @@ class StorageViewSet(viewsets.ViewSet):
|
|||
"path": conn.get_target_path(),
|
||||
"type": conn.get_type(),
|
||||
"autostart": conn.get_autostart(),
|
||||
"volumes": conn.update_volumes()
|
||||
"volumes": conn.update_volumes(),
|
||||
}
|
||||
|
||||
serializer = StorageSerializer(infoset, many=False, context={'request': request})
|
||||
serializer = StorageSerializer(
|
||||
infoset, many=False, context={"request": request}
|
||||
)
|
||||
return Response(serializer.data)
|
||||
|
||||
|
||||
@action(detail=True, methods=['post'])
|
||||
@action(detail=True, methods=["post"])
|
||||
def start(self, request, pk=None, compute_pk=None):
|
||||
compute = get_object_or_404(Compute, pk=compute_pk)
|
||||
|
||||
conn = wvmStorage(compute.hostname, compute.login, compute.password, compute.type, pk)
|
||||
conn = wvmStorage(
|
||||
compute.hostname, compute.login, compute.password, compute.type, pk
|
||||
)
|
||||
ret = conn.start()
|
||||
conn.close()
|
||||
return Response({'status': 'Pool start command send: ' + str(ret)})
|
||||
return Response({"status": "Pool start command send: " + str(ret)})
|
||||
|
||||
@action(detail=True, methods=['post'])
|
||||
@action(detail=True, methods=["post"])
|
||||
def stop(self, request, pk=None, compute_pk=None):
|
||||
compute = get_object_or_404(Compute, pk=compute_pk)
|
||||
|
||||
conn = wvmStorage(compute.hostname, compute.login, compute.password, compute.type, pk)
|
||||
conn = wvmStorage(
|
||||
compute.hostname, compute.login, compute.password, compute.type, pk
|
||||
)
|
||||
ret = conn.stop()
|
||||
conn.close()
|
||||
return Response({'status': 'Pool stop command send: ' + str(ret)})
|
||||
return Response({"status": "Pool stop command send: " + str(ret)})
|
||||
|
||||
@action(detail=True, methods=['post'])
|
||||
@action(detail=True, methods=["post"])
|
||||
def refresh(self, request, pk=None, compute_pk=None):
|
||||
compute = get_object_or_404(Compute, pk=compute_pk)
|
||||
|
||||
conn = wvmStorage(compute.hostname, compute.login, compute.password, compute.type, pk)
|
||||
conn = wvmStorage(
|
||||
compute.hostname, compute.login, compute.password, compute.type, pk
|
||||
)
|
||||
ret = conn.refresh()
|
||||
conn.close()
|
||||
return Response({'status': 'Pool refresh command send: ' + str(ret)})
|
||||
return Response({"status": "Pool refresh command send: " + str(ret)})
|
||||
|
||||
@action(detail=True, methods=['post'])
|
||||
@action(detail=True, methods=["post"])
|
||||
def XML_description(self, request, pk=None, compute_pk=None):
|
||||
compute = get_object_or_404(Compute, pk=compute_pk)
|
||||
|
||||
conn = wvmStorage(compute.hostname, compute.login, compute.password, compute.type, pk)
|
||||
conn = wvmStorage(
|
||||
compute.hostname, compute.login, compute.password, compute.type, pk
|
||||
)
|
||||
ret = conn._XMLDesc(0)
|
||||
conn.close()
|
||||
return Response({'return': str(ret)})
|
||||
return Response({"return": str(ret)})
|
||||
|
||||
|
||||
class VolumeViewSet(viewsets.ViewSet):
|
||||
|
|
@ -89,14 +102,17 @@ class VolumeViewSet(viewsets.ViewSet):
|
|||
"""
|
||||
A simple ViewSet for listing or retrieving Storage Volumes.
|
||||
"""
|
||||
|
||||
serializer_class = VolumeSerializer
|
||||
lookup_value_regex = "[^/]+"
|
||||
|
||||
def list(self, request, storage_pk=None, compute_pk=None):
|
||||
|
||||
|
||||
compute = get_object_or_404(Compute, pk=compute_pk)
|
||||
|
||||
conn = wvmStorage(compute.hostname, compute.login, compute.password, compute.type, storage_pk)
|
||||
conn = wvmStorage(
|
||||
compute.hostname, compute.login, compute.password, compute.type, storage_pk
|
||||
)
|
||||
state = conn.is_active()
|
||||
|
||||
if state:
|
||||
|
|
@ -105,29 +121,37 @@ class VolumeViewSet(viewsets.ViewSet):
|
|||
else:
|
||||
volume_queryset = None
|
||||
conn.close()
|
||||
serializer = VolumeSerializer(volume_queryset, many=True, context={'request': request})
|
||||
serializer = VolumeSerializer(
|
||||
volume_queryset, many=True, context={"request": request}
|
||||
)
|
||||
|
||||
return Response(serializer.data)
|
||||
|
||||
def retrieve(self, request, storage_pk=None, compute_pk=None, pk=None):
|
||||
compute = get_object_or_404(Compute, pk=compute_pk)
|
||||
|
||||
conn = wvmStorage(compute.hostname, compute.login, compute.password, compute.type, storage_pk)
|
||||
conn = wvmStorage(
|
||||
compute.hostname, compute.login, compute.password, compute.type, storage_pk
|
||||
)
|
||||
state = conn.is_active()
|
||||
|
||||
|
||||
if state:
|
||||
volume_queryset = conn.get_volume_details(pk)
|
||||
else:
|
||||
volume_queryset = None
|
||||
conn.close()
|
||||
serializer = VolumeSerializer(volume_queryset, many=False, context={'request': request})
|
||||
serializer = VolumeSerializer(
|
||||
volume_queryset, many=False, context={"request": request}
|
||||
)
|
||||
|
||||
return Response(serializer.data)
|
||||
|
||||
def create(self, request, storage_pk=None, compute_pk=None):
|
||||
compute = get_object_or_404(Compute, pk=compute_pk)
|
||||
|
||||
conn = wvmStorage(compute.hostname, compute.login, compute.password, compute.type, storage_pk)
|
||||
conn = wvmStorage(
|
||||
compute.hostname, compute.login, compute.password, compute.type, storage_pk
|
||||
)
|
||||
|
||||
serializer = VolumeSerializer(data=request.data)
|
||||
if serializer.is_valid():
|
||||
|
|
@ -135,28 +159,30 @@ class VolumeViewSet(viewsets.ViewSet):
|
|||
if state:
|
||||
conn.refresh()
|
||||
ret = conn.create_volume(
|
||||
serializer.validated_data['name'],
|
||||
serializer.validated_data['size'],
|
||||
serializer.validated_data['type'],
|
||||
serializer.validated_data['meta_prealloc'],
|
||||
serializer.validated_data["name"],
|
||||
serializer.validated_data["size"],
|
||||
serializer.validated_data["type"],
|
||||
serializer.validated_data["meta_prealloc"],
|
||||
int(app_settings.INSTANCE_VOLUME_DEFAULT_OWNER_UID),
|
||||
int(app_settings.INSTANCE_VOLUME_DEFAULT_OWNER_GID),
|
||||
)
|
||||
conn.close()
|
||||
return Response({'status': 'Volume: ' + ret + ' is created'})
|
||||
return Response({"status": "Volume: " + ret + " is created"})
|
||||
else:
|
||||
return Response({'status': 'Pool is not active'})
|
||||
return Response({"status": "Pool is not active"})
|
||||
else:
|
||||
return Response({'status': 'Data is not right for create volume'})
|
||||
return Response({"status": "Data is not right for create volume"})
|
||||
|
||||
def destroy(self, request, storage_pk=None, compute_pk=None, pk=None):
|
||||
compute = get_object_or_404(Compute, pk=compute_pk)
|
||||
|
||||
conn = wvmStorage(compute.hostname, compute.login, compute.password, compute.type, storage_pk)
|
||||
conn = wvmStorage(
|
||||
compute.hostname, compute.login, compute.password, compute.type, storage_pk
|
||||
)
|
||||
|
||||
if conn.is_active():
|
||||
conn.del_volume(pk)
|
||||
conn.close()
|
||||
return Response({'status': 'Volume: ' + pk + ' is deleted'})
|
||||
return Response({"status": "Volume: " + pk + " is deleted"})
|
||||
else:
|
||||
return Response({'status': 'Pool is not active'})
|
||||
return Response({"status": "Pool is not active"})
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue