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)
失敗