# Copyright (c) 2017-2018 Wind River Systems, 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. # See the License for the specific language governing permissions and # limitations under the License. import logging import json import traceback from keystoneauth1.exceptions import HttpError from rest_framework import status from rest_framework.response import Response from rest_framework.views import APIView from common.exceptions import VimDriverNewtonException from newton_base.util import VimDriverUtils from common.msapi import extsys logger = logging.getLogger(__name__) class Flavors(APIView): service = {'service_type': 'compute', 'interface': 'public'} keys_mapping = [ ("project_id", "tenantId"), ("ram", "memory"), ("vcpus", "vcpu"), ("OS-FLV-EXT-DATA:ephemeral", "ephemeral"), ("os-flavor-access:is_public", "isPublic"), ("extra_specs", "extraSpecs"), ] def __init__(self): super(Flavors, self).__init__() self._logger = logger def _convert_extra_specs(self, extra_specs_vfc, extra_specs_openstack, reverse=False): if reverse == False: # from extraSpecs to extra_specs for spec in extra_specs_vfc: extra_specs_openstack[spec['keyName']] = spec['value'] else: for k,v in list(extra_specs_openstack.items()): spec={} spec['keyName']=k spec['value']=v extra_specs_vfc.append(spec) def get(self, request, vimid="", tenantid="", flavorid=""): logger.info("vimid, tenantid, flavorid = %s,%s,%s" % (vimid, tenantid, flavorid)) if request.data: logger.debug("With data = %s" % request.data) pass try: # prepare request resource to vim instance query = VimDriverUtils.get_query_part(request) vim = VimDriverUtils.get_vim_info(vimid) sess = VimDriverUtils.get_session(vim, tenantid) self.service['region_name'] = vim['openstack_region_id'] \ if vim.get('openstack_region_id') \ else vim['cloud_region_id'] resp = self._get_flavor(sess, request, flavorid) content = resp.json() if flavorid: flavor = content.pop("flavor", None) extraResp = self._get_flavor_extra_specs(sess, flavor["id"]) extraContent = extraResp.json() if extraContent["extra_specs"]: extra_specs_vfc = [] self._convert_extra_specs(extra_specs_vfc, extraContent["extra_specs"], True) flavor["extraSpecs"] = extra_specs_vfc VimDriverUtils.replace_key_by_mapping(flavor, self.keys_mapping) content.update(flavor) else: wanted = None # check if query contains name="flavorname" if query: for queryone in query.split('&'): k,v = queryone.split('=') if k == "name": wanted = v break if wanted: oldFlavors = content.pop("flavors", None) content["flavors"] = [] for flavor in oldFlavors: if wanted == flavor["name"]: content["flavors"].append(flavor) # iterate each flavor to get extra_specs for flavor in content["flavors"]: extraResp = self._get_flavor_extra_specs(sess, flavor["id"]) extraContent = extraResp.json() if extraContent["extra_specs"]: extra_specs_vfc = [] self._convert_extra_specs(extra_specs_vfc, extraContent["extra_specs"], True) flavor["extraSpecs"] = extra_specs_vfc VimDriverUtils.replace_key_by_mapping(flavor, self.keys_mapping) # add extra keys vim_dict = { "vimName": vim["name"], "vimId": vim["vimId"], "cloud-owner": vim["cloud_owner"], "cloud-region-id": vim["cloud_region_id"], "tenantId": tenantid, } content.update(vim_dict) logger.info("response with status = %s" % resp.status_code) return Response(data=content, status=resp.status_code) except VimDriverNewtonException as e: logger.error("response with status = %s" % e.status_code) return Response(data={'error': e.content}, status=e.status_code) except HttpError as e: logger.error("HttpError: status:%s, response:%s" % (e.http_status, e.response.json())) return Response(data=e.response.json(), status=e.http_status) except Exception as e: logger.error(traceback.format_exc()) return Response(data={'error': str(e)}, status=status.HTTP_500_INTERNAL_SERVER_ERROR) def _get_flavor_extra_specs(self, sess, flavorid): if flavorid: # prepare request resource to vim instance req_resouce = "/flavors/%s/os-extra_specs" % flavorid logger.info("making request with URI:%s" % req_resouce) resp = sess.get(req_resouce, endpoint_filter=self.service) logger.info("request returns with status %s" % resp.status_code) if resp.status_code == status.HTTP_200_OK: logger.debug("with content:%s" % resp.json()) pass return resp return {} def _get_flavor(self, sess, request, flavorid=""): if sess: # prepare request resource to vim instance req_resouce = "/flavors" if flavorid: req_resouce += "/%s" % flavorid else: req_resouce += "/detail" query = VimDriverUtils.get_query_part(request) if query: req_resouce += "?%s" % query logger.info("making request with URI:%s" % req_resouce) resp = sess.get(req_resouce, endpoint_filter=self.service) logger.info("request returns with status %s" % resp.status_code) if resp.status_code == status.HTTP_200_OK: logger.debug("with content:%s" % resp.json()) pass return resp return {} def post(self, request, vimid="", tenantid="", flavorid=""): logger.info("vimid, tenantid, flavorid = %s,%s,%s" % (vimid, tenantid, flavorid)) if request.data: logger.debug("With data = %s" % request.data) pass sess = None resp = None resp_body = None try: # prepare request resource to vim instance vim = VimDriverUtils.get_vim_info(vimid) sess = VimDriverUtils.get_session(vim, tenantid) self.service['region_name'] = vim['openstack_region_id'] \ if vim.get('openstack_region_id') \ else vim['cloud_region_id'] #check if the flavor is already created: name or id tmpresp = self._get_flavor(sess, request) content = tmpresp.json() #iterate each flavor to get extra_specs existed = False for flavor in content["flavors"]: if flavor["name"] == request.data["name"]: existed = True break elif hasattr(request.data, "id") and flavor["id"] == request.data["id"]: existed = True break if existed: extraResp = self._get_flavor_extra_specs(sess, flavor["id"]) extraContent = extraResp.json() if extraContent["extra_specs"]: extra_specs_vfc = [] self._convert_extra_specs(extra_specs_vfc, extraContent["extra_specs"], True) flavor["extraSpecs"] = extra_specs_vfc VimDriverUtils.replace_key_by_mapping(flavor, self.keys_mapping) vim_dict = { "vimName": vim["name"], "vimId": vim["vimId"], "cloud-owner": vim["cloud_owner"], "cloud-region-id": vim["cloud_region_id"], "tenantId": tenantid, "returnCode": 0, } flavor.update(vim_dict) return Response(data=flavor, status=tmpresp.status_code) extra_specs_vfc = request.data.pop("extraSpecs", None) #create flavor first resp = self._create_flavor(sess, request) if resp.status_code == 200: resp_body = resp.json()["flavor"] else: return resp flavorid = resp_body['id'] if extra_specs_vfc: extra_specs_openstack={} self._convert_extra_specs(extra_specs_vfc, extra_specs_openstack, False) extraResp = self._create_flavor_extra_specs(sess, extra_specs_openstack, flavorid) if extraResp.status_code == 200: #combine the response body and return tmp_specs_vfc = [] tmpRespBody = extraResp.json() self._convert_extra_specs(tmp_specs_vfc, tmpRespBody['extra_specs'], True) resp_body.update({"extraSpecs":tmp_specs_vfc}) else: #rollback self._delete_flavor(self, request, vimid, tenantid, flavorid) return extraResp VimDriverUtils.replace_key_by_mapping(resp_body, self.keys_mapping) vim_dict = { "vimName": vim["name"], "vimId": vim["vimId"], "cloud-owner": vim["cloud_owner"], "cloud-region-id": vim["cloud_region_id"], "tenantId": tenantid, "returnCode": 1, } resp_body.update(vim_dict) return Response(data=resp_body, status=resp.status_code) except VimDriverNewtonException as e: logger.error("response with status = %s" % e.status_code) if sess and resp and resp.status_code == 200: self._delete_flavor(sess, flavorid) return Response(data={'error': e.content}, status=e.status_code) except HttpError as e: logger.error("HttpError: status:%s, response:%s" % (e.http_status, e.response.json())) return Response(data=e.response.json(), status=e.http_status) except Exception as e: logger.error(traceback.format_exc()) if sess and resp and resp.status_code == 200: self._delete_flavor(sess, flavorid) return Response(data={'error': str(e)}, status=status.HTTP_500_INTERNAL_SERVER_ERROR) def _create_flavor(self, sess, request): # prepare request resource to vim instance req_resouce = "/flavors" flavor = request.data VimDriverUtils.replace_key_by_mapping(flavor, self.keys_mapping, True) req_body = json.JSONEncoder().encode({"flavor": flavor}) logger.info("making request with URI:%s" % req_resouce) logger.debug("with data:%s" % req_body) resp = sess.post(req_resouce, data=req_body, endpoint_filter=self.service) logger.info("request returns with status %s" % resp.status_code) return resp def _create_flavor_extra_specs(self, sess, extraspecs, flavorid): # prepare request resource to vim instance req_resouce = "/flavors" if flavorid: req_resouce += "/%s/os-extra_specs" % flavorid else: raise VimDriverNewtonException(message="VIM newton exception", content="internal bug in creating flavor extra specs", status_code=500) req_body = json.JSONEncoder().encode({"extra_specs": extraspecs}) logger.info("making request with URI:%s" % req_resouce) logger.debug("with data:%s" % req_body) resp = sess.post(req_resouce, data=req_body, endpoint_filter=self.service) logger.info("request returns with status %s" % resp.status_code) return resp def delete(self, request, vimid="", tenantid="", flavorid=""): logger.info("vimid, tenantid, flavorid = %s,%s,%s" % (vimid, tenantid, flavorid)) if request.data: logger.debug("With data = %s" % request.data) pass try: # prepare request resource to vim instance vim = VimDriverUtils.get_vim_info(vimid) sess = VimDriverUtils.get_session(vim, tenantid) self.service['region_name'] = vim['openstack_region_id'] \ if vim.get('openstack_region_id') \ else vim['cloud_region_id'] #delete extra specs one by one resp = self._delete_flavor_extra_specs(sess, flavorid) #delete flavor resp = self._delete_flavor(sess, flavorid) #return results return Response(status=resp.status_code) except VimDriverNewtonException as e: logger.error("response with status = %s" % e.status_code) return Response(data={'error': e.content}, status=e.status_code) except HttpError as e: logger.error("HttpError: status:%s, response:%s" % (e.http_status, e.response.json())) return Response(data=e.response.json(), status=e.http_status) except Exception as e: logger.error(traceback.format_exc()) return Response(data={'error': str(e)}, status=status.HTTP_500_INTERNAL_SERVER_ERROR) def _delete_flavor_extra_specs(self, sess, flavorid): #delete extra specs one by one resp = self._get_flavor_extra_specs(sess, flavorid) extra_specs = resp.json() if extra_specs and extra_specs["extra_specs"]: for k, _ in list(extra_specs["extra_specs"].items()): # just try to delete extra spec, but do not care if succeeded self._delete_flavor_one_extra_spec(sess, flavorid, k) return resp def _delete_flavor_one_extra_spec(self, sess, flavorid, extra_spec_key): # prepare request resource to vim instance try: req_resouce = "/flavors" if flavorid and extra_spec_key: req_resouce += "/%s" % flavorid req_resouce += "/os-extra_specs/%s" % extra_spec_key else: raise VimDriverNewtonException(message="VIM newton exception", content="internal bug in deleting flavor extra specs: %s" % extra_spec_key, status_code=status.HTTP_500_INTERNAL_SERVER_ERROR) logger.info("making request with URI:%s" % req_resouce) resp = sess.delete(req_resouce, endpoint_filter=self.service) logger.info("request returns with status %s" % resp.status_code) return resp except HttpError as e: logger.error("HttpError: status:%s, response:%s" % (e.http_status, e.response.json())) return Response(data=e.response.json(), status=e.http_status) except Exception as e: logger.error(traceback.format_exc()) return Response(data={'error': str(e)}, status=status.HTTP_500_INTERNAL_SERVER_ERROR) def _delete_flavor(self, sess, flavorid): # prepare request resource to vim instance req_resouce = "/flavors" if flavorid: req_resouce += "/%s" % flavorid else: raise VimDriverNewtonException(message="VIM newton exception", content="internal bug in deleting flavor", status_code=500) logger.info("making request with URI:%s" % req_resouce) resp = sess.delete(req_resouce, endpoint_filter=self.service) logger.info("request returns with status %s" % resp.status_code) return resp class APIv1Flavors(Flavors): def __init__(self): super(APIv1Flavors, self).__init__() self._logger = logger def get(self, request, cloud_owner="", cloud_region_id="", tenantid="", flavorid=""): self._logger.info("%s, %s" % (cloud_owner, cloud_region_id)) vimid = extsys.encode_vim_id(cloud_owner, cloud_region_id) return super(APIv1Flavors, self).get(request, vimid, tenantid, flavorid) def post(self, request, cloud_owner="", cloud_region_id="", tenantid="", flavorid=""): self._logger.info("%s, %s" % (cloud_owner, cloud_region_id)) vimid = extsys.encode_vim_id(cloud_owner, cloud_region_id) return super(APIv1Flavors, self).post(request, vimid, tenantid, flavorid) def delete(self, request, cloud_owner="", cloud_region_id="", tenantid="", flavorid=""): self._logger.info("%s, %s" % (cloud_owner, cloud_region_id)) vimid = extsys.encode_vim_id(cloud_owner, cloud_region_id) return super(APIv1Flavors, self).delete(request, vimid, tenantid, flavorid)