3.2.0

sop-sdk/sdk-python/common/OpenClient.py

@@ -0,0 +1,122 @@
+#!/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)