subprocess模块

一、什么是subprocess模块
二、subprocess语法
三、官方介绍
四、基本操作方法
1. subprocess的run、call、check_call、check_output函数
2. subprocess的getoutput、getstatusoutput函数
五、 subprocess.Popen类
1. 介绍
2.操作
（1）subprocess.Popen对象常用方法（如PopenObject为subprocess.Popen对象）
（2）subprocess.Popen对象的文本或字节流控制

TOC

一、什么是subprocess模块

subprocess:
sub: 子
process: 进程

可以通过python代码给操作系统发送命令，并将执行结果返回。

二、subprocess语法

import subprocess
while True:
    # 1.让用户输入终端命令
    cmd_str = input('请输入终端命令:').strip()
    # Popen(cmd命令, shell=True,
    # stdout=subprocess.PIPE, stderr=subprocess.PIPE)
    # 调用Popen就会将用户的终端命令发送给本地操作系统的终端
    # 得到一个对象，对象中包含着正确或错误的结果。
    obj = subprocess.Popen(
        cmd_str, shell=True,
        stdout=subprocess.PIPE,
        stderr=subprocess.PIPE
    )

    success = obj.stdout.read().decode('gbk')
    if success:
        print(success, '正确的结果')

    error = obj.stderr.read().decode('gbk')
    if error:
        print(error, '错误的结果')

---

三、官方介绍

subprocess模块可以生成新的进程，连接到它们的input/output/error管道，同时获取它们的返回码。

四、基本操作方法

1. subprocess的run、call、check_call、check_output函数

• subprocess.run(args[, stdout, stderr, shell ...])：执行args命令，返回值为CompletedProcess类；
  若未指定stdout，则命令执行后的结果输出到屏幕上，函数返回值CompletedProcess中包含有args和returncode；
  若指定有stdout，则命令执行后的结果输出到stdout中，函数返回值CompletedProcess中包含有args、returncode和stdout；
  若执行成功，则returncode为0；若执行失败，则returncode为1；
  若想获取args命令执行后的输出结果，命令为：output = subprocess.run(args, stdout=subprocess.PIPE).stdout
• subprocess.call(args[, stdout, ...])：执行args命令，返回值为命令执行状态码；
  若未指定stdout，则命令执行后的结果输出到屏幕；
  若指定stdout，则命令执行后的结果输出到stdout；
  若执行成功，则函数返回值为0；若执行失败，则函数返回值为1；
  （类似os.system）
• subprocess.check_call(args[, stdout, ...])：执行args命令，返回值为命令执行状态码；
  若未指定stdout，则命令执行后的结果输出到屏幕；
  若指定stdout，则命令执行后的结果输出到stdout；
  若执行成功，则函数返回值为0；若执行失败，抛出异常；
  （类似subprocess.run(args, check=True)）
• subprocess.check_output(args[, stderr, ...])：执行args命令，返回值为命令执行的输出结果；
  若执行成功，则函数返回值为命令输出结果；若执行失败，则抛出异常；
  （类似subprocess.run(args, check=True, stdout=subprocess.PIPE).stdout）

（1） args：启动进程的参数，默认为字符串序列（列表或元组），也可为字符串（设为字符串时一般需将shell参数赋值为True）；
（2） shell：shell为True，表示args命令通过shell执行，则可访问shell的特性；
（3） check：check为True时，表示执行命令的进程以非0状态码退出时会抛出；subprocess.CalledProcessError异常；check为False时，状态码为非0退出时不会抛出异常；
（4） stdout、stdin、stderr：分别表示程序标准标输出、输入、错误信息；
run函数返回值为CompletedProcess类，若需获取执行结果，可通过获取返回值的stdout和stderr来捕获；
check_output函数若需捕获错误信息，可通过stderr=subprocess.STDOUT来获取；

