Skip to content

Commit ca8038e

Browse files
committed
Fix issue when connection error
1 parent 3feb0fb commit ca8038e

1 file changed

Lines changed: 6 additions & 2 deletions

File tree

python-lib/rest_api_client.py

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -136,8 +136,12 @@ def request(self, method, url, can_raise_exeption=True, **kwargs):
136136
request_start_time = time.time()
137137
self.time_last_request = request_start_time
138138
error_message = None
139+
status_code = None
140+
response_headers = None
139141
try:
140142
response = self.request_with_redirect_retry(method, url, **kwargs)
143+
status_code = response.status_code
144+
response_headers = response.headers
141145
except Exception as err:
142146
self.pagination.is_last_batch_empty = True
143147
error_message = "Error: {}".format(err)
@@ -146,8 +150,8 @@ def request(self, method, url, can_raise_exeption=True, **kwargs):
146150

147151
request_finish_time = time.time()
148152
self.set_metadata("request_duration", request_finish_time - request_start_time)
149-
self.set_metadata("status_code", response.status_code)
150-
self.set_metadata("response_headers", "{}".format(response.headers))
153+
self.set_metadata("status_code", status_code)
154+
self.set_metadata("response_headers", "{}".format(response_headers))
151155

152156
if error_message:
153157
return {} if self.behaviour_when_error=="ignore" else {DKUConstants.REPONSE_ERROR_KEY: error_message}

0 commit comments

Comments
 (0)