urequests: Added support for redirects.

diefesson · pi-anl · commit c2d466101265 · 2022-06-19T10:09:43.000+10:00
diff --git a/python-ecosys/urequests/urequests.py b/python-ecosys/urequests/urequests.py
@@ -36,6 +36,7 @@ def json(self):
 def request(
     method, url, data=None, json=None, headers={}, stream=None, parse_headers=True, auth=None
 ):
+    redirect = None  # redirection url, None means no redirection
     chunked = data and is_chunked_data(data)
 
     if auth is not None:
@@ -119,7 +120,10 @@ def request(
                 if b"chunked" in l:
                     raise ValueError("Unsupported " + l.decode())
             elif l.startswith(b"Location:") and not 200 <= status <= 299:
-                raise NotImplementedError("Redirects not yet supported")
+                if status in [301, 302, 303, 307, 308]:
+                    redirect = l[10:-2].decode()
+                else:
+                    raise NotImplementedError("Redirect %d not yet supported" % status)
             if parse_headers is False:
                 pass
             elif parse_headers is True:
@@ -132,12 +136,19 @@ def request(
         s.close()
         raise
 
-    resp = Response(s)
-    resp.status_code = status
-    resp.reason = reason
-    if resp_d is not None:
-        resp.headers = resp_d
-    return resp
+    if redirect:
+        s.close()
+        if status in [301, 302, 303]:
+            return request("GET", redirect, None, None, headers, stream)
+        else:
+            return request(method, redirect, data, json, headers, stream)
+    else:
+        resp = Response(s)
+        resp.status_code = status
+        resp.reason = reason
+        if resp_d is not None:
+            resp.headers = resp_d
+        return resp
 
 
 def head(url, **kw):
