dagster-dingtalk 0.1.29a1__py3-none-any.whl → 0.1.30__py3-none-any.whl

dagster_dingtalk/app_client.py

@@ -48,10 +48,17 @@ class DingTalkClient:
 
 
     def __init_clients(self):
-        self.__oapi_client = httpx.Client(base_url="https://oapi.dingtalk.com/")
-
-        self.__api_client = httpx.Client(base_url="https://api.dingtalk.com/")
+        self.__oapi_client = httpx.Client(
+            base_url="https://oapi.dingtalk.com/",
+            timeout=httpx.Timeout(60.0),
+            limits=httpx.Limits(max_connections=100),
+        )
 
+        self.__api_client = httpx.Client(
+            base_url="https://api.dingtalk.com/",
+            timeout=httpx.Timeout(10.0),
+            limits=httpx.Limits(max_connections=100),
+        )
 
     def __file_cache_read(self) -> dict:
         if self.__access_token_file_cache.exists():
@@ -64,13 +71,12 @@ class DingTalkClient:
         else:
             return {}
 
-    def __file_cache_write(self):
-        with self.__token_lock:
-            try:
-                with open(self.__access_token_file_cache, 'wb') as f:
-                    f.write(pickle.dumps(self.__access_token_cache))
-            except Exception as e:
-                logging.error(f"AccessToken 缓存写入失败: {e}")
+    def __file_cache_write(self, access_token_cache: dict):
+        try:
+            with open(self.__access_token_file_cache, 'wb') as f:
+                f.write(pickle.dumps(access_token_cache))
+        except Exception as e:
+            logging.error(f"AccessToken 缓存写入失败: {e}")
 
     def __get_access_token(self) -> str:
         with self.__token_lock:
@@ -87,16 +93,17 @@ class DingTalkClient:
                         json={"appKey": self.__client_id, "appSecret": self.__client_secret},
                     ).json()
                     # 提前 1 分钟进行续期
-                    self.__access_token_cache = {
+                    access_token_cache = {
                         "access_token": response.get("accessToken"),
                         "expire_in": response.get("expireIn") + int(time.time()) - 60
                     }
-                    self.__file_cache_write()
+                    self.__access_token_cache = access_token_cache
+                    self.__file_cache_write(self.__access_token_cache)
                 except Exception as e:
                     logging.error(f"AccessToken 获取失败: {e}")
                     raise
 
-        return self.__access_token_cache["access_token"]
+        return access_token_cache["access_token"]
 
     def oapi(self, method: str, path: str, **kwargs) -> httpx.Response:
         params = kwargs.get("params", {})

{dagster_dingtalk-0.1.29a1.dist-info → dagster_dingtalk-0.1.30.dist-info}/METADATA

@@ -1,10 +1,10 @@
 Metadata-Version: 2.4
 Name: dagster-dingtalk
-Version: 0.1.29a1
+Version: 0.1.30
 Summary: A dagster plugin for the DingTalk
 Author-email: YiZixuan <sqkkyzx@qq.com>
 Requires-Python: >=3.11
-Requires-Dist: dagster
+Requires-Dist: dagster==1.10.0
 Requires-Dist: httpx
 Requires-Dist: pydantic
 Description-Content-Type: text/markdown

{dagster_dingtalk-0.1.29a1.dist-info → dagster_dingtalk-0.1.30.dist-info}/RECORD

@@ -1,9 +1,9 @@
 dagster_dingtalk/__init__.py,sha256=Qw145M6Igl-NXuFUS-SHpKxyqvVboQAksXvhBnv0wBM,425
-dagster_dingtalk/app_client.py,sha256=YbovNnH9a20TN3E5Z8ctuMEOeslonL2oz44ofRcdBpY,35765
+dagster_dingtalk/app_client.py,sha256=dCQo0GMkVnEhpMHrbTm4OmCEYfJrrX19XCEVih5O-dg,36052
 dagster_dingtalk/operations.py,sha256=yH_rXJpWc8rwzkiJqjqHKy5iIED0Xi3qbSOv6SVr-eA,3923
 dagster_dingtalk/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 dagster_dingtalk/resources.py,sha256=RtoK-lbHKknzYDRq1bredLMxHm0GhycJ1IxQ7hjgLBw,17868
 dagster_dingtalk/version.py,sha256=7jeKY88H3WyDN-Lrs5HYlAPbc4flvLTYlLgkgnZOy-g,26
-dagster_dingtalk-0.1.29a1.dist-info/METADATA,sha256=g_Ks_mB3_mLYbVcxW_TFVVZTL2tUL51U1vdEGyiMIZM,7752
-dagster_dingtalk-0.1.29a1.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-dagster_dingtalk-0.1.29a1.dist-info/RECORD,,
+dagster_dingtalk-0.1.30.dist-info/METADATA,sha256=M2rEfThb0FR2vGAIGzTJ1hSncjw9rVPYLF1c-q4yci0,7758
+dagster_dingtalk-0.1.30.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+dagster_dingtalk-0.1.30.dist-info/RECORD,,