dagster-dingtalk 0.1.5b2__py3-none-any.whl → 0.1.6__py3-none-any.whl

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.dist-info → dagster_dingtalk-0.1.6.dist-info}/METADATA

@@ -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.6.dist-info/RECORD

@@ -0,0 +1,7 @@
+dagster_dingtalk/__init__.py,sha256=ktvoURpkJwIzcyQfUvnel1KA4DukRgavAgLl7f0Cy_0,440
+dagster_dingtalk/operations.py,sha256=xJJlOVmFjpaDTMkHZXxj5LbXqRtIQwREl9ZJdXIMOyE,788
+dagster_dingtalk/resources.py,sha256=P5PW7xpIZ5FXyWsgIODR7RNRzysYeq7q0yrofIA3RdQ,15015
+dagster_dingtalk/version.py,sha256=sXLh7g3KC4QCFxcZGBTpG2scR7hmmBsMjq6LqRptkRg,22
+dagster_dingtalk-0.1.6.dist-info/METADATA,sha256=oQkzvE9EHHrNTT8kMLn2EiQw0Gh0uQKhVxjXAyIAR2Y,1659
+dagster_dingtalk-0.1.6.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
+dagster_dingtalk-0.1.6.dist-info/RECORD,,

dagster_dingtalk-0.1.5b2.dist-info/RECORD

@@ -1,7 +0,0 @@
-dagster_dingtalk/__init__.py,sha256=ktvoURpkJwIzcyQfUvnel1KA4DukRgavAgLl7f0Cy_0,440
-dagster_dingtalk/operations.py,sha256=xJJlOVmFjpaDTMkHZXxj5LbXqRtIQwREl9ZJdXIMOyE,788
-dagster_dingtalk/resources.py,sha256=zZkrrSff8lW59D6FEw1_01GyhcMAgL6TN3hOD21p9AE,15159
-dagster_dingtalk/version.py,sha256=sXLh7g3KC4QCFxcZGBTpG2scR7hmmBsMjq6LqRptkRg,22
-dagster_dingtalk-0.1.5b2.dist-info/METADATA,sha256=xeX9IPCeb_vufDg7NvaATUO5PKfLFeFgx_lb_7NdVh4,1661
-dagster_dingtalk-0.1.5b2.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
-dagster_dingtalk-0.1.5b2.dist-info/RECORD,,