Если вы знаете, как выглядит приглашение, вы можете использовать строковые манипуляции, чтобы получить желаемый результат как @ Майкл предложил :
import sys
from subprocess import Popen, PIPE
p = Popen([sys.executable, "names.py"], stdin=PIPE, stdout=PIPE,
universal_newlines=True)
Вывод = p.communicate("Sue\n")[0]
prompt = "name? "
print(Вывод.replace(prompt, prompt + "\n"))
Вывод
I am Joe.
What is your name?
Hi Sue !
If you don't know how the prompt looks like then the timeout-based solution suggested by @Exp might not work if the child process uses block-buffering when it is run non-interactively. Though it does work for the names.py
. Here's a timeout-based solution that uses select
instead of threads to read the Вывод:
import os
import sys
from select import select
from subprocess import Popen, PIPE
timeout = 1 # seconds
with Popen([sys.executable, 'names.py'],
stdin=PIPE, stdout=PIPE, bufsize=0) as p:
while True:
ready = select([p.stdout], [], [], timeout)[0]
if ready: # there is something to read
data = os.read(p.stdout.fileno(), 512)
if not data: # EOF
break
sys.stdout.buffer.write(data) # echo subprocess Вывод
elif p.poll() is None: # timeout, but subprocess is still running
print("") # print newline after the prompt
p.stdin.write(b"Sue\n") # answer the prompt
else: # subprocess exited
break
It produces the same Вывод as the first code example after a delay.
В общем случае pexpect
может использоваться для эмуляции интерактивного режима для подпроцесса.
Если вы знаете, как выглядит приглашение:
import sys
import pexpect
print(pexpect.run(sys.executable + " -mnames", events={r'name\? ': 'Sue\n'}))
# note: it echos our answer too (it can be avoided if necessary)
Вывод
I am Joe.
What is your name? Sue
Hi Sue !
Вот решение на основе тайм-аута, которое позволяет избежать повторения ответа:
import sys
import pexpect # pip install pexpect-u
child = pexpect.spawn(sys.executable + " -mnames", timeout=1)
child.logfile_read = sys.stdout # echo subprocess Вывод
child.expect(pexpect.TIMEOUT)
print("") # print newline after the prompt
child.setecho(False) # don't echo our answer
child.sendline('Sue')
child.expect(pexpect.EOF)
child.close()
Чтобы воспроизвести его с помощью subprocess
, можно использовать модуль pty
:
import os
import pty
import sys
from select import select
from subprocess import Popen, STDOUT
timeout = 1 # seconds
master_fd, slave_fd = pty.openpty()
with Popen([sys.executable, 'names.py'],
stdin=slave_fd, stdout=slave_fd, stderr=STDOUT,
bufsize=0) as p:
while True:
ready = select([master_fd], [], [], timeout)[0]
if ready: # there is something to read
data = os.read(master_fd, 512)
if not data: # EOF
break
sys.stdout.buffer.write(data) # echo subprocess Вывод
elif p.poll() is None: # timeout, but subprocess is still running
# assume that child process waits for input after printing the prompt
answer = b"Sue\n"
os.write(master_fd, answer) # asnwer the prompt
os.read(master_fd, len(answer)) # don't echo our answer
else: # subprocess exited
break
os.close(slave_fd)
os.close(master_fd)