feat: 支持自定义网络接入点

Signed-off-by: 金喜 <[email protected]>

Author: alijinxi

dingtalk_stream/stream.py

@@ -22,12 +22,12 @@
 from .frames import EventMessage
 from .frames import CallbackMessage
 from .log import setup_default_logger
-from .utils import DINGTALK_OPENAPI_ENDPOINT
+from .utils import get_dingtalk_endpoint
 from .version import VERSION_STRING
 
 
 class DingTalkStreamClient(object):
-    OPEN_CONNECTION_API = DINGTALK_OPENAPI_ENDPOINT + '/v1.0/gateway/connections/open'
+    OPEN_CONNECTION_API = get_dingtalk_endpoint() + '/v1.0/gateway/connections/open'
     TAG_DISCONNECT = 'disconnect'
 
     def __init__(self, credential: Credential, logger: logging.Logger = None):
@@ -191,7 +191,8 @@ def get_access_token(self):
             'appSecret': self.credential.client_secret,
         }
         try:
-            response = requests.post(DINGTALK_OPENAPI_ENDPOINT + '/v1.0/oauth2/accessToken',
+            url = get_dingtalk_endpoint() + '/v1.0/oauth2/accessToken'
+            response = requests.post(url,
                                      headers=request_headers,
                                      data=json.dumps(values))
             response.raise_for_status()

dingtalk_stream/utils.py

@@ -1,3 +1,11 @@
 # -*- coding:utf-8 -*-
 
+import os
+
 DINGTALK_OPENAPI_ENDPOINT = "https://api.dingtalk.com"
+
+def get_dingtalk_endpoint():
+    endpoint_env = os.getenv('DINGTALK_OPENAPI_ENDPOINT')
+    if endpoint_env:
+        return endpoint_env
+    return DINGTALK_OPENAPI_ENDPOINT

dingtalk_stream/version.py

@@ -1 +1 @@
-VERSION_STRING = '0.19.1'
+VERSION_STRING = '0.20.0'