您的位置:首页 > 运维架构 > Shell

Python获取shell管道数据/输入的2种方法: subprocess子进程 和 sys.stdin

2013-07-05 15:06 2096 查看
今天发现了更方便的方式,就是用subprocess模块,创建子进程,然后用管道来进行交互,而这种方式在shell中非常普遍,比如:cat xxx.file | test.py 就是用的管道,另外,在hadoop中stream模式就是用的管道。
其实在python中,和shell脚本,其他程序交互的方式有很多,比如:
os.system(cmd),os.system只是执行一个shell命令,不能输入、且无返回
os.open(cmd),可以交互,但是是一次性的,调用都少次都会创建和销毁多少次进程,性能太差

所以,建议用subprocess,但是subprocess复杂一些,可以参考python docs: http://docs.python.org/library/subprocess.html
先看一个简单的例子,调用ls命令,两者之间是没有交互的:
[python] view plaincopyimport subprocess p = subprocess.Popen('ls')

再看在程序中获取输出的例子:
[c-sharp] view plaincopyimport subprocess p = subprocess.Popen('ls',stdout=subprocess.PIPE) print p.stdout.readlines()

再看看有输入,有输出的例子,父进程发送'say hi',子进程输出 test say hi,父进程获取输出并打印

[python] view plaincopy#test1.py import sys line = sys.stdin.readline() print 'test',line #run.py from subprocess import * p =Popen('./test1.py',stdin=PIPE,stdout=PIPE) p.stdin.write('say hi/n') print p.stdout.readline() #result test say hi

看看连续输入和输出的例子
test.py
[python] view plaincopyimport sys while True: line = sys.stdin.readline() if not line:break sys.stdout.write(line) sys.stdout.flush()

run.py
[python] view plaincopyimport sys from subprocess import * proc = Popen('./test.py',stdin=PIPE,stdout=PIPE,shell=True) for line in sys.stdin: proc.stdin.write(line) proc.stdin.flush() output = proc.stdout.readline() sys.stdout.write(output)

注意,run.py的flush和test.py中的flush,要记得清空缓冲区,否则程序得不到正确的输入和输出
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: