You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
122 lines
3.7 KiB
122 lines
3.7 KiB
#!/usr/bin/python
|
|
# -*- coding: UTF-8 -*-
|
|
|
|
import json
|
|
import time
|
|
|
|
import requests
|
|
|
|
from common import SignUtil, RequestTypes
|
|
from common.RequestType import RequestType
|
|
|
|
_headers = {'Accept-Encoding': 'identity'}
|
|
|
|
|
|
class OpenClient:
|
|
"""调用客户端"""
|
|
__app_id = ''
|
|
__private_key = ''
|
|
__url = ''
|
|
|
|
def __init__(self, app_id, private_key, url):
|
|
"""客户端
|
|
|
|
:param app_id: 应用ID
|
|
:type app_id: str
|
|
|
|
:param private_key: 应用私钥
|
|
:type private_key: str
|
|
|
|
:param url: 请求URL
|
|
:type url: str
|
|
"""
|
|
self.__app_id = app_id
|
|
self.__private_key = private_key
|
|
self.__url = url
|
|
|
|
def execute(self, request, token=None):
|
|
"""
|
|
|
|
:param request: 请求对象，BaseRequest的子类
|
|
|
|
:param token: (Optional) token
|
|
:type token: str
|
|
|
|
:return: 返回请求结果
|
|
:rtype: BaseResponse
|
|
"""
|
|
biz_model = request.biz_model
|
|
request_type = request.get_request_type()
|
|
if not isinstance(request_type, RequestType):
|
|
raise Exception('get_request_type返回错误类型，正确方式：RequestTypes.XX')
|
|
|
|
params = biz_model.__dict__
|
|
if request.files is not None:
|
|
response = self._post_file(request, params, token)
|
|
elif request_type == RequestTypes.GET:
|
|
response = self._get(request, params, token)
|
|
elif request_type == RequestTypes.POST_FORM:
|
|
response = self._post_form(request, params, token)
|
|
elif request_type == RequestTypes.POST_JSON:
|
|
response = self._post_json(request, params, token)
|
|
elif request_type == RequestTypes.POST_UPLOAD:
|
|
response = self._post_file(request, params, token)
|
|
else:
|
|
raise Exception('get_request_type设置错误')
|
|
|
|
return self._parse_response(response, request)
|
|
|
|
def _get(self, request, params, token):
|
|
all_params = self._build_params(request, params, token)
|
|
return requests.get(self.__url, all_params, headers=_headers).text
|
|
|
|
def _post_form(self, request, params, token):
|
|
all_params = self._build_params(request, params, token)
|
|
return requests.post(self.__url, data=all_params, headers=_headers).text
|
|
|
|
def _post_json(self, request, params, token):
|
|
all_params = self._build_params(request, params, token)
|
|
return requests.post(self.__url, json=all_params, headers=_headers).text
|
|
|
|
def _post_file(self, request, params, token):
|
|
all_params = self._build_params(request, params, token)
|
|
return requests.request('POST', self.__url, data=all_params, files=request.files, headers=_headers).text
|
|
|
|
def _build_params(self, request, params, token):
|
|
"""构建所有的请求参数
|
|
|
|
:param request: 请求对象
|
|
:type request: request.BaseRequest
|
|
|
|
:param params: 业务请求参数
|
|
:type params: dict
|
|
|
|
:param token: token
|
|
:type token: str
|
|
|
|
:return: 返回请求参数
|
|
:rtype: str
|
|
"""
|
|
all_params = {
|
|
'app_id': self.__app_id,
|
|
'method': request.get_method(),
|
|
'charset': 'UTF-8',
|
|
'sign_type': 'RSA2',
|
|
'timestamp': time.strftime("%Y-%m-%d %H:%M:%S", time.localtime()),
|
|
'version': request.get_version()
|
|
}
|
|
|
|
if token is not None:
|
|
all_params['access_token'] = token
|
|
|
|
# 添加业务参数
|
|
all_params.update(params)
|
|
|
|
# 构建sign
|
|
sign = SignUtil.create_sign(all_params, self.__private_key, 'RSA2')
|
|
all_params['sign'] = sign
|
|
return all_params
|
|
|
|
def _parse_response(self, resp, request):
|
|
response_dict = json.loads(resp)
|
|
return request.parse_response(response_dict)
|
|
|