python - 使用subprocess调用外部程序，p.stdout读取内容卡死

Question

{代码...} 这个卡死如何解决，这时候结束当前进程也没办法结束或者如何判断卡死，然后p.kill()或“ctrl+c”当前进程ssh执行p.stdin.write('exit
')也不行，因为ssh exit之后还会跳回登录界面，进程还是没有结束

ringa_lee · Answer

while p.stdout.readline()!=' ':
    line = p.stdout.readline()    #如果p.stdout中内容被读完之后，程序会卡在这里
    #line = line.strip()
    print line
    r.write(line)

首先，我指出这段是有问题的，这个是打印问题， p.stdout.readline() != ''，不应该放在while中的，因为读取数据后，就会舍弃掉了，这样就不能把完整数据打印了，和写入到sad.txt文件了,
其次我认为应该给p.stdout 搞成非阻塞,就可以解决卡死问题

import subprocess
import os
import fcntal

r=open("sad.txt",'a')
p = subprocess.Popen("ssh.exe root@192.168.58.154", stdin = subprocess.PIPE, stdout = subprocess.PIPE, stderr = subprocess.PIPE, shell = False)
flags = fcntl.fcntl(p.stdout, fcntl.F_GETFL)
flags |= os.O_NONBLOCK
fcntl.fcntl(p.stdout, fcntl.F_SETFL, flags)
p.stdin.write('password
')
p.stdin.write('ifconfig
')

while True:
    try:
        line = p.stdout.readline()    #如果p.stdout中内容被读完之后，程序会卡在这里
        if line == '':
            break
        #line = line.strip()
        print line
        r.write(line)
    except IOError:
        break

PHPz · Answer

这个卡死应该是由于ssh这个进程没退出造成的。

高洛峰 · Answer

之所以卡死是因为读完最后一行后管道空了，而且针对ifconfig命令来说，最后一行不是空格而是 :

p.stdin.write('ifconfig eth0
')
for line in iter(p.stdout.readline, '
'):
    print line.strip()

为防止出现因IO引起的死等，官方推荐的做法是使用communicate，比如这样：

out, err = p.communicate('ifconfig')