config file

Author: tigersk

.gitignore

@@ -139,3 +139,4 @@ cython_debug/
 
 .idea
 .DS_Store
+*.properties

requirements.txt

@@ -8,4 +8,5 @@ rsa
 stomp.py
 getmac
 cryptography
-backoff
+backoff
+jproperties

tigeropen/common/consts/service_types.py

@@ -4,7 +4,9 @@
 
 @author: gaoan
 """
+USER_TOKEN_REFRESH = "user_token_refresh"
 
+# 订单
 ORDER_NO = "order_no"
 PREVIEW_ORDER = "preview_order"
 PLACE_ORDER = "place_order"

tigeropen/common/util/web_utils.py

@@ -69,11 +69,13 @@ def do_request(method, url, query_string=None, headers=None, params=None, timeou
         connection.connect()
     except Exception as e:
         raise RequestException('[' + THREAD_LOCAL.uuid + ']' + method + ' connect failed. url: ' + url
+                               + ' headers: ' + str(headers)
                                + ' params: ' + str(params) + ' detail: ' + str(e))
     try:
         connection.request(method, url, body=json.dumps(params), headers=headers)
     except Exception as e:
         raise RequestException('[' + THREAD_LOCAL.uuid + ']' + method + ' request failed. url: ' + url
+                               + ' headers: ' + str(headers)
                                + ' params: ' + str(params) + ' detail: ' + str(e))
     response = connection.getresponse()
     result = response.read()
@@ -82,7 +84,8 @@ def do_request(method, url, query_string=None, headers=None, params=None, timeou
         if charset:
             result = result.decode(charset)
         raise ResponseException('[' + THREAD_LOCAL.uuid + ']invalid http status ' + str(response.status) +
-                                ',detail body:' + result + ' params: ' + str(params))
+                                ' headers: ' + str(headers) +
+                                ' detail body:' + result + ' params: ' + str(params))
     try:
         response.close()
         connection.close()

tigeropen/tiger_open_client.py

@@ -15,7 +15,8 @@
 from tigeropen.common.consts import OPEN_API_SERVICE_VERSION, THREAD_LOCAL
 from tigeropen.common.consts.params import P_TIMESTAMP, P_TIGER_ID, P_METHOD, P_CHARSET, P_VERSION, P_SIGN_TYPE, \
     P_DEVICE_ID, P_NOTIFY_URL, COMMON_PARAM_KEYS, P_SIGN
-from tigeropen.common.consts.service_types import USER_LICENSE, PLACE_ORDER, CANCEL_ORDER, MODIFY_ORDER
+from tigeropen.common.consts.service_types import USER_LICENSE, PLACE_ORDER, CANCEL_ORDER, MODIFY_ORDER, \
+    USER_TOKEN_REFRESH
 from tigeropen.common.exceptions import ResponseException, RequestException
 from tigeropen.common.request import OpenApiRequest
 from tigeropen.common.response import TigerResponse
@@ -58,6 +59,7 @@ def __init__(self, client_config, logger=None):
         self.__device_id = self.__get_device_id()
         self.__init_license()
         self.__refresh_server_info()
+        self.__refresh_token()
 
     def __init_license(self):
         if self.__config.license is None and self.__config.enable_dynamic_domain:
@@ -156,6 +158,9 @@ def __parse_response(self, response_str, timestamp=None):
 
         return response_content
 
+    def _update_header(self):
+        self.__headers['Authorization'] = self.__config.token
+
     def _get_retry_deco(self, service):
         if service not in SKIP_RETRY_SERVICES and self.__config.retry_max_tries > 0:
             return backoff.on_exception(backoff.fibo,
@@ -169,6 +174,7 @@ def _get_retry_deco(self, service):
     执行接口请求
     """
     def execute(self, request, url=None):
+        self._update_header()
         if url is None:
             url = self.__config.server_url
         THREAD_LOCAL.uuid = str(uuid.uuid1())
@@ -198,3 +204,19 @@ def query_license(self):
             if response.is_success():
                 return json.loads(response.data).get('license')
         self.__logger.error(f"failed to query license, response: {response_content}")
