前言

subprocess庫(kù)提供了一個(gè)API創(chuàng)建子進(jìn)程并與之通信。這對(duì)于運(yùn)行生產(chǎn)或消費(fèi)文本的程序尤其有好處，因?yàn)檫@個(gè)API支持通過(guò)新進(jìn)行的標(biāo)準(zhǔn)輸入和輸出通道來(lái)回傳數(shù)據(jù)。

本篇，將詳細(xì)介紹Python創(chuàng)建附加進(jìn)行的庫(kù)：subprocess。

run(運(yùn)行外部命令)

subprocess庫(kù)本身可以替換os.system()，os.spawnv()等函數(shù)?，F(xiàn)在我們來(lái)通過(guò)subprocess庫(kù)運(yùn)行一個(gè)外部命令，但不采用os.system()。示例如下：

import subprocess

completed = subprocess.run('whoami')
print(completed.returncode)

這里我們運(yùn)行了一個(gè)windows系統(tǒng)常用的whoami命令，返回當(dāng)前用戶的名稱，輸出如下：

這里，我們使用了subprocess.run調(diào)用了子進(jìn)程運(yùn)行windows命令。它返回一個(gè)CompletedProcess實(shí)例，它包含了與進(jìn)行有關(guān)的信息。returncode為子進(jìn)程的退出狀態(tài)碼。通常情況下，退出狀態(tài)碼為0則表示進(jìn)程成功運(yùn)行了；一個(gè)負(fù)值-N表示這個(gè)子進(jìn)程被信號(hào)N終止了。

該函數(shù)還有許多參數(shù)，比如shell，默認(rèn)值為False表示直接運(yùn)行命令，如果主動(dòng)賦值為T(mén)rue則會(huì)創(chuàng)建一個(gè)中間shell進(jìn)程，由這個(gè)進(jìn)程運(yùn)行命令。

import subprocess

completed = subprocess.run('echo 123',shell=True)
print(completed.returncode)

比如這里，我們打印123。

該庫(kù)還有一個(gè)call()函數(shù)，subprocess.run有一個(gè)check參數(shù)，如果沒(méi)有設(shè)置該參數(shù)，等價(jià)于調(diào)用了call()函數(shù)。check默認(rèn)值為False。

對(duì)于run()函數(shù)啟動(dòng)的進(jìn)程，它的標(biāo)準(zhǔn)輸入輸出通道會(huì)綁定到父進(jìn)程的輸入輸出。這說(shuō)明調(diào)用程序無(wú)法捕獲命令的輸出。不過(guò)，我們可以通過(guò)為stdout和stderr參數(shù)傳入PIPE來(lái)捕獲輸出，以備以后處理。

import subprocess

completed = subprocess.run('whoami',stdout=subprocess.PIPE)
print(completed.returncode)
print(len(completed.stdout))
print(completed.stdout.decode('UTF-8'))

運(yùn)行之后，效果如下：

如果設(shè)置run()函數(shù)的參數(shù)check=True與stdout為PIPE，等價(jià)于調(diào)用了check_output()函數(shù)。

通過(guò)Shell返回消息

本例會(huì)通過(guò)一個(gè)子shell運(yùn)行命令，在命令返回錯(cuò)誤碼并退出之前，將詳細(xì)輸入到控制臺(tái)。實(shí)例如下：

import subprocess

try:
    completed = subprocess.run(
        'echoa 123',
        shell=True,
        stdout=subprocess.PIPE,
        stderr=subprocess.PIPE, )
except subprocess.CalledProcessError as err:
    print("ERROR:", err)
else:
    print("else")
    print(completed.returncode)
    print(len(completed.stdout))
    print(completed.stdout.decode('UTF-8'))
    print(len(completed.stderr))
    print(completed.stderr.decode('gbk'))

運(yùn)行之后，效果如下：

這里我們輸入了一個(gè)錯(cuò)誤的命令，可以看到因?yàn)槊铄e(cuò)誤，并沒(méi)有輸出命令的執(zhí)行結(jié)果，0和64中間就是completed.stdout，為空。而命令將錯(cuò)誤消息返回了。這是因?yàn)槲覀冊(cè)O(shè)置了stdout與stderr為subprocess.PIPE，表明這些通道要開(kāi)放。這樣我們才能獲取子shell運(yùn)行的結(jié)果獲取所運(yùn)行的錯(cuò)誤提示。（讀者可以將命令改正確后可以發(fā)現(xiàn)錯(cuò)誤消息沒(méi)有了，正確執(zhí)行結(jié)果會(huì)輸出。這就是subprocess庫(kù)創(chuàng)建進(jìn)程的通信機(jī)制）

