Re-try requests on 504 error and raise correct error

This commit is contained in:
Lilly Rose Berner 2021-07-09 10:54:22 +02:00 committed by GitHub
parent 8fb998b599
commit 0aa825557d
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -321,8 +321,8 @@ class HTTPClient:
continue
# we've received a 500 or 502, unconditional retry
if response.status in {500, 502}:
# we've received a 500, 502, or 504, unconditional retry
if response.status in {500, 502, 504}:
await asyncio.sleep(1 + tries * 2)
continue
@ -331,7 +331,7 @@ class HTTPClient:
raise Forbidden(response, data)
elif response.status == 404:
raise NotFound(response, data)
elif response.status == 503:
elif response.status >= 500:
raise DiscordServerError(response, data)
else:
raise HTTPException(response, data)