From 41c79c1fdd4a4045a9e6659e3b1487654aeff786 Mon Sep 17 00:00:00 2001 From: caoqianming Date: Thu, 25 Apr 2024 18:25:38 +0800 Subject: [PATCH] fix: kingClient get_token request 2 --- apps/third/king/k.py | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) diff --git a/apps/third/king/k.py b/apps/third/king/k.py index 3de05a8f..46636358 100644 --- a/apps/third/king/k.py +++ b/apps/third/king/k.py @@ -35,23 +35,25 @@ class KingClient(HandleLogMixin): # is_ok, _ = self.request(**kapis['heartbeat'], raise_exception=False, timeout=10) # if is_ok == 'success': # return - if cache.get('king_token_request', None) != 'requesting': + if cache.get('king_token_request') != 'requesting': json = { 'username': settings.KING_USERNAME, 'password': settings.KING_PASSWORD } - cache.set('king_token_request', 'requesting') + cache.set('king_token_request', 'requesting', timeout=None) _, res = self.request( **kapis['login'], json=json, timeout=10) cache.set('king_token', res['Authorization'], timeout=None) - cache.set('king_token_request', 'done') + cache.set('king_token_request', 'done', timeout=None) def request(self, url: str, method: str = 'post', params=dict(), json=dict(), timeout=20, raise_exception=True): if not self.king_enabled: raise ParseError('亚控对接未启用') count = 6 - while cache.get('king_token_request', None) == 'requesting': + while cache.get('king_token_request') == 'requesting': time.sleep(0.5) + if cache.get('king_token_request') == 'done': + break count = count - 1 if count < 0: break