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

View File

@ -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