需要注意的是，如果需要抑制輸出效果，可以將stdout與stderr設(shè)置為subprocess.DEVNULL。不過(guò)改了之后，上面代碼肯定會(huì)報(bào)錯(cuò)，因?yàn)楣艿狸P(guān)閉，通信也就關(guān)閉了。也就是沒(méi)有這些參數(shù)了。

直接處理管道

subprocess庫(kù)還有一個(gè)非常重要的類Popen，它是用來(lái)建立其他API的底層API，對(duì)更復(fù)雜的進(jìn)程交互很有用。

比如run()，call()，check_call()和check_output()函數(shù)都是Popen類的包裝器。直接使用Popen可以更好的控制如何運(yùn)行命令以及如何處理輸入和輸出流。Popen的構(gòu)造函數(shù)利用參數(shù)建立新進(jìn)程，使父進(jìn)程可以通過(guò)管道與之通信。

下面，我們來(lái)分別介紹進(jìn)程間通信的方式。

與進(jìn)程的單項(xiàng)通信

要運(yùn)行一個(gè)進(jìn)程并讀取它的所有輸出，可以設(shè)置stdout為PIPE并調(diào)用communicate()函數(shù)。示例如下：

import subprocess

prc = subprocess.Popen('whoami', stdout=subprocess.PIPE)
stdout_value = prc.communicate()[0].decode('utf-8')
print(repr(stdout_value))

如上面代碼所示，Popen會(huì)在內(nèi)部管理數(shù)據(jù)讀取。運(yùn)行之后，效果如下：

如果你需要調(diào)用一個(gè)管道，并完成寫(xiě)數(shù)據(jù)的操作，可以設(shè)置stdin為PIPE。

import subprocess

prc = subprocess.Popen(["cmd", "/c", 'type', '-'], stdin=subprocess.PIPE)
prc.communicate('stdin'.encode('UTF-8'))

與進(jìn)程的雙向通信

要完成進(jìn)程的雙向通信，可以直接將stdin與stdout都設(shè)置為PIPE即可。示例如下：

import subprocess

cmd = "cmd /c type E:/Project/debug.log"
cmd.encode('utf-8')
prc = subprocess.Popen(cmd, stdin=subprocess.PIPE, stdout=subprocess.PIPE)
msg = 'stdin'.encode('UTF-8')
stdout_value = prc.communicate(msg)[0].decode('utf-8')
print(repr(stdout_value))

至于如果命令行錯(cuò)誤需要捕獲錯(cuò)誤消息，可以直接將stderr也設(shè)置為PIPE。

連接管道段

在Linux系統(tǒng)中，我們可以將多個(gè)命令連接成一個(gè)管線，即可以把它們的輸入輸出串聯(lián)在一起。通過(guò)Popen我們也可以完成類似的操作，只需要將一個(gè)Popen實(shí)例的stdout屬性被用左管線中下一個(gè)Popen實(shí)例的stdin參數(shù)即可。至于最后肯定還是要設(shè)置為PIPE，畢竟我們還是要獲取多個(gè)管道段消息結(jié)果，示例如下：

import subprocess

cmd1 = "cmd /c type E:/Project/debug.log"
proc1 = subprocess.Popen(cmd1, stdout=subprocess.PIPE, encoding='utf-8')

cmd2 = "tree /F | findstr 拒絕訪問(wèn)"
proc2 = subprocess.Popen(cmd1, stdout=subprocess.PIPE, stdin=proc1.stdout, encoding='utf-8')

result = proc2.stdout

for line in result:
    print(line.decode('utf-8').strip())

sys的命令交互

在我們學(xué)習(xí)Python時(shí)，一般使用input()進(jìn)行用戶輸入數(shù)據(jù)。但是其實(shí)sys庫(kù)也可以進(jìn)行輸入輸出判斷，但它涉及的是進(jìn)程間的交互，示例如下：

import sys

sys.stderr.write('開(kāi)始\n')
sys.stderr.flush()

while True:
    next_line = sys.stdin.readline()
    sys.stderr.flush()
    if next_line.strip() == "9599":
        break
    sys.stdout.write(next_line)
    sys.stdout.flush()
sys.stderr.write('結(jié)束\n')
sys.stderr.flush()

運(yùn)行之后，效果如下：

到此這篇關(guān)于Python中使用subprocess庫(kù)創(chuàng)建附加進(jìn)程的文章就介紹到這了,更多相關(guān)Python附加進(jìn)程內(nèi)容請(qǐng)搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

最新評(píng)論