Skip to content

Commit b03804b

Browse files
committed
more fixes
1 parent 013b883 commit b03804b

2 files changed

Lines changed: 9 additions & 9 deletions

File tree

datacrunch/authentication/authentication.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -80,7 +80,7 @@ def refresh(self) -> dict:
8080
url, json=payload, headers=self._generate_headers())
8181

8282
# if refresh token is also expired, authenticate again:
83-
if response.status_code == 400:
83+
if response.status_code == 401 or response.status_code == 400:
8484
return self.authenticate()
8585
else:
8686
handle_error(response)

datacrunch/http_client/http_client.py

Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -49,11 +49,11 @@ def post(self, url: str, json: dict = None, **kwargs) -> requests.Response:
4949
:return: Response object
5050
:rtype: requests.Response
5151
"""
52+
self._refresh_token_if_expired()
53+
5254
url = self._add_base_url(url)
5355
headers = self._generate_headers()
5456

55-
self._refresh_token_if_expired()
56-
5757
response = requests.post(url, json=json, headers=headers, **kwargs)
5858
handle_error(response)
5959

@@ -76,11 +76,11 @@ def put(self, url: str, json: dict = None, **kwargs) -> requests.Response:
7676
:return: Response object
7777
:rtype: requests.Response
7878
"""
79+
self._refresh_token_if_expired()
80+
7981
url = self._add_base_url(url)
8082
headers = self._generate_headers()
8183

82-
self._refresh_token_if_expired()
83-
8484
response = requests.put(url, json=json, headers=headers, **kwargs)
8585
handle_error(response)
8686

@@ -103,11 +103,11 @@ def get(self, url: str, params: dict = None, **kwargs) -> requests.Response:
103103
:return: Response object
104104
:rtype: requests.Response
105105
"""
106+
self._refresh_token_if_expired()
107+
106108
url = self._add_base_url(url)
107109
headers = self._generate_headers()
108110

109-
self._refresh_token_if_expired()
110-
111111
response = requests.get(url, params=params, headers=headers, **kwargs)
112112
handle_error(response)
113113

@@ -130,11 +130,11 @@ def delete(self, url: str, json: dict = None, **kwargs) -> requests.Response:
130130
:return: Response object
131131
:rtype: requests.Response
132132
"""
133+
self._refresh_token_if_expired()
134+
133135
url = self._add_base_url(url)
134136
headers = self._generate_headers()
135137

136-
self._refresh_token_if_expired()
137-
138138
response = requests.delete(url, headers=headers, json=json, **kwargs)
139139
handle_error(response)
140140

0 commit comments

Comments
 (0)