fix: cleanup script

This commit is contained in:
Barrett Ruth 2025-10-24 16:17:56 -04:00
parent 11b6056d8c
commit 3fdb74a3d8

View file

@ -12,8 +12,8 @@ async def pump(
data = await reader.readline() data = await reader.readline()
if not data: if not data:
break break
sys.stdout.buffer.write(data) _ = sys.stdout.buffer.write(data)
sys.stdout.flush() _ = sys.stdout.flush()
if writer: if writer:
writer.write(data) writer.write(data)
await writer.drain() await writer.drain()
@ -42,9 +42,9 @@ async def main(interactor_cmd: Sequence[str], interactee_cmd: Sequence[str]) ->
asyncio.create_task(pump(interactor.stdout, interactee.stdin)), asyncio.create_task(pump(interactor.stdout, interactee.stdin)),
asyncio.create_task(pump(interactee.stdout, interactor.stdin)), asyncio.create_task(pump(interactee.stdout, interactor.stdin)),
] ]
await asyncio.wait(tasks, return_when=asyncio.ALL_COMPLETED) _ = await asyncio.wait(tasks, return_when=asyncio.ALL_COMPLETED)
await interactor.wait() _ = await interactor.wait()
await interactee.wait() _ = await interactee.wait()
if __name__ == "__main__": if __name__ == "__main__":
@ -55,4 +55,4 @@ if __name__ == "__main__":
interactor_cmd = shlex.split(sys.argv[1]) interactor_cmd = shlex.split(sys.argv[1])
interactee_cmd = shlex.split(sys.argv[2]) interactee_cmd = shlex.split(sys.argv[2])
asyncio.run(main(interactor_cmd, interactee_cmd)) _ = asyncio.run(main(interactor_cmd, interactee_cmd))