问题描述

在python代码中直接获取终端执行命令获得的结果,以判断做出下一步动作

补充知识:

    执行一个 Shell 命令行时通常会自动打开三个标准文件,即:

  • 标准输入文件(stdin)
  • 标准输出文件(stdout)
  • 标准错误输出文件(stderr)

stdin通常对应终端的键盘,后两个文件对应终端的屏幕,进程从标准输入文件中得到输入数据,将正常输出数据输出到标准输出文件,而将错误信息送到标准错误文件中。

所以就要使用到subprocess模块:允许我们启动一个新进程,并连接到它们的输入/输出/错误管道,从而获取返回值。

且Popen方法是subprocess的核心,子进程的创建和管理都靠它处理,常用参数有:

  • args:shell命令,可以是字符串或者序列类型(如:list,元组)
  • bufsize:缓冲区大小。当创建标准流的管道对象时使用,默认-1。
    0:不使用缓冲区
    1:表示行缓冲,仅当universal_newlines=True时可用,也就是文本模式
    正数:表示缓冲区大小
    负数:表示使用系统默认的缓冲区大小。
  • stdin, stdout, stderr:分别表示程序的标准输入、输出、错误句柄
  • preexec_fn:只在 Unix 平台下有效,用于指定一个可执行对象(callable object),它将在子进程运行之前被调用
  • shell:如果该参数为 True,将通过操作系统的 shell 执行指定的命令。
  • cwd:用于设置子进程的当前目录。
  • env:用于指定子进程的环境变量。如果 env = None,子进程的环境变量将从父进程中继承
import subprocess
sp = subprocess.Popen("your cmd", stdin=subprocess.PIPE, stdout=subprocess.PIPE,shell=True)
for line in sp.stdout.realines():    
    print line

其中 shell = True 的时候会直接在终端执行整条命令,而 False 的时候需要把命令按空格使用逗号分隔开来,目的是让 Python 清楚这条命令的所有细节,具体相当于

False:cmd =[ 'ls' , '-a' ]  True:cmd = 'ls -a'

又为避免程序死锁,需要用到Popen.communicate()方法,它会把输出放在内存,而不是管道里。communicate()函数返回一个tuple(标准输出和错误),发送数据到标准输入,从标准输出和错误读取数据直到遇到结束符,等待进程结束。


代码示例

import subprocess


def run_cmd(cmd):
    return subprocess.Popen(cmd,
                            shell=False,
                            stdout=subprocess.PIPE,
                            stderr=subprocess.PIPE).communicate()

std_out_and_err = run_cmd(['ls', '-a'])
print(std_out_and_err)
# ('.\n..\ncontinuegetstdout.py\n', '')