fix: retry on transient Cloudflare errors

python-poetry · Apr 3, 2023 · 6a57c2b · 6a57c2b
1 parent c776541
commit 6a57c2b
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 1 deletion.
diff --git a/src/poetry/utils/constants.py b/src/poetry/utils/constants.py
@@ -7,4 +7,18 @@
 RETRY_AFTER_HEADER = "retry-after"
 
 # Server response codes to retry requests on.
-STATUS_FORCELIST = [429, 500, 501, 502, 503, 504]
+STATUS_FORCELIST = [
+    429,
+    500,
+    501,
+    502,
+    503,
+    504,
+    # Cloudflare internal errors may indicate transient errors on the origin server
+    # https://github.com/cloudflare/cloudflare-docs/blob/9842ebbd3b3539ee64ac75daf766c091b90c7512/content/spectrum/reference/logs.md?plain=1#L41-L45
+    520,
+    521,
+    522,
+    523,
+    524,
+]
diff --git a/tests/utils/test_authenticator.py b/tests/utils/test_authenticator.py
@@ -282,6 +282,11 @@ def callback(
         (502, 5),
         (503, 5),
         (504, 5),
+        (520, 5),
+        (521, 5),
+        (522, 5),
+        (523, 5),
+        (524, 5),
     ],
 )
 def test_authenticator_request_retries_on_status_code(