Make status_code checks more robust

mnot · Dec 4, 2023 · a523e88 · a523e88
1 parent 4893c21
commit a523e88
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 3 deletions.
diff --git a/redbot/resource/active_check/etag_validate.py b/redbot/resource/active_check/etag_validate.py
@@ -29,7 +29,10 @@ def modify_request_headers(
         return base_headers
 
     def preflight(self) -> bool:
-        if 300 <= self.base.response.status_code <= 399:
+        if (
+            self.base.response.status_code
+            and 300 <= self.base.response.status_code <= 399
+        ):
             return False
         etag = self.base.response.headers.parsed.get("etag", None)
         if etag:

diff --git a/redbot/resource/active_check/lm_validate.py b/redbot/resource/active_check/lm_validate.py
@@ -58,7 +58,10 @@ def modify_request_headers(
         return base_headers
 
     def preflight(self) -> bool:
-        if 300 <= self.base.response.status_code <= 399:
+        if (
+            self.base.response.status_code
+            and 300 <= self.base.response.status_code <= 399
+        ):
             return False
         if self.base.response.headers.parsed.get("last-modified", None):
             return True

diff --git a/redbot/resource/active_check/range.py b/redbot/resource/active_check/range.py
@@ -44,7 +44,10 @@ def modify_request_headers(
         return base_headers
 
     def preflight(self) -> bool:
-        if 300 <= self.base.response.status_code <= 399:
+        if (
+            self.base.response.status_code
+            and 300 <= self.base.response.status_code <= 399
+        ):
             return False
         if self.base.response.status_code == 206:
             return False