Python武器库开发-常用模块之subprocess模块(十九)

常用模块之subprocess模块(十九)

subprocess模块介绍

subprocess 模块允许我们启动一个新进程，并连接到它们的输入/输出/错误管道，从而获取返回值。subprocess 它可以用来调用第三方工具（例如：exe、另一个python文件、命令行工具）。subprocess 模块用于启动新的模块，它可以用于执行新的命令，获取进程的输出、输入和等待进程的结束。

在python中，多进程协同大概有三种方式：

1. os.system函数
2. mulitprocessing模块
3. subprocess模块

我们可以在python中使用以下的语句，引入subprocess模块：

import subprocess

subprocess模块中的常用函数

在subprocess模块中的常用函数如下表所示：

subprocess模块常用参数说明

subprocess.run()函数

先来一个正确执行命令

语法：subprocess.run(args, *, stdin=None, input=None, stdout=None, stderr=None, shell=False, timeout=None, check=False, universal_newlines=False)

实例：

import subprocess
subprocess.run(["ls", "-l"])

以上实例的输出结果：

subprocess.call()函数

语法：subprocess.call(args, *, stdin=None, stdout=None, stderr=None, shell=False, timeout=None)

实例：

import subprocess
subprocess.call(['ls',  '-l'])
subprocess.call('ls -l', shell=True)
subprocess.call(['ls',  '-l'], stdout=subprocess.DEVNULL)
subprocess.call(['ls',  '-l', '/test'])

以上实例的输出结果：

subprocess.check_call()函数

语法：subprocess.check_call(args, *, stdin=None, stdout=None, stderr=None, shell=False, timeout=None)

实例：

import subprocess
subprocess.check_call(['ls',  '-l'])
subprocess.check_call('ls -l', shell=True)
subprocess.check_call('ls -l /test', shell=True)

以上实例的输出结果：

subprocess.check_output()函数

语法：subprocess.check_output(args, *, stdin=None, stderr=None, shell=False, universal_newlines=False, timeout=None)

实例：

import subprocess
ret = subprocess.check_output(['ls',  '-l'])
print(ret)
ret = subprocess.check_output(['ls',  '-l'], universal_newlines=True)
print(ret)

以上实例的输出结果：

subprocess.getstatusoutput()函数

语法：subprocess.getstatusoutput(cmd)

实例：

import subprocess
retcode, output = subprocess.getstatusoutput('ls -l')
print(retcode)
print(output)
retcode, output = subprocess.getstatusoutput('ls -l /test')
print(retcode)
print(output)

以上实例的输出结果：

subprocess.getoutput()函数

语法：subprocess.getoutput(cmd)

实例：

import subprocess
ret = subprocess.getoutput('ls -l')
print(ret)

以上实例的输出结果：

subprocess.Popen介绍

该类用于在一个新的进程中执行一个子程序。前面我们提到过，上面介绍的这些函数都是基于subprocess.Popen类实现的，通过使用这些被封装后的高级函数可以很方面的完成一些常见的需求。由于subprocess模块底层的进程创建和管理是由Popen类来处理的，因此，当我们无法通过上面哪些高级函数来实现一些不太常见的功能时就可以通过subprocess.Popen类提供的灵活的api来完成。

subprocess.Popen常用参数说明

subprocess.Popen的语法：class subprocess.Popen(args, bufsize=-1, executable=None, stdin=None, stdout=None, stderr=None, preexec_fn=None, close_fds=True, shell=False, cwd=None, env=None, universal_newlines=False,startup_info=None, creationflags=0, restore_signals=True, start_new_session=False, pass_fds=())

subprocess.Popen常用参数如下表所示：

subprocess.Popen类的实例可调用的方法

subprocess.Popen类的实例可调用的方法如下表所示：

关于communicate()方法的说明：

• 该方法中的可选参数 input应该是将被发送给子进程的数据，或者如没有数据发送给子进程，该参数应该是None。input参数的数据类型必须是字节串，如果universal_newlines参数值为True，则input参数的数据类型必须是字符串。
• 该方法返回一个元组(stdout_data,stderr_data)，这些数据将会是字节穿或字符串（如果universal_newlines的值为True）。
• 如果在timeout指定的秒数后该进程还没有结束，将会抛出一个TimeoutExpired异常。捕获这个异常，然后重新尝试通信不会丢失任何输出的数据。但是超时之后子进程并没有被杀死，为了合理的清除相应的内容，一个好的应用应该手动杀死这个子进程来结束通信。
• 需要注意的是，这里读取的数据是缓冲在内存中的，所以，如果数据大小非常大或者是无限的，就不应该使用这个方法

实列一控制电脑逐步执行命令：

import subprocessobj = subprocess.Popen(["python"], stdin=subprocess.PIPE,stdout=subprocess.PIPE,stderr=subprocess.PIPE,universal_newlines=True)
obj.stdin.write("print(1)\n")
obj.stdin.write('print(2) \n')
out,err = obj.communicate()
print(out)

以上代码输出的实例：

实例二获得输出的数据：

import subprocess
p = subprocess.Popen('ls -al', stdout=subprocess.PIPE, shell=True)
print(p.stdout.read())

以上代码输出的实例：

实例三实现shell中管道的功能

import subprocessp1 = subprocess.Popen(['df', '-Th'], stdout=subprocess.PIPE)
p2 = subprocess.Popen(['grep', 'data'], stdin=p1.stdout, stdout=subprocess.PIPE)
out,err = p2.communicate()
print(out)

以上代码输出的实例：