greco: set timeout to 60 seconds (#34989)

This commit is contained in:
Thomas NOËL 2019-07-22 14:55:25 +02:00
parent d0ba673d72
commit af0a5067e0
1 changed files with 5 additions and 3 deletions

View File

@ -93,7 +93,7 @@ class Greco(BaseResource):
headers = {'Authorization': 'Basic %s' % self.token_authorization}
resp = self.requests.post(self.token_url, headers=headers,
data={'grant_type': 'client_credentials'},
verify=self.verify_cert).json()
verify=self.verify_cert, timeout=60).json()
token = '%s %s' % (resp.get('token_type'), resp.get('access_token'))
timeout = int(resp.get('expires_in'))
cache.set(cache_key, token, timeout)
@ -168,13 +168,15 @@ class Greco(BaseResource):
request.headers['Authorization'] = self.instance.get_token()
resp = self.instance.requests.post(request.url, data=request.message,
headers=request.headers,
verify=self.instance.verify_cert)
verify=self.instance.verify_cert,
timeout=60)
if resp.status_code == 401:
# ask for a new token, and retry
request.headers['Authorization'] = self.instance.get_token(renew=True)
resp = self.instance.requests.post(request.url, data=request.message,
headers=request.headers,
verify=self.instance.verify_cert)
verify=self.instance.verify_cert,
timeout=60)
return Reply(resp.status_code, resp.headers, resp.content)
return Client(url=self.wsdl_url, transport=Transport(self, attachments))