Python中的Subprocess模塊

marek 發(fā)布于2019-07-25 11:46 / 1242人閱讀

摘要：以前我一直用處理一些系統(tǒng)管理任務(wù)因?yàn)槲艺J(rèn)為那是運(yùn)行命令最簡(jiǎn)單的方式我們能從官方文檔里讀到應(yīng)該用模塊來(lái)運(yùn)行系統(tǒng)命令模塊允許我們創(chuàng)建子進(jìn)程連接他們的輸入輸出錯(cuò)誤管道，還有獲得返回值。模塊打算來(lái)替代幾個(gè)過(guò)時(shí)的模塊和函數(shù)，比如命令。

以前我一直用os.system()處理一些系統(tǒng)管理任務(wù),因?yàn)槲艺J(rèn)為那是運(yùn)行l(wèi)inux命令最簡(jiǎn)單的方式.
我們能從Python官方文檔里讀到應(yīng)該用subprocess 模塊來(lái)運(yùn)行系統(tǒng)命令.subprocess模塊允許我們創(chuàng)建子進(jìn)程,連接他們的輸入/輸出/錯(cuò)誤管道，還有獲得返回值。
subprocess模塊打算來(lái)替代幾個(gè)過(guò)時(shí)的模塊和函數(shù)，比如： os.system, os.spawn*, os.popen*, popen2.*命令。
讓我們來(lái)看一下subprocess 有哪些不同的函數(shù).

subprocess.call()

執(zhí)行由參數(shù)提供的命令.
我們可以用數(shù)組作為參數(shù)運(yùn)行命令,也可以用字符串作為參數(shù)運(yùn)行命令(通過(guò)設(shè)置參數(shù)shell=True)
注意,參數(shù)shell默認(rèn)為False
我們用subprocess.call()來(lái)做一個(gè)統(tǒng)計(jì)磁盤(pán)的例子:

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

下面的例子把shell設(shè)置為True

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

注意,python官方文檔里對(duì)參數(shù)shell=True陳述了一個(gè)警告:

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

現(xiàn)在,我們來(lái)看看輸入與輸出

Input and Output

subprocess 模塊能阻止輸出,當(dāng)你不關(guān)心標(biāo)準(zhǔn)輸出的時(shí)候是非常方便的.
它也使你通過(guò)一種正確的方式管理輸入/輸出,有條理地整合python腳本中的的shell命令.

Return Codes

通過(guò)subprocess.call的返回值你能夠判定命令是否執(zhí)行成功.
每一個(gè)進(jìn)程退出時(shí)都會(huì)返回一個(gè)狀態(tài)碼，你可以根據(jù)這個(gè)狀態(tài)碼寫(xiě)一些代碼。

stdin, stdout and stderr

我在使用subprocess 時(shí),有一個(gè)微妙的部分是怎么使用管道把命令連接起來(lái).
管道表明一個(gè)新的子管道應(yīng)該被創(chuàng)建.
默認(rèn)的設(shè)置為None,意味著沒(méi)有重定向發(fā)生
標(biāo)準(zhǔn)錯(cuò)誤可以指向標(biāo)準(zhǔn)輸出,表明子進(jìn)程的錯(cuò)誤信息會(huì)被捕獲到和標(biāo)準(zhǔn)輸出同一個(gè)文件.

subprocess.Popen()

subprocess 模塊中基本的進(jìn)程創(chuàng)建和管理由Popen 類來(lái)處理.
subprocess.popen是用來(lái)替代os.popen的.
我們來(lái)做一些真實(shí)的例子,subprocess.Popen需要一個(gè)數(shù)組作為參數(shù):

import subprocess

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

print p.communicate()

>>>("hello world
", None)

注意,雖然你可以使用 "shell=True",但并不推薦這樣的方式.
如果你知道你只用幾個(gè)有限的函數(shù),比如Popen和PIPE,你可以單單指定這幾個(gè)函數(shù):

from subprocess import Popen, PIPE

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

print p1.communicate()

Popen.communicate()

communicate()函數(shù)返回一個(gè)tuple(標(biāo)準(zhǔn)輸出和錯(cuò)誤).
Popen.communicate() 和進(jìn)程溝通:發(fā)送數(shù)據(jù)到標(biāo)準(zhǔn)輸入.從標(biāo)準(zhǔn)輸出和錯(cuò)誤讀取數(shù)據(jù)直到遇到結(jié)束符.等待進(jìn)程結(jié)束.
輸入?yún)?shù)應(yīng)該是一個(gè)字符串,以傳遞給子進(jìn)程,如果沒(méi)有數(shù)據(jù)的話應(yīng)該是None.
基本上,當(dāng)你用 communicate()函數(shù)的時(shí)候意味著你要執(zhí)行命令了.

用subprocess寫(xiě)Ping程序

我們先問(wèn)用戶地址,然后用ping請(qǐng)求這個(gè)地址.

# 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

譯自