redirects control

anaconda · Oct 4, 2024 · f10c47d · f10c47d
1 parent 255d5c2
commit f10c47d
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/ae5_tools/api.py b/ae5_tools/api.py
@@ -539,6 +539,7 @@ def _format_response(self, response, format, columns=None, record_type=None):
     def _api(self, method, endpoint, **kwargs):
         format = kwargs.pop("format", None)
         subdomain = kwargs.pop("subdomain", None)
+        override_redirects = kwargs.pop("override_redirects", False)
         isabs, endpoint = endpoint.startswith("/"), endpoint.lstrip("/")
         if subdomain:
             subdomain += "."
@@ -556,7 +557,10 @@ def _api(self, method, endpoint, **kwargs):
                 allow_retry = False
         while True:
             try:
-                response = getattr(self.session, method)(url, allow_redirects=False, **kwargs)
+                if override_redirects:
+                    response = getattr(self.session, method)(url, **kwargs)
+                else:
+                    response = getattr(self.session, method)(url, allow_redirects=False, **kwargs)
             except requests.exceptions.ConnectionError:
                 raise AEUnexpectedResponseError("Unable to connect", method, url, **kwargs)
             except requests.exceptions.Timeout: