Python中的Subprocess模塊

之前我一直用os.system()處理一些系統管理任務,由於我認爲那是運行linux命令最簡單的方式.
咱們能從Python官方文檔裏讀到應該用subprocess 模塊來運行系統命令.subprocess模塊容許咱們建立子進程,鏈接他們的輸入/輸出/錯誤管道，還有得到返回值。
subprocess模塊打算來替代幾個過期的模塊和函數，好比： os.system, os.spawn*, os.popen*, popen2.*命令。
讓咱們來看一下subprocess 有哪些不一樣的函數.

subprocess.call()

執行由參數提供的命令.
咱們能夠用數組做爲參數運行命令,也能夠用字符串做爲參數運行命令(經過設置參數shell=True)
注意,參數shell默認爲False
咱們用subprocess.call()來作一個統計磁盤的例子:

subprocess.call(['df', '-h'])

下面的例子把shell設置爲True

subprocess.call('du -hs $HOME', shell=True)

注意,python官方文檔裏對參數shell=True陳述了一個警告:

Invoking the system shell with shell=True can be a security hazard if combined
with untrusted input

如今,咱們來看看輸入與輸出

Input and Output

subprocess 模塊能阻止輸出,當你不關心標準輸出的時候是很是方便的.
它也使你經過一種正確的方式管理輸入/輸出,有條理地整合python腳本中的的shell命令.

Return Codes

經過subprocess.call的返回值你可以斷定命令是否執行成功.
每個進程退出時都會返回一個狀態碼，你能夠根據這個狀態碼寫一些代碼。

stdin, stdout and stderr

我在使用subprocess 時,有一個微妙的部分是怎麼使用管道把命令鏈接起來.
管道代表一個新的子管道應該被建立.
默認的設置爲None,意味着沒有重定向發生
標準錯誤能夠指向標準輸出,代表子進程的錯誤信息會被捕獲到和標準輸出同一個文件.

subprocess.Popen()

subprocess 模塊中基本的進程建立和管理由Popen 類來處理.
subprocess.popen是用來替代os.popen的.
咱們來作一些真實的例子,subprocess.Popen須要一個數組做爲參數:

import subprocess

p = subprocess.Popen(["echo", "hello world"], stdout=subprocess.PIPE)

print p.communicate()

>>>('hello world
', None)

注意,雖然你可使用 "shell=True",但並不推薦這樣的方式.
若是你知道你只用幾個有限的函數,好比Popen和PIPE,你能夠單單指定這幾個函數:

from subprocess import Popen, PIPE

p1 = Popen(["dmesg"], stdout=PIPE)

print p1.communicate()

Popen.communicate()

communicate()函數返回一個tuple(標準輸出和錯誤).
Popen.communicate() 和進程溝通:發送數據到標準輸入.從標準輸出和錯誤讀取數據直到遇到結束符.等待進程結束.
輸入參數應該是一個字符串,以傳遞給子進程,若是沒有數據的話應該是None.
基本上,當你用 communicate()函數的時候意味着你要執行命令了.

用subprocess寫Ping程序

咱們先問用戶地址,而後用ping請求這個地址.

# Import the module
import subprocess

# Ask the user for input
host = raw_input("Enter a host to ping: ")    

# Set up the echo command and direct the output to a pipe
p1 = subprocess.Popen(['ping', '-c 2', host], stdout=subprocess.PIPE)

# Run the command
output = p1.communicate()[0]

print output

譯自