multivimbroker/multivimbroker/forwarder/base.py

   1 # Copyright (c) 2017-2018 VMware, Inc.
   2 #
   3 # Licensed under the Apache License, Version 2.0 (the "License");
   4 # you may not use this file except in compliance with the License.
   5 # You may obtain a copy of the License at:
   6 #
   7 #       http://www.apache.org/licenses/LICENSE-2.0
   8 #
   9 # Unless required by applicable law or agreed to in writing, software
  10 # distributed under the License is distributed on an "AS IS" BASIS,
  11 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  12
  13 import logging
  14
  15 from django.http import HttpResponse
  16 from rest_framework import status
  17
  18 import multivimbroker.pub.exceptions as exceptions
  19 from multivimbroker.pub.utils.syscomm import getHeadersKeys
  20 from multivimbroker.pub.utils.syscomm import getMultivimDriver
  21 from multivimbroker.pub.utils.restcall import req_by_msb
  22
  23
  24 logger = logging.getLogger(__name__)
  25
  26
  27 class BaseHandler(object):
  28
  29     def _request(self, route_uri, method, body="", headers=None):
  30
  31         try:
  32             retcode, content, status_code, resp = \
  33                 req_by_msb(route_uri, method, body, headers)
  34             if retcode != 0:
  35                 # Execptions are handled within req_by_msb
  36                 logger.error("Status code is %s, detail is %s.",
  37                              status_code, content)
  38
  39         except exceptions.NotFound as e:
  40             return HttpResponse(str(e), status=status.HTTP_404_NOT_FOUND)
  41
  42         except Exception as e:
  43             content = e
  44             status_code = status.HTTP_500_INTERNAL_SERVER_ERROR
  45             logger.exception("exception: %s" % e)
  46
  47         response = HttpResponse(content, status=status_code)
  48         for k in getHeadersKeys(resp):
  49             response[k] = resp[k]
  50         return response
  51
  52     def send(self, vimid, full_path, body, method, headers=None):
  53
  54         try:
  55             url = getMultivimDriver(vimid, full_path=full_path)
  56         except exceptions.VimBrokerException as e:
  57             logging.exception("vimbroker exception: %s" % e)
  58             return HttpResponse(e.content, status=e.status_code)
  59         except Exception as e:
  60             logging.exception("unkown exception: %s" % e)
  61             return HttpResponse(str(e),
  62                                 status=status.HTTP_500_INTERNAL_SERVER_ERROR)
  63
  64         return self._request(url, method, body=body, headers=headers)