# subprocess.run使用
def subprocess_run():
    print("**** subprocess.run ****")
    print("----------")
    result1 = subprocess.run(["adb", "devices"])
    print("result1:", result1)
    print("----------")
    result2 = subprocess.run("adb devices", shell=True, check=True)
    print("result2:", result2)
    print("----------")
    result3 = subprocess.run(["adb", "devices"], stdout=subprocess.PIPE)
    print("result3:", result3)
    print(type(result3))
subprocess_run()
"""结果
**** subprocess.run ****
----------
List of devices attached
338b123f0504 device

result1: CompletedProcess(args=['adb', 'devices'], returncode=0)
----------
List of devices attached
338b123f0504 device

result2: CompletedProcess(args='adb devices', returncode=0)
----------
result3: CompletedProcess(args=['adb', 'devices'], returncode=0, stdout=b'List of devices attached 
338b123f0504	device

')
<class 'subprocess.CompletedProcess'>
"""

# subprocess.call使用
def subprocess_call():
    print("**** subprocess.call ****")
    print("----------")
    result1 = subprocess.call(["adb", "devices"])
    print("result1:", result1)
    print("----------")
    result2 = subprocess.call(["adb", "devices"], stdout=subprocess.PIPE)
    print("result2:", result2)
subprocess_call()
"""结果
**** subprocess.call ****
----------
List of devices attached
338b123f0504 device

result1: 0
----------
result2: 0
"""

# subprocess.check_call
def subprocess_check_call():
    print("**** subprocess.check_call ****")
    print("----------")
    result1 = subprocess.check_call(["adb", "devices"])
    print("result1:", result1)
    print("----------")
    result2 = subprocess.check_call(["adb", "devices"], stdout=subprocess.PIPE)
    print("result2:", result2)
subprocess_check_call()
"""结果
**** subprocess.check_call ****
----------
List of devices attached
338b123f0504 device

result1: 0
----------
result2: 0
"""

# subprocess.check_output
def subprocess_check_output():
    print("**** subprocess.check_output ****")
    print("----------")
    result1 = subprocess.check_output(["adb", "devices"])
    print("result1:", result1)
    print("----------")
    result2 = subprocess.run(["adb", "devices"], stdout=subprocess.PIPE).stdout
    print("result2:", result2)
subprocess_check_output()
"""结果
**** subprocess.check_output ****
----------
result1: b'List of devices attached 
338b123f0504	device

'
----------
result2: b'List of devices attached 
338b123f0504	device

'
"""

2. subprocess的getoutput、getstatusoutput函数

• subprocess.getoutput(cmd)：执行cmd命令，返回值为命令执行的输出结果（字符串类型）；
  注：执行失败，不会抛出异常（类似os.popen(cmd).read()）；
• subprocess.getstatusoutput(cmd)：执行cmd命令，返回值为元组类型(命令执行状态, 命令执行的输出结果)；
  元组中命令执行状态为0，表示执行成功；命令执行状态为1，表示执行失败；

cmd：参数，字符串类型；

# subprocess.getoutput或getstatusoutput使用
def subprocess_get_output():
    print("**** subprocess.getoutput ****")
    result1 = subprocess.getoutput("adb devices")
    print("result1:", result1)
    print(type(result1))

    print("**** subprocess.getstatusoutput ****")
    result2 = subprocess.getstatusoutput("adb devices")
    print("result2:", result2)
    print(type(result2))
subprocess_get_output()
"""结果
**** subprocess.getoutput ****
result1: List of devices attached
338b123f0504 device

<class 'str'>
**** subprocess.getstatusoutput ****
result2: (0, 'List of devices attached 
338b123f0504	device
')
<class 'tuple'>
"""

五、 subprocess.Popen类

1. 介绍

subprocess.Popen类用于在一个新进程中执行一个子程序，上述subprocess函数均是基于subprocess.Popen类；

2.操作

• subprocess.Popen(args[, bufsize, stdin, stdout, stderr, ...])：Popen类的构造函数，返回结果为subprocess.Popen对象；

