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 from multivimbroker.pub.utils.restcall import req_by_msb_multipart
  23
  24
  25 logger = logging.getLogger(__name__)
  26
  27
  28 class BaseHandler(object):
  29
  30     def _request(self, route_uri, method, body="", headers=None):
  31
  32         try:
  33             if "multipart" in route_uri:
  34                 return self._multipart_req(route_uri, method, body, headers)
  35             retcode, content, status_code, resp = \
  36                 req_by_msb(route_uri, method, body, headers)
  37             if retcode != 0:
  38                 # Execptions are handled within req_by_msb
  39                 logger.error("Status code is %s, detail is %s.",
  40                              status_code, content)
  41
  42         except exceptions.NotFound as e:
  43             return HttpResponse(str(e), status=status.HTTP_404_NOT_FOUND)
  44
  45         except Exception as e:
  46             content = e
  47             status_code = status.HTTP_500_INTERNAL_SERVER_ERROR
  48             logger.exception("exception: %s" % e)
  49
  50         response = HttpResponse(content, status=status_code)
  51         if retcode == 0:
  52             for k in getHeadersKeys(resp):
  53                 response[k] = resp[k]
  54         return response
  55
  56     def _multipart_req(self, route_uri, method, body, headers=None):
  57
  58         try:
  59             retcode, content, status_code, resp = \
  60                 req_by_msb_multipart(route_uri, method, body, headers)
  61             if retcode != 0:
  62                 # Execptions are handled within req_by_msb
  63                 logger.error("Status code is %s, detail is %s.",
  64                              status_code, content)
  65
  66         except exceptions.NotFound as e:
  67             return HttpResponse(str(e), status=status.HTTP_404_NOT_FOUND)
  68
  69         except Exception as e:
  70             content = e
  71             status_code = status.HTTP_500_INTERNAL_SERVER_ERROR
  72             logger.exception("exception: %s" % e)
  73
  74         response = HttpResponse(content, status=status_code)
  75         return response
  76
  77     def send(self, vimid, full_path, body, method, headers=None):
  78
  79         try:
  80             url = getMultivimDriver(vimid, full_path=full_path)
  81         except exceptions.VimBrokerException as e:
  82             logging.exception("vimbroker exception: %s" % e)
  83             return HttpResponse(e.content, status=e.status_code)
  84         except Exception as e:
  85             logging.exception("unkown exception: %s" % e)
  86             return HttpResponse(str(e),
  87                                 status=status.HTTP_500_INTERNAL_SERVER_ERROR)
  88
  89         return self._request(url, method, body=body, headers=headers)