connectors: close stdin after writing any input

Turns out some applications will wait on stdin being closed before
going anything (ollama), see: #1355

Author: Fizzadar

src/pyinfra/connectors/ssh.py

@@ -397,8 +397,10 @@ def execute_command() -> Tuple[int, CommandOutput]:
                 get_pty=_get_pty,
             )
 
+            # Write any stdin and then close it
             if _stdin:
                 write_stdin(_stdin, stdin_buffer)
+            stdin_buffer.close()
 
             combined_output = read_output_buffers(
                 stdout_buffer,

src/pyinfra/connectors/util.py

@@ -44,11 +44,14 @@ def run_local_process(
 ) -> tuple[int, "CommandOutput"]:
     process = Popen(command, shell=True, stdout=PIPE, stderr=PIPE, stdin=PIPE)
 
-    if stdin:
-        write_stdin(stdin, process.stdin)
-
     assert process.stdout is not None
     assert process.stderr is not None
+    assert process.stdin is not None
+
+    # Write any stdin and then close it
+    if stdin:
+        write_stdin(stdin, process.stdin)
+    process.stdin.close()
 
     combined_output = read_output_buffers(
         process.stdout,