subprocess的使用

python写流程控制的时候，有时需要等待运行结果完成才能进行下一步。有几个想法：

• 一是反复循环判断直到生成结果文件（且文件中有预期结果），低效易出错；
• 二是用subprocess模块来判断进程结果，较为常用；
• 三是用drmaa包，功能强大，但使用相对复杂。
  这次介绍如何用subprocess来实现。subprocess中的poll函数有以下几种返回码：

0 正常结束
1 sleep
2 子进程不存在
5 kill
None 在运行

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, 
startupinfo=None, creationflags=0,restore_signals=True, start_new_session=False, pass_fds=(),
*, encoding=None, errors=None)
 

常用参数

• 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
 
proc = subprocess.Popen(['python', 'test.py'], stdout=subprocess.PIPE)
while True:
  flag = 1
  if proc.poll() is None:
    flag = 0
  if flag:
    break
  else:
    print("Task is running")
    time.sleep(100)

Popen 对象方法

• poll(): 检查进程是否终止，如果终止返回 returncode，否则返回 None。
• wait(timeout): 等待子进程终止。
• communicate(input,timeout): 和子进程交互，发送和读取数据。
• send_signal(singnal): 发送信号到子进程 。
• terminate(): 停止子进程,也就是发送SIGTERM信号到子进程。
• kill(): 杀死子进程。发送 SIGKILL 信号到子进程。

实例

import time
import subprocess
 
def cmd(command):
    subp = subprocess.Popen(command,shell=True,stdout=subprocess.PIPE,stderr=subprocess.PIPE,encoding="utf-8")
    subp.wait(2)
    if subp.poll() == 0:
        print(subp.communicate()[1])
    else:
        print("失败")
 
 
 
cmd("java -version")
cmd("exit 1")

输出结果

java version "1.8.0_31"
Java(TM) SE Runtime Environment (build 1.8.0_31-b13)
Java HotSpot(TM) 64-Bit Server VM (build 25.31-b07, mixed mode)
失败