api_client: don't accept cookies

Flared · Aug 20, 2024 · 5013272 · 5013272
1 parent 7c38128
commit 5013272
Showing 1 changed file with 17 additions and 7 deletions.
diff --git a/flareio/api_client.py b/flareio/api_client.py
@@ -2,6 +2,7 @@
 
 from datetime import datetime
 from datetime import timedelta
+from http.cookiejar import DefaultCookiePolicy
 from requests.adapters import HTTPAdapter
 from urllib.parse import urljoin
 from urllib.parse import urlparse
@@ -32,19 +33,28 @@ def __init__(
     @staticmethod
     def _create_session() -> requests.Session:
         session = requests.Session()
-        retries = Retry(
-            total=5,
-            backoff_factor=2,
-            status_forcelist=[429, 502, 503, 504],
-            allowed_methods={"GET", "POST"},
-            backoff_max=15,
+
+        # Don't accept cookies.
+        session.cookies.set_policy(
+            policy=DefaultCookiePolicy(
+                allowed_domains=[],
+            ),
         )
+
+        # Enable retries
         session.mount(
             "https://",
             HTTPAdapter(
-                max_retries=retries,
+                max_retries=Retry(
+                    total=5,
+                    backoff_factor=2,
+                    status_forcelist=[429, 502, 503, 504],
+                    allowed_methods={"GET", "POST"},
+                    backoff_max=15,
+                )
             ),
         )
+
         return session
 
     def generate_token(self) -> str: