Sending To The Stdin Of A Program In Python3
Solution 1:
What you need is something along the lines of (in main.py
):
from subprocess import Popen, PIPE, STDOUT
x = Popen(['some_child.exe', 'parameter'], stdout=PIPE, stdin=PIPE, stderr=STDOUT)
while x.poll() is None:
child_output = x.stdout.readline()
print(child_output)
if b'Do you want to send the argument?'in child_output:
x.stdin.write(b'yes\n')
x.stdin.flush()
x.stdout.close()
x.stdin.close()
You're assuming child.exe
(in your mockup demo, python.exe
) is communicating with main.py
via sys.stdin/stdout
, however these I/O's are used to communicate with the shell that spawned the process.
Much like the childs stdout/stdin
will be communicating with the shell that spawned that process, in this case Popen()
.
Each spawned child process of subprocess.Popen(...)
will be isolated with it's own stdout/stdin/stderr
, otherwise every subprocess would make a huge mess of your main process stdout/stdin. This means you'll have to check for output on that particular subprocess and write to it accordingly as done in the above example.
One way to look at it is this:
You're starting main.py
, and you communicate with it via sys.stdout
and sys.stdin
. Each input()
in main.py
will output something to sys.stdout
so you can read it.
Exactly the same logic applies to child.exe
where every input()
will output something to it's sys.stdout
(- But remember - sys
is not a shared variable across processes).
import sys
if sys.argv[1] == 'start':
inp = input('Do you want to send the argument?\n').lower()
if inp == 'no':
sys.exit()
elif inp == 'yes':
#Somehow send '1' to the stdin of 1.py while it is running
sys.stdout.write('1')
sys.stdout.flush()
But a simple print(1)
would do the same because it will essentially output the 1
to sys.stdout
for you.
Edit 2018: Don't forget to close your inputs and outputs, as they might leave open file descriptors on your file system, hogging resources and causing problems later in life.
Other conveyers of information
Assuming you have control of the code to child.exe
and you can modify the communication pipe in any way, some other options are:
- sockets - Use regular sockets to communicate, on *nix the most efficient would be Unix sockets.
- Some other solutions can be found here: Best way to return a value from a python script
More cautionary tails!
.readline()
will assume there's a\n
somewhere in your data, most likely at the end. I switched to.readline()
for two reasons,.read()
will hang and wait forEOF
unless you specify exactly how many bytes to read, if I'm not out on a bicycle. To be able to read all kinds of output you need to incorporate select.select() into your code - or a buffer of some sort where you callx.stdout.read(1)
to read one byte at a time. Because if you try to read.read(1024)
and there's not 1024 bytes in the buffer, your read will hang until there are 1024 characters.I left a bug in your
child.py
code on purpose (mine works) - It's trivial and basic Python - in hopes that it's a learning experience on how to debug errors (you mentioned you're not good at it, this is a way to learn).
Post a Comment for "Sending To The Stdin Of A Program In Python3"