• args：需要执行的系统命令，可为字符串序列（列表或元组，shell为默认值False即可，建议为序列），也可为字符串（使用字符串时，需将shell赋值为True）；
• shell：默认为False，若args为序列时，shell=False；若args为字符串时，shell=True，表示通过shell执行命令；
• stdout、stdin、stderr：分别表示子程序标准输出、标准输入、标准错误，可为subprocess.PIPE、一个有效的文件描述符、文件对象或None。
  若为subprocess.PIPE：代表打开通向标准流的管道，创建一个新的管道；
  若为None：表示没有任何重定向，子进程会继承父进程；
  stderr也可为subprocess.STDOUT：表示将子程序的标准错误输出重定向到了标准输出
• bufsize：指定缓冲策略，0表示不缓冲，1表示行缓冲，其它整数表示缓冲区大小，负数表示使用系统默认值0；
• cwd：默认值为None；若非None，则表示将会在执行这个子进程之前改变当前工作目录；
• env：用于指定子进程的环境变量。若env为None，那么子进程的环境变量将从父进程中继承；若env非None，则表示子程序的环境变量由env值来设置，它的值必须是一个映射对象。
• universal_newlines： 不同系统的换行符不同。若True，则该文件对象的stdin，stdout和stderr将会以文本流方式打开；否则以二进制流方式打开。

（1）subprocess.Popen对象常用方法（如PopenObject为subprocess.Popen对象）

• PopenObject.poll() ：用于检查命令是否已经执行结束，若结束返回状态码；若未结束返回None；
• PopenObject.wait([timeout, endtime])：等待子进程结束，并返回状态码；若超过timeout(s)进程仍未结束，则抛出异常；
• PopenObject.send_signal(signal)：发送信号signal给子进程；
• PopenObject.terminate()：停止子进程；
• PopenObject.kill()：杀死子进程；
• PopenObject.communicate([input, timeout])：与进程进行交互（如发送数据到stdin、读取stdout和stderr数据），它会阻塞父进程，直到子进程完成；
  input：表示将发送到子进程的字符串数据，默认为None；
  timeout：超时判断，若超过timeout秒后仍未结束则抛出TimeoutExpired异常；
  communicate返回值：一个元组(stdout_data, stderr_data)

（2）subprocess.Popen对象的文本或字节流控制

• PopenObject.stdin：
  若PopenObject中stdin为PIPE，则返回一个可写流对象；若encoding或errors参数被指定或universal_newlines参数为True，则此流是一个文件流，否则为字节流。
  若PopenObject中stdin不是PIPE，则属性为None。
  stdin输入流非None，可执行写操作即PopenObject.stdin.write(s)
• PopenObject.stdout：
  若PopenObject中stdout为PIPE，则返回一个可读流对象；若encoding或errors参数被指定或universal_newlines参数为True，则此流是一个文件流，否则为字节流。
  若PopenObject中stdout不是PIPE，则属性为None。
  stdout输出流非None，可执行读操作即PopenObject.stdout.read()或.readlines()
• PopenObject.stderr：
  若PopenObject中stderr为PIPE，则返回一个可读流对象；若encoding或errors参数被指定或universal_newlines参数为True，则此流是一个文件流，否则为字节流。
  若PopenObject中stderr不是PIPE，则属性为None。
  stderr错误流非None，可执行读操作即PopenObject.stderr.read()或.readlines()

