请求自动登录digikey同步token接口

lib/apidigikey.py

@@ -41,13 +41,21 @@ class DGTokenCrawler(token_lib.BaseTokenCrawler):
         self.redirect_uri = 'https://bigdata.ichunt.com/digikey_auth.php'
         self._pool = redis.ConnectionPool(host=apidigikey_ip, port=6379, db=0, password=psd)
         self._redis = redis.Redis(connection_pool=self._pool)
+        self.refresh_token_url = "127.0.0.1:8721/refresh"
 
     def get_params_list(self):
         return [[i] for i in range(THREAD_NUM)]
 
     def collect_token(self):
         try:
-            access_token, refresh_token = self.get_token_by_refresh_token()
+            token_json = self.get_token_by_refresh_token()
+            if not token_json.get('access_token', ''):
+                dd_send_msg('digikey自动登录同步token')
+                resp = requests.get(self.refresh_token_url, timeout=(10, 30)).json()
+                if resp['status'] == 500:
+                    raise Exception(resp['msg'])
+                token_json = resp['data']
+            access_token, refresh_token = token_json['access_token'], token_json['refresh_token']
             apidigikey_log.info(access_token)
             apidigikey_log.info(refresh_token)
             return access_token, refresh_token
@@ -83,7 +91,8 @@ class DGTokenCrawler(token_lib.BaseTokenCrawler):
         }
         resp = requests.post(p_url, headers=p_headers, data=p_data, timeout=(15, 20))
         token_json = resp.json()
-        return token_json['access_token'], token_json['refresh_token']
+
+        return token_json
 
     def process(self, is_main):
         access_token, refresh_token = self.collect_token()