zcdsj
/
factory
5
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix: kingClient get_token request 2

This commit is contained in:
caoqianming 2024-04-25 18:25:38 +08:00
parent facecacc38
commit 41c79c1fdd
1 changed files with 6 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
apps/third/king/k.py
View File

@ -35,23 +35,25 @@ class KingClient(HandleLogMixin):
# is_ok, _ = self.request(**kapis['heartbeat'], raise_exception=False, timeout=10) # is_ok, _ = self.request(**kapis['heartbeat'], raise_exception=False, timeout=10)
# if is_ok == 'success': # if is_ok == 'success':
# return # return
if cache.get('king_token_request', None) != 'requesting': if cache.get('king_token_request') != 'requesting':
json = { json = {
'username': settings.KING_USERNAME, 'username': settings.KING_USERNAME,
'password': settings.KING_PASSWORD 'password': settings.KING_PASSWORD
} }
cache.set('king_token_request', 'requesting') cache.set('king_token_request', 'requesting', timeout=None)
_, res = self.request( _, res = self.request(
**kapis['login'], json=json, timeout=10) **kapis['login'], json=json, timeout=10)
cache.set('king_token', res['Authorization'], timeout=None) 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): def request(self, url: str, method: str = 'post', params=dict(), json=dict(), timeout=20, raise_exception=True):
if not self.king_enabled: if not self.king_enabled:
raise ParseError('亚控对接未启用') raise ParseError('亚控对接未启用')
count = 6 count = 6
while cache.get('king_token_request', None) == 'requesting': while cache.get('king_token_request') == 'requesting':
time.sleep(0.5) time.sleep(0.5)
if cache.get('king_token_request') == 'done':
break
count = count - 1 count = count - 1
if count < 0: if count < 0:
break break