Skip to content

Commit 9eb4d33

Browse files
authored
Use requests_retry_session to retry get_reviews/dismiss_review GitHub requests. (#574)
1 parent 890defd commit 9eb4d33

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

scripts/gha/github.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -168,7 +168,7 @@ def dismiss_review(token, pull_number, review_id, message):
168168
url = f'{FIREBASE_URL}/pulls/{pull_number}/reviews/{review_id}/dismissals'
169169
headers = {'Accept': 'application/vnd.github.v3+json', 'Authorization': f'token {token}'}
170170
data = {'message': message}
171-
with requests.put(url, headers=headers, data=json.dumps(data),
171+
with requests_retry_session().put(url, headers=headers, data=json.dumps(data),
172172
stream=True, timeout=TIMEOUT) as response:
173173
logging.info("dismiss_review: %s response: %s", url, response)
174174
return response.json()
@@ -185,7 +185,7 @@ def get_reviews(token, pull_number):
185185
params = {'per_page': per_page, 'page': page}
186186
page = page + 1
187187
keep_going = False
188-
with requests.get(url, headers=headers, params=params,
188+
with requests_retry_session().get(url, headers=headers, params=params,
189189
stream=True, timeout=TIMEOUT) as response:
190190
logging.info("get_reviews: %s response: %s", url, response)
191191
results = results + response.json()

0 commit comments

Comments
 (0)