Skip to content

Commit ff0f040

Browse files
committed
Fix regenerate_token method
1 parent 5e77b0c commit ff0f040

File tree

1 file changed

+22
-1
lines changed

1 file changed

+22
-1
lines changed

src/onelogin/api/client.py

Lines changed: 22 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -245,7 +245,7 @@ def regenerate_token(self):
245245

246246
response = requests.post(url, headers=headers, json=data)
247247
if response.status_code == 200:
248-
token = OneLoginToken(json_data)
248+
token = OneLoginToken(response.json())
249249
self.access_token = token.access_token
250250
self.refresh_token = token.refresh_token
251251
self.expiration = token.created_at + datetime.timedelta(seconds=token.expires_in)
@@ -1942,3 +1942,24 @@ def get_embed_apps(self, token, email):
19421942
except Exception as e:
19431943
self.error = 500
19441944
self.error_description = e.args[0]
1945+
1946+
def execute_call(method, url, headers, params=None):
1947+
response = None
1948+
tries = 0
1949+
while (tries < 2):
1950+
if method == 'get':
1951+
response = requests.get(url, headers=headers, params=data)
1952+
elif method == 'post':
1953+
response = requests.get(url, headers=headers, params=data)
1954+
elif method == 'put':
1955+
response = requests.put(url, headers=headers, params=data)
1956+
elif method == 'delete':
1957+
response = requests.delete(url, headers=headers, params=data)
1958+
1959+
if response.status_code == 401:
1960+
self.clean_error()
1961+
self.prepare_token()
1962+
tries += 1
1963+
else:
1964+
break;
1965+
return response

0 commit comments

Comments
 (0)