如果我这样做:
import subprocess
from cStringIO import StringIO
subprocess.Popen(['grep','f'],stdout=subprocess.PIPE,stdin=StringIO('one\ntwo\nthree\nfour\nfive\nsix\n')).communicate()[0]
我得到:
Traceback (most recent call last):
File "<stdin>", line 1, in ?
File "/build/toolchain/mac32/python-2.4.3/lib/python2.4/subprocess.py", line 533, in __init__
(p2cread, p2cwrite,
File "/build/toolchain/mac32/python-2.4.3/lib/python2.4/subprocess.py", line 830, in _get_handles
p2cread = stdin.fileno()
AttributeError: 'cStringIO.StringI' object has no attribute 'fileno'
显然是cStringIO。StringIO对象的嘎嘎声不够接近文件鸭子,不适合subprocess.Popen。我怎么解决这个问题呢?
Popen.communicate()的文档:
请注意,如果要将数据发送到
这个过程是标准的,你需要
创建Popen对象
stdin =管。同样,要得到任何东西
除了结果元组中的None,
你需要给stdout=PIPE和/或
stderr =管。
取代os.popen *
pipe = os.popen(cmd, 'w', bufsize)
# ==>
pipe = Popen(cmd, shell=True, bufsize=bufsize, stdin=PIPE).stdin
警告使用communication()代替
Stdin.write (), stdout.read()或
Stderr.read()避免死锁
到任何其他操作系统管道缓冲区
填充和阻塞孩子
的过程。
所以你的例子可以写成这样:
from subprocess import Popen, PIPE, STDOUT
p = Popen(['grep', 'f'], stdout=PIPE, stdin=PIPE, stderr=STDOUT)
grep_stdout = p.communicate(input=b'one\ntwo\nthree\nfour\nfive\nsix\n')[0]
print(grep_stdout.decode())
# -> four
# -> five
# ->
在Python 3.5+(编码为3.6+)上,你可以使用subprocess.run,将输入作为字符串传递给外部命令,并在一次调用中获得其退出状态和输出作为字符串返回:
#!/usr/bin/env python3
from subprocess import run, PIPE
p = run(['grep', 'f'], stdout=PIPE,
input='one\ntwo\nthree\nfour\nfive\nsix\n', encoding='ascii')
print(p.returncode)
# -> 0
print(p.stdout)
# -> four
# -> five
# ->
在Python 3.7+上这样做:
my_data = "whatever you want\nshould match this f"
subprocess.run(["grep", "f"], text=True, input=my_data)
并且您可能希望添加capture_output=True以获得以字符串形式运行命令的输出。
在旧版本的Python中,将text=True替换为universal_newlines=True:
subprocess.run(["grep", "f"], universal_newlines=True, input=my_data)
这对于grep来说有点过分了,但是通过我的学习,我已经了解了Linux命令expect和python库pexpect
期望:对话与互动程序
pexpect:用于生成子应用程序的Python模块;控制他们;并在他们的输出中响应预期的模式。
import pexpect
child = pexpect.spawn('grep f', timeout=10)
child.sendline('text to match')
print(child.before)
使用pexpect使用交互式shell应用程序(如ftp)非常简单
import pexpect
child = pexpect.spawn ('ftp ftp.openbsd.org')
child.expect ('Name .*: ')
child.sendline ('anonymous')
child.expect ('Password:')
child.sendline ('noah@example.com')
child.expect ('ftp> ')
child.sendline ('ls /pub/OpenBSD/')
child.expect ('ftp> ')
print child.before # Print the result of the ls command.
child.interact() # Give control of the child to the user.