bounded_subprocess 2.1.0__tar.gz → 2.2.0__tar.gz

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: bounded_subprocess
-Version: 2.1.0
+Version: 2.2.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.2.0}/pyproject.toml

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

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

@@ -33,7 +33,7 @@ async def run(
             _STDIN_WRITE_TIMEOUT,
             sleep_interval=SLEEP_BETWEEN_READS,
         )
-        state.close_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.2.0}/src/bounded_subprocess/util.py

@@ -9,7 +9,7 @@ import asyncio
 
 MAX_BYTES_PER_READ = 1024
 SLEEP_BETWEEN_READS = 0.1
-_STDIN_WRITE_TIMEOUT = 1.0
+_STDIN_WRITE_TIMEOUT = 3
 
 
 class Result:
@@ -147,6 +147,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