Merge pull request #767 from python-discord/extend-download-timeout

Use a httpx.Client for netlify static builds

1 files changed, 8 insertions, 3 deletions

diff --git a/static-builds/netlify_build.py b/static-builds/netlify_build.py
index f3a53f72..36520c28 100644
--- a/static-builds/netlify_build.py
+++ b/static-builds/netlify_build.py
@@ -28,6 +28,11 @@ def raise_response(response: httpx.Response) -> None:
 
 
 if __name__ == "__main__":
+    client = httpx.Client(
+        follow_redirects=True,
+        timeout=3 * 60,
+    )
+
     owner, repo = parse.urlparse(os.getenv("REPOSITORY_URL")).path.lstrip("/").split("/")[0:2]
 
     download_url = "/".join([
@@ -40,19 +45,19 @@ if __name__ == "__main__":
         os.getenv("ARTIFACT_NAME"),
     ])
     print(f"Fetching download URL from {download_url}")
-    response = httpx.get(download_url, follow_redirects=True)
+    response = client.get(download_url)
     raise_response(response)
 
     # The workflow is still pending, retry in a bit
     while response.status_code == 202:
         print(f"{response.json()['error']}. Retrying in 10 seconds.")
         time.sleep(10)
-        response = httpx.get(download_url, follow_redirects=True)
+        response = client.get(download_url)
 
     raise_response(response)
     url = response.json()["url"]
     print(f"Downloading build from {url}")
-    zipped_content = httpx.get(url, follow_redirects=True, timeout=3 * 60)
+    zipped_content = client.get(url)
     zipped_content.raise_for_status()
 
     zip_file = Path("temp.zip")