bounded_subprocess 2.1.0__tar.gz → 2.3.0__tar.gz

{bounded_subprocess-2.1.0 → bounded_subprocess-2.3.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: bounded_subprocess
-Version: 2.1.0
+Version: 2.3.0
 Summary: A library to facilitate running subprocesses that may misbehave.
 Project-URL: Homepage, https://github.com/arjunguha/bounded_subprocess
 Project-URL: Bug Tracker, https://github.com/arjunguha/bounded_subprocess

{bounded_subprocess-2.1.0 → bounded_subprocess-2.3.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "bounded_subprocess"
-version = "2.1.0"
+version = "2.3.0"
 authors = [
   { name="Arjun Guha" },
   { name="Ming-Ho Yee" },

{bounded_subprocess-2.1.0 → bounded_subprocess-2.3.0}/src/bounded_subprocess/bounded_subprocess.py

@@ -7,6 +7,7 @@ from .util import (
     SLEEP_BETWEEN_READS,
     write_loop_sync,
     _STDIN_WRITE_TIMEOUT,
+    SLEEP_BETWEEN_WRITES,
 )
 
 
@@ -16,6 +17,7 @@ def run(
     max_output_size: int = 2048,
     env=None,
     stdin_data: Optional[str] = None,
+    stdin_write_timeout: Optional[int] = None,
 ) -> Result:
     """
     Runs the given program with arguments. After the timeout elapses, kills the process
@@ -24,12 +26,16 @@ def run(
     """
     state = BoundedSubprocessState(args, env, max_output_size, stdin_data is not None)
     if stdin_data is not None:
-        write_loop_sync(
+        ok = write_loop_sync(
             state.write_chunk,
             stdin_data.encode(),
-            _STDIN_WRITE_TIMEOUT,
-            sleep_interval=SLEEP_BETWEEN_READS,
+            stdin_write_timeout if stdin_write_timeout is not None else 15,
+            sleep_interval=SLEEP_BETWEEN_WRITES,
         )
+        if not ok:
+            state.terminate()
+            return Result(True, -1, "", "failed to write to stdin")
+
         state.close_stdin()
 
     # We sleep for 0.1 seconds in each iteration.

{bounded_subprocess-2.1.0 → bounded_subprocess-2.3.0}/src/bounded_subprocess/bounded_subprocess_async.py

@@ -6,6 +6,7 @@ from .util import (
     SLEEP_BETWEEN_READS,
     write_loop_async,
     _STDIN_WRITE_TIMEOUT,
+    SLEEP_BETWEEN_WRITES,
 )
 
 
@@ -15,6 +16,7 @@ async def run(
     max_output_size: int = 2048,
     env=None,
     stdin_data: Optional[str] = None,
+    stdin_write_timeout: Optional[int] = None,
 ) -> Result:
     """
     Runs the given program with arguments. After the timeout elapses, kills the process
@@ -27,13 +29,16 @@ async def run(
     # async here? It's just the sleep between reads.
     state = BoundedSubprocessState(args, env, max_output_size, stdin_data is not None)
     if stdin_data is not None:
-        await write_loop_async(
+        ok = await write_loop_async(
             state.write_chunk,
             stdin_data.encode(),
-            _STDIN_WRITE_TIMEOUT,
-            sleep_interval=SLEEP_BETWEEN_READS,
+            stdin_write_timeout if stdin_write_timeout is not None else 15,
+            sleep_interval=SLEEP_BETWEEN_WRITES,
         )
-        state.close_stdin()
+        if not ok:
+            state.terminate()
+            return Result(True, -1, "", "failed to write to stdin")
+        await state.close_stdin_async(_STDIN_WRITE_TIMEOUT)
 
     # We sleep for 0.1 seconds in each iteration.
     max_iterations = timeout_seconds * 10

{bounded_subprocess-2.1.0 → bounded_subprocess-2.3.0}/src/bounded_subprocess/util.py

@@ -9,7 +9,8 @@ import asyncio
 
 MAX_BYTES_PER_READ = 1024
 SLEEP_BETWEEN_READS = 0.1
-_STDIN_WRITE_TIMEOUT = 1.0
+_STDIN_WRITE_TIMEOUT = 15
+SLEEP_BETWEEN_WRITES = 0.01
 
 
 class Result:
@@ -147,6 +148,20 @@ class BoundedSubprocessState:
         except BrokenPipeError:
             pass
 
+    async def close_stdin_async(self, timeout: int) -> None:
+        if self.p.stdin is None:
+            return
+        for _ in range(timeout):
+            try:
+                self.p.stdin.close()
+                return
+            except BlockingIOError:
+                await asyncio.sleep(1)
+            except BrokenPipeError:
+                return
+
+
+
     def try_read(self) -> bool:
         """
         Reads from the process. Returning False indicates that we should stop

{bounded_subprocess-2.1.0 → bounded_subprocess-2.3.0}/uv.lock

@@ -4,7 +4,7 @@ requires-python = ">=3.9"
 
 [[package]]
 name = "bounded-subprocess"
-version = "1.5.0"
+version = "2.2.0"
 source = { editable = "." }
 dependencies = [
     { name = "typeguard" },