catalog/pub/utils/restcall.py

   1 # Copyright 2017 ZTE Corporation.
   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 # See the License for the specific language governing permissions and
  13 # limitations under the License.
  14
  15 import sys
  16 import traceback
  17 import logging
  18 import urllib
  19 import uuid
  20 import httplib2
  21 import base64
  22
  23 from catalog.pub.config.config import MSB_SERVICE_IP, MSB_SERVICE_PORT
  24
  25 rest_no_auth, rest_oneway_auth, rest_bothway_auth = 0, 1, 2
  26 HTTP_200_OK, HTTP_201_CREATED, HTTP_204_NO_CONTENT, HTTP_202_ACCEPTED = '200', '201', '204', '202'
  27 status_ok_list = [HTTP_200_OK, HTTP_201_CREATED, HTTP_204_NO_CONTENT, HTTP_202_ACCEPTED]
  28 HTTP_404_NOTFOUND, HTTP_403_FORBIDDEN, HTTP_401_UNAUTHORIZED, HTTP_400_BADREQUEST = '404', '403', '401', '400'
  29
  30 logger = logging.getLogger(__name__)
  31
  32
  33 def call_req(base_url, user, passwd, auth_type, resource, method, content='', additional_headers={}):
  34     callid = str(uuid.uuid1())
  35     logger.debug("[%s]call_req('%s','%s','%s',%s,'%s','%s','%s')" % (
  36         callid, base_url, user, passwd, auth_type, resource, method, content))
  37     ret = None
  38     resp_status = ''
  39     try:
  40         full_url = combine_url(base_url, resource)
  41         headers = {'content-type': 'application/json', 'accept': 'application/json'}
  42         if user:
  43             headers['Authorization'] = 'Basic %s' % base64.b64encode(bytes('%s:%s' % (user, passwd), "utf-8")).decode()
  44         ca_certs = None
  45         if additional_headers:
  46             headers.update(additional_headers)
  47         for retry_times in range(3):
  48             http = httplib2.Http(ca_certs=ca_certs, disable_ssl_certificate_validation=(auth_type == rest_no_auth))
  49             http.follow_all_redirects = True
  50             try:
  51                 resp, resp_content = http.request(full_url, method=method.upper(), body=content, headers=headers)
  52                 resp_status, resp_body = resp['status'], resp_content
  53                 logger.debug("[%s][%d]status=%s)" % (callid, retry_times, resp_status))
  54                 if headers['accept'] == 'application/json':
  55                     resp_body = resp_content.decode('UTF-8')
  56                     logger.debug("resp_body=%s", resp_body)
  57                 if resp_status in status_ok_list:
  58                     ret = [0, resp_body, resp_status]
  59                 else:
  60                     ret = [1, resp_body, resp_status]
  61                 break
  62             except Exception as ex:
  63                 if 'httplib.ResponseNotReady' in str(sys.exc_info()):
  64                     logger.debug("retry_times=%d", retry_times)
  65                     logger.error(traceback.format_exc())
  66                     ret = [1, "Unable to connect to %s" % full_url, resp_status]
  67                     continue
  68                 raise ex
  69     except urllib.error.URLError as err:
  70         ret = [2, str(err), resp_status]
  71     except Exception as ex:
  72         logger.error(traceback.format_exc())
  73         logger.error("[%s]ret=%s" % (callid, str(sys.exc_info())))
  74         res_info = str(sys.exc_info())
  75         if 'httplib.ResponseNotReady' in res_info:
  76             res_info = "The URL[%s] request failed or is not responding." % full_url
  77         ret = [3, res_info, resp_status]
  78     except:
  79         logger.error(traceback.format_exc())
  80         ret = [4, str(sys.exc_info()), resp_status]
  81
  82     logger.debug("[%s]ret=%s" % (callid, str(ret)))
  83     return ret
  84
  85
  86 def req_by_msb(resource, method, content=''):
  87     base_url = "http://%s:%s/" % (MSB_SERVICE_IP, MSB_SERVICE_PORT)
  88     return call_req(base_url, "", "", rest_no_auth, resource, method, content)
  89
  90
  91 def upload_by_msb(resource, method, file_data={}):
  92     headers = {'Content-Type': 'application/octet-stream'}
  93     full_url = "http://%s:%s/%s" % (MSB_SERVICE_IP, MSB_SERVICE_PORT, resource)
  94     http = httplib2.Http()
  95     resp, resp_content = http.request(full_url, method=method.upper(), body=file_data, headers=headers)
  96     resp_status, resp_body = resp['status'], resp_content.decode('UTF-8')
  97     if resp_status not in status_ok_list:
  98         logger.error("Status code is %s, detail is %s.", resp_status, resp_body)
  99         return [1, "Failed to upload file.", resp_status]
 100     logger.debug("resp_body=%s", resp_body)
 101     return [0, resp_body, resp_status]
 102
 103
 104 def combine_url(base_url, resource):
 105     full_url = None
 106     if base_url.endswith('/') and resource.startswith('/'):
 107         full_url = base_url[:-1] + resource
 108     elif base_url.endswith('/') and not resource.startswith('/'):
 109         full_url = base_url + resource
 110     elif not base_url.endswith('/') and resource.startswith('/'):
 111         full_url = base_url + resource
 112     else:
 113         full_url = base_url + '/' + resource
 114     return full_url