Python 调用外部命令

时间：2022-03-31 07:53:26

python 可以使用 os 模块来调用外部的 Linux Shell 命令，常用的方法如下：

os.system()：结果输出在终端上，捕获不到
os.popen() ： 结果返回一个对象，即标准输出
os.popen2()：结果返回两个对象，分别是标准输入，标准输出
os.popen3()：结果返回三个对象，分别是标准输入，标准输出，标准错误输出
os.popen4()：结果返回两个对象，分别是标准输入，标准输出（标准输出中包括标准错误输出）

In [2]: stdout = os.system('ls /data')    # os.system() --- 结果输出在终端上，会返回执行命令的状态码，我们可以用变量来接收

1.txt  2.txt  3.txt

In [3]: print(stdout)

0
In [7]: stdout = os.popen('ls /data')    # os.popen() --- 结果返回一个对象，即标准输出，标准输出需要用read()、readlines()等方法读取

In [8]: print stdout.read()

1.txt

2.txt

3.txt
In [24]: stdin, stdout = os.popen2('cat')    # os.popen2() --- 结果返回两个对象，分别是标准输入，标准输出

In [25]: stdin.write('hello\n')              # 我们使用 write() 来进行输入，使用 close() 来结束输入，使用 read() 来读取输出

In [26]: stdin.write('world\n')

In [27]: stdin.close()

In [28]: print stdout.read()

hello

world

使用 os 模块来调用 shell 命令已经过时了，官方建议我们使用 subprocess 模块，subprocess 会开启一个子进程（即shell）来执行命令：

subprocess.call() ：用于执行 shell 命令，但结果不能被捕获到，只会输出到终端，如 subprocess.call('ls -l /data', shell=True)
subprocess.check_call() ：用法与 call() 一致，不同的是，如果命令执行失败，会抛出 CalledProcessError 异常，我们可以根据这个异常去决定需要做什么
subprocess.Popen() ：用于执行 shell 命令，结果返回三个对象，分别是标准输入，标准输出，标准错误输出

In [1]: from subprocess import Popen, PIPE

In [2]: p = Popen('ls /data', stdin=PIPE, stdout=PIPE, stderr=PIPE, shell=True)    # stdin用于接收标准输入，可选参数，可以通过 p.stdin.write('xxx') 添加标准输入

                                                                                   # stdout用于接收标准输出，可选参数，可以通过 p.stdout.read() 读取标准输出

In [3]: p.stdout.read()                                                            # stderr用于接收标准错误输出，可选参数，可以通过 p.stderr.read() 读取标准错误输出

Out[3]: '1.txt\n2.txt\n3.txt\n'                                                    # shell=True用于表明该命令是在shell下执行的，如果不加上该参数会报错

                                                                                   # 如果我们没有用PIPE来接收标准输出和标准错误输出，则默认会打印到终端上

In [4]: p.stderr.read()                                                            # 另外要注意的是，当我们写成 p = Popen(...) 时，命令就已经执行了，执行后的结果放在管道(PIPE)里面

Out[4]: ''                                                                         # subprocess.Popen()其他常用方法

线上的用法：

import subprocess

def runCMD(self, ip, cmd, logfile):

    out = ""

    sshcmd = ''' ssh root@%s -n '%s' ''' % (ip, cmd)

    p = subprocess.Popen(sshcmd, shell=True, stdout=subprocess.PIPE, stderr=subprocess.STDOUT)

    while True:

        buff = p.stdout.readline()

        if p.poll() != None and buff == "":

            break

        if buff:

            out += buff

            self.writeLogFile(logfile, buff)

    return p.returncode, out

相关文章