def subprocess_Popen1():
    print("***通过communicate函数分别输出PopenObject对象的输出流和错误流***")
    args = [["adb", "devices"], ["adb", "devices11"]]
    for arg in args:
        popen_object = subprocess.Popen(arg, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
        object_stdout, object_stderr = popen_object.communicate()
        output = {"popen_object": popen_object,
                  "object_stdout": object_stdout,
                  "object_stderr": object_stderr}
        print(output)
    """
    {'popen_object': <subprocess.Popen object at 0x0000000002212400>, 'object_stdout': b'List of devices attached 
106D111805005938	device

', 'object_stderr': b''}
    {'popen_object': <subprocess.Popen object at 0x0000000002577C18>, 'object_stdout': b'', 'object_stderr': b'Android Debug Bridge version 1.0.31

 -a .....}
    """

    print("***通过stdout和stderr方法输出PopenObject对象输出流和错误流***")
    p0 = subprocess.Popen(["adb", "devices"], stdout=subprocess.PIPE, stderr=subprocess.PIPE)
    object_stdout = p0.stdout.read()
    p0.stdout.close()
    object_stderr = p0.stderr.read()
    p0.stderr.close()
    print(object_stdout) # 结果：b'List of devices attached 
338b123f0504	device

'
    print(object_stderr) # 结果：b''

    print("***Popen对象stdin写入功能：使用stdout和stderr输出")
    args = ["python", "python1"]
    for arg in args:
        p4 = subprocess.Popen([arg], shell=True, stdout=subprocess.PIPE,
                              stdin=subprocess.PIPE, stderr=subprocess.PIPE, universal_newlines=True)
        p4.stdin.write("print('hello')")
        p4.stdin.close()
        out = p4.stdout.read()
        p4.stdout.close()
        err = p4.stderr.read()
        p4.stderr.close()
        print("out：%s err：%s" % (out, err))
    """
    ***Popen对象stdin写入功能
    out：hello
    err：
    out： err：'python1' 不是内部或外部命令，也不是可运行的程序或批处理文件。
    """

    print("***Popen对象stdin写入功能：使用communicate输出")
    p4 = subprocess.Popen(["python"], stdout=subprocess.PIPE,
                          stdin=subprocess.PIPE, stderr=subprocess.PIPE, universal_newlines=True)
    p4.stdin.write("print('hello')")
    output = p4.communicate()
    print(output) # 结果：('hello
', '')

    print("***不含encoding参数***")
    p1 = subprocess.Popen("adb devices", shell=True, stdout=subprocess.PIPE)
    out1 = p1.stdout.readlines()
    print(out1) # 结果: [b'List of devices attached 
', b'106D111805005938	device
', b'
']

    print("***含encoding参数***")
    p2 = subprocess.Popen("adb devices", shell=True, stdout=subprocess.PIPE, encoding="utf-8")
    out2 = p2.stdout.readlines()
    print(out2) # 结果: ['List of devices attached 
', '106D111805005938	device
', '
']

    print("***Popen对象检查命令是否结束，等待进程结束")
    print(p2.poll()) # 结果: None
    print(p2.wait()) # 结果: 0
    print(p2.poll()) # 结果: 0

    print("***Popen对象communicate函数，它会阻塞父进程直至子进程完成")
    p3 = subprocess.Popen("adb devices", shell=True, stdout=subprocess.PIPE)
    out = p3.communicate()[0]
    print(out) # 结果：b'List of devices attached 
338b123f0504	device

'
    print(p3.poll()) # 结果：0
subprocess_Popen1()



def subprocess_Popen2():
    """
    1. 通过管道功能，实现adb shell ps | findstr top功能
    2. 直接为args赋值为一个字符串，实现adb shell ps | findstr top功能
    :return:
    """
    print("***通过管道方式***")
    p1 = subprocess.Popen(["adb", "shell", "ps"], stdout=subprocess.PIPE)
    p2 = subprocess.Popen(["findstr", "top"], stdin=p1.stdout, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
    out, err = p2.communicate()
    print(out, err) # 结果：b'shell 8508 8504 2600 1044 c004e5f8 b6f40938 S top

' b''
    print("***通过传一个字符串方式***")
    p3 = subprocess.Popen("adb shell ps | findstr top", shell=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
    out, err = p3.communicate()
    print(out, err) # 结果：b'shell 8508 8504 2600 1044 c004e5f8 b6f40938 S top

' b''
subprocess_Popen2()

来自为知笔记(Wiz)