zrb 1.6.1__tar.gz → 1.6.2__tar.gz

{zrb-1.6.1 → zrb-1.6.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: zrb
-Version: 1.6.1
+Version: 1.6.2
 Summary: Your Automation Powerhouse
 Home-page: https://github.com/state-alchemists/zrb
 License: AGPL-3.0-or-later

{zrb-1.6.1 → zrb-1.6.2}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "zrb"
-version = "1.6.1"
+version = "1.6.2"
 description = "Your Automation Powerhouse"
 authors = ["Go Frendi Gunawan <gofrendiasgard@gmail.com>"]
 license = "AGPL-3.0-or-later"

{zrb-1.6.1 → zrb-1.6.2}/src/zrb/util/cmd/command.py

@@ -64,11 +64,25 @@ async def run_command(
         stream, print_method: Callable[..., None], max_lines: int
     ) -> str:
         lines = []
+        if hasattr(stream, "_limit"):
+            # The limit is set to 10 MB
+            stream._limit = 1024 * 1024 * 10
         while True:
-            line = await stream.readline()
+            line = None
+            try:
+                line = await stream.readline()
+                line = line.decode("utf-8").rstrip()
+            except asyncio.exceptions.CancelledError:
+                break
+            except asyncio.exceptions.LimitOverrunError as e:
+                # Recover by reading a limited chunk instead
+                await stream.read(e.consumed)
+                line = "<output line too long>"
+            except BaseException as e:
+                line = f"Error while reading stream {type(e)} {e}"
+                pass
             if not line:
                 break
-            line = line.decode("utf-8").rstrip()
             lines.append(line)
             if len(lines) > max_lines:
                 lines.pop(0)  # Keep only the last max_lines