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

#!/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(),
            'biz_content': json.dumps(params)
        }

        if token is not None:
            all_params['access_token'] = token

        # 构建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)