python-ecosys/urequests: add streaming API support

mkomon · mkomon · commit 25ebe4a261e7 · 2021-05-30T23:27:59.000+02:00
-implement iterator protocol for Response so it can be used with streaming APIs
-set socket blocking to False for responses to requests with `stream` argument so that stream APIs can be checked from any loop
diff --git a/python-ecosys/urequests/urequests.py b/python-ecosys/urequests/urequests.py
@@ -7,6 +7,17 @@ def __init__(self, f):
         self.encoding = "utf-8"
         self._cached = None
 
+    def __iter__(self):
+        return self
+
+    def __next__(self):
+        data = b""
+        chunk = self.raw.read()
+        while chunk:
+            data += chunk
+            chunk = self.raw.read()
+        return data
+
     def close(self):
         if self.raw:
             self.raw.close()
@@ -105,6 +116,10 @@ def request(method, url, data=None, json=None, headers={}, stream=None):
     resp = Response(s)
     resp.status_code = status
     resp.reason = reason
+
+    if stream:
+        resp.raw.setblocking(False)
+
     return resp
 
 
