1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
|
# Copyright (c) 2017 VMware, Inc.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at:
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
import json
from rest_framework import status
from rest_framework.response import Response
from rest_framework.views import APIView
from vio.pub.msapi import extsys
from vio.pub.vim.vimapi.glance import OperateImage
from vio.swagger import image_utils
from vio.pub.exceptions import VimDriverVioException
class GetDeleteImageView(APIView):
def get(self, request, vimid, tenantid, imageid):
try:
vim_info = extsys.get_vim_by_id(vimid)
except VimDriverVioException as e:
return Response(data={'error': str(e)}, status=e.status_code)
image_op = OperateImage.OperateImage(vim_info)
try:
image = image_op.get_vim_image(imageid)
vim_rsp = image_utils.vim_formatter(vim_info, tenantid)
rsp = image_utils.image_formatter(image)
rsp.update(vim_rsp)
return Response(data=rsp, status=status.HTTP_200_OK)
except Exception as e:
return Response(data={'error': str(e)},
status=status.HTTP_500_INTERNAL_SERVER_ERROR)
def delete(self, request, vimid, tenantid, imageid):
try:
vim_info = extsys.get_vim_by_id(vimid)
except VimDriverVioException as e:
return Response(data={'error': str(e)}, status=e.status_code)
image_op = OperateImage.OperateImage(vim_info)
try:
image_op.delete_vim_image(imageid)
return Response(status=status.HTTP_204_NO_CONTENT)
except Exception as e:
return Response(data={'error': str(e)},
status=status.HTTP_500_INTERNAL_SERVER_ERROR)
class CreateListImagesView(APIView):
def get(self, request, vimid, tenantid):
try:
vim_info = extsys.get_vim_by_id(vimid)
except VimDriverVioException as e:
return Response(data={'error': str(e)}, status=e.status_code)
query_data = dict(request.query_params)
image_instance = OperateImage.OperateImage(vim_info)
try:
images = image_instance.get_vim_images(**query_data)
rsp = {}
rsp['images'] = []
vim_rsp = image_utils.vim_formatter(vim_info, tenantid)
for image in images:
rsp['images'].append(image_utils.image_formatter(image))
rsp.update(vim_rsp)
return Response(data=rsp, status=status.HTTP_200_OK)
except Exception as e:
return Response(data={'error': str(e)},
status=status.HTTP_500_INTERNAL_SERVER_ERROR)
def post(self, request, vimid, tenantid):
try:
vim_info = extsys.get_vim_by_id(vimid)
except VimDriverVioException as e:
return Response(data={'error': str(e)}, status=e.status_code)
try:
req_body = json.loads(request.body)
except Exception as e:
return Response(data={'error': 'Fail to decode request body.'},
status=status.HTTP_500_INTERNAL_SERVER_ERROR)
vim_rsp = image_utils.vim_formatter(vim_info, tenantid)
image_instance = OperateImage.OperateImage(vim_info)
try:
images = image_instance.get_vim_images()
for image in images:
if image.name == req_body.get('name'):
image_info = image_instance.get_vim_image(image.id)
rsp = image_utils.image_formatter(image_info)
rsp['returnCode'] = '0'
rsp.update(vim_rsp)
return Response(data=rsp, status=status.HTTP_200_OK)
param = image_utils.req_body_formatter(req_body)
image = image_instance.create_vim_image(vimid, tenantid,
imagePath=req_body.get('imagePath'),
**param)
rsp = image_utils.image_formatter(image)
rsp.update(vim_rsp)
rsp['returnCode'] = '1'
return Response(data=rsp, status=status.HTTP_202_ACCEPTED)
except Exception as e:
return Response(data={'error': str(e)},
status=status.HTTP_500_INTERNAL_SERVER_ERROR)
|
