lcm/pub/utils/restcall.py

   1 # Copyright 2016 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 urllib2
  19 import uuid
  20 import httplib2
  21 import requests
  22
  23 from lcm.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_Location = ''
  39     resp_status = ''
  40     try:
  41         full_url = combine_url(base_url, resource)
  42         headers = {'content-type': 'application/json', 'accept': 'application/json'}
  43         if user:
  44             headers['Authorization'] = 'Basic ' + ('%s:%s' % (user, passwd)).encode("base64")
  45         ca_certs = None
  46         if additional_headers:
  47             headers.update(additional_headers)
  48         for retry_times in range(3):
  49             http = httplib2.Http(ca_certs=ca_certs, disable_ssl_certificate_validation=(auth_type == rest_no_auth))
  50             http.follow_all_redirects = True
  51             try:
  52                 resp, resp_content = http.request(full_url, method=method.upper(), body=content, headers=headers)
  53                 resp_status, resp_body = resp['status'], resp_content
  54                 resp_Location = resp.get('Location', "")
  55                 logger.debug("[%s][%d]status=%s)" % (callid, retry_times, resp_status))
  56                 if headers['accept'] == 'application/json':
  57                     resp_body = resp_content.decode('UTF-8')
  58                     logger.debug("resp_body=%s", resp_body)
  59                 if resp_status in status_ok_list:
  60                     ret = [0, resp_body, resp_status, resp_Location]
  61                 else:
  62                     ret = [1, resp_body, resp_status, resp_Location]
  63                 break
  64             except Exception as ex:
  65                 if 'httplib.ResponseNotReady' in str(sys.exc_info()):
  66                     logger.debug("retry_times=%d", retry_times)
  67                     logger.error(traceback.format_exc())
  68                     ret = [1, "Unable to connect to %s" % full_url, resp_status, resp_Location]
  69                     continue
  70                 raise ex
  71     except urllib2.URLError as err:
  72         ret = [2, str(err), resp_status, resp_Location]
  73     except Exception as ex:
  74         logger.error(traceback.format_exc())
  75         logger.error("[%s]ret=%s" % (callid, str(sys.exc_info())))
  76         res_info = str(sys.exc_info())
  77         if 'httplib.ResponseNotReady' in res_info:
  78             res_info = "The URL[%s] request failed or is not responding." % full_url
  79         ret = [3, res_info, resp_status, resp_Location]
  80     except:
  81         logger.error(traceback.format_exc())
  82         ret = [4, str(sys.exc_info()), resp_status, resp_Location]
  83
  84     logger.debug("[%s]ret=%s" % (callid, str(ret)))
  85     return ret
  86
  87
  88 def req_by_msb(resource, method, content=''):
  89     logger.debug("resource: %s, method: %s, content: %s" % (resource, method, content))
  90     base_url = "http://%s:%s/" % (MSB_SERVICE_IP, MSB_SERVICE_PORT)
  91     return call_req(base_url, "", "", rest_no_auth, resource, method, content)
  92
  93
  94 def upload_by_msb(resource, method, file_data):
  95     headers = {'accept': 'application/json'}
  96     full_url = "http://%s:%s/%s" % (MSB_SERVICE_IP, MSB_SERVICE_PORT, resource)
  97     r = requests.post(full_url, files=file_data, headers=headers)
  98     resp_status, resp_body = str(r.status_code), r.text
  99     if resp_status not in status_ok_list:
 100         logger.error("Status code is %s, detail is %s.", resp_status, resp_body)
 101         return [1, "Failed to upload file.", resp_status]
 102     logger.debug("resp_body=%s", resp_body)
 103     return [0, resp_body, resp_status]
 104
 105
 106 def combine_url(base_url, resource):
 107     full_url = None
 108     if base_url.endswith('/') and resource.startswith('/'):
 109         full_url = base_url[:-1] + resource
 110     elif base_url.endswith('/') and not resource.startswith('/'):
 111         full_url = base_url + resource
 112     elif not base_url.endswith('/') and resource.startswith('/'):
 113         full_url = base_url + resource
 114     else:
 115         full_url = base_url + '/' + resource
 116     return full_url