dagster-dingtalk 0.1.5b2__tar.gz → 0.1.6__tar.gz

{dagster_dingtalk-0.1.5b2 → dagster_dingtalk-0.1.6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: dagster-dingtalk
-Version: 0.1.5b2
+Version: 0.1.6
 Summary: A dagster plugin for the DingTalk
 Author: YiZixuan
 Author-email: sqkkyzx@qq.com

{dagster_dingtalk-0.1.5b2 → dagster_dingtalk-0.1.6}/dagster_dingtalk/resources.py

@@ -176,13 +176,14 @@ class DingTalkResource(ConfigurableResource):
     def _get_access_token(self) -> str:
         access_token_cache = Path("/tmp/.dingtalk_cache")
 
-        if access_token_cache.exists():
+        try:
             with open(access_token_cache, 'rb') as f:
                 all_access_token: Dict[str, Tuple[str, int]] = pickle.loads(f.read())
-        else:
+                access_token, expire_in = all_access_token.get(self.AppID, ('', 0))
+        except Exception as e:
+            print(e)
             all_access_token = {}
-
-        access_token, expire_in = all_access_token.get(self.AppID, ('', 0))
+            access_token, expire_in = all_access_token.get(self.AppID, ('', 0))
 
         if access_token and expire_in < int(time.time()):
             return access_token
@@ -196,29 +197,35 @@ class DingTalkResource(ConfigurableResource):
             expire_in:int = response.json().get("expireIn") + int(time.time()) - 60
             with open(access_token_cache, 'wb') as f:
                 all_access_token[self.AppID] = (access_token, expire_in)
-                f.write(pickle.dumps(access_token_cache))
+                f.write(pickle.dumps(all_access_token))
             return access_token
 
+    def _init_client(self):
+        if not hasattr(self, '_client'):
+            self._client = DingTalkClient(
+                self._get_access_token(),
+                self.AppID,
+                self.AgentID,
+                self.RobotCode or self.ClientId
+            )
+
     def setup_for_execution(self, context: InitResourceContext) -> None:
-        self._client = DingTalkClient(self._get_access_token(), self.AppID, self.AgentID, self.RobotCode or self.ClientId)
+        self._init_client()
 
     def teardown_after_execution(self, context: InitResourceContext) -> None:
         self._client.api.close()
         self._client.oapi.close()
 
     def 智能人事(self):
-        if not hasattr(self, '_client'):
-            self._client = DingTalkClient(self._get_access_token(), self.AppID, self.AgentID, self.RobotCode or self.ClientId)
+        self._init_client()
         return API_智能人事(self._client)
 
     def 通讯录管理(self):
-        if not hasattr(self, '_client'):
-            self._client = DingTalkClient(self._get_access_token(), self.AppID, self.AgentID, self.RobotCode or self.ClientId)
+        self._init_client()
         return API_通讯录管理(self._client)
 
     def 文档文件(self):
-        if not hasattr(self, '_client'):
-            self._client = DingTalkClient(self._get_access_token(), self.AppID, self.AgentID, self.RobotCode or self.ClientId)
+        self._init_client()
         return API_文档文件(self._client)
 
 

{dagster_dingtalk-0.1.5b2 → dagster_dingtalk-0.1.6}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "dagster-dingtalk"
-version = "0.1.5.b2"
+version = "0.1.6"
 description = "A dagster plugin for the DingTalk"
 authors = ["YiZixuan <sqkkyzx@qq.com>"]
 readme = "README.md"