+
+    def __refresh_token(self):
+        request = OpenApiRequest(method=USER_TOKEN_REFRESH)
+
+        response_content = None
+        try:
+            response_content = self.execute(request)
+        except Exception as e:
+            self.__logger.error(e, exc_info=True)
+        if response_content:
+            response = TigerResponse()
+            response.parse_response_content(response_content)
+            if response.is_success():
+                return json.loads(response.data).get('token')
+        self.__logger.error(f"failed to refresh token, response: {response_content}")
+

tigeropen/tiger_open_config.py

@@ -6,7 +6,9 @@
 """
 import json
 import logging
+import os
 
+from jproperties import Properties
 from pytz import timezone
 from tigeropen.common.consts import Language, ServiceType
 from tigeropen.common.util.account_util import AccountUtil
@@ -47,9 +49,12 @@
                            'MidihTvHHf+tJ0PYD0o3PruI0hl3qhEjHTAxb75T5YD3SGK4IBhHn/Rk6mhqlGgI+bBrBVYaXixm' \
                            'HfRo75RpUUuWACyeqQkZckgR0McxuW9xRMIa2cXZOoL1E4SL4lXKGhKoWbwIDAQAB'
 
+DEFAULT_PROPS_FILE = 'tiger_openapi_config.properties'
+DEFAULT_TOKEN_FILE = 'tiger_openapi_token.properties'
+
 
 class TigerOpenClientConfig:
-    def __init__(self, sandbox_debug=False, enable_dynamic_domain=True):
+    def __init__(self, sandbox_debug=False, enable_dynamic_domain=True, props_path='.'):
         # 开发者应用id
         self._tiger_id = ''
         # 授权账户
@@ -86,17 +91,21 @@ def __init__(self, sandbox_debug=False, enable_dynamic_domain=True):
             self._quote_server_url = SANDBOX_SERVER_URL
             self._socket_host_port = SANDBOX_SOCKET_HOST_PORT
 
+        self.log_level = None
+        self.log_path = None
+        self.retry_max_time = 60
+        self.retry_max_tries = 5
+        self.props_path = props_path
+        self.token = None
+        self._load_props()
+        self._load_token()
+
         self.domain_conf = dict()
         self.enable_dynamic_domain = enable_dynamic_domain
         if enable_dynamic_domain:
             self.domain_conf = self.query_domains()
             self.refresh_server_info()
 
-        self.log_level = None
-        self.log_path = None
-        self.retry_max_time = 60
-        self.retry_max_tries = 5
-
     @property
     def tiger_id(self):
         return self._tiger_id
@@ -221,6 +230,53 @@ def secret_key(self):
     def secret_key(self, value):
         self._secret_key = value
 
+    @property
+    def token(self):
+        return self._token
+
+    @token.setter
+    def token(self, value):
+        self._token = value
+
+    def _get_props_path(self, filename):
+        if self.props_path is not None and os.path.exists(self.props_path):
+            if os.path.isdir(self.props_path):
+                full_path = os.path.join(self.props_path, filename)
+            else:
+                full_path = self.props_path
+            return full_path
+        return None
+
+    def _load_props(self):
+        full_path = self._get_props_path(DEFAULT_PROPS_FILE)
+        if full_path and os.path.exists(full_path):
+            try:
+                p = Properties()
+                with open(full_path, "rb") as f:
+                    p.load(f, "utf-8")
+                    if not self.tiger_id:
+                        self.tiger_id = getattr(p.get('tiger_id'), 'data', '')
+                    if not self.private_key:
+                        self.private_key = getattr(p.get('private_key_pk1'), 'data', '')
+                    if not self.account:
+                        self.account = getattr(p.get('account'), 'data', '')
+                    if not self.license:
+                        self.license = getattr(p.get('license'), 'data', '')
+            except Exception as e:
+                logging.error(e, exc_info=True)
+
+    def _load_token(self):
+        full_path = self._get_props_path(DEFAULT_TOKEN_FILE)
+        if full_path and os.path.exists(full_path):
+            try:
+                p = Properties()
+                with open(full_path, "r+b") as f:
+                    p.load(f, "utf-8")
+                    if not self.token:
+                        self.token = getattr(p.get('token'), 'data', '')
+            except Exception as e:
+                logging.error(e, exc_info=True)
+
     def refresh_server_info(self):
         if self.enable_dynamic_domain and self.domain_conf:
             if self.license: