subprocess

官方中文文檔

介紹參考文檔，我的直觀感受和實(shí)際用法是：subprocess可以開(kāi)啟一個(gè)子進(jìn)程來(lái)運(yùn)行cmd命令。那就意味著可以在一個(gè)py文件里運(yùn)行另一個(gè)py文件

例1-快速使用subprocess

新建一個(gè)目錄，目錄下有兩個(gè)文件

|-demo
? ? |-main.py
? ? |-hello.py

在hello.py中

# hello.py
print('hello world!')

在main.py中

import subprocess
subprocess.run(['python', 'hello.py'])

執(zhí)行main.py文件得到如下結(jié)果

hello world!

例2-subprocess.run()的返回值

修改代碼如下：

# main.py
import subprocess
res = subprocess.run(['python', 'hello.py'])
print("args:", res.args)
print("returncode", res.returncode)

運(yùn)行后

hello world!
args: ['python', 'hello.py']
returncode: 0

returncode 表示你run的這個(gè)py文件過(guò)程是否正確，如果正確，返回0，否則返回1

例3-全面的返回值介紹

• args：被用作啟動(dòng)進(jìn)程的參數(shù)，可能是列表或字符串
• returncode：子進(jìn)程的退出狀態(tài)碼
• stdout：從子進(jìn)程捕獲到的標(biāo)準(zhǔn)輸出，但是沒(méi)設(shè)置subprocess.run()中的stdout參數(shù)時(shí)，這一項(xiàng)是None。
• stderr：捕獲到的子進(jìn)程標(biāo)準(zhǔn)錯(cuò)誤，沒(méi)設(shè)置subprocess.run()中的stderr參數(shù)時(shí)，這一項(xiàng)是None。
• check_returncode()：如果 returncode 非零, 拋出 CalledProcessError.

修改main.py

# main.py
import subprocess
res = subprocess.run(['python', 'hello.py'])
print("args:", res.args)
print("returncode", res.returncode)
print("stdout", res.stdout)
print("stderr", res.stderr)

結(jié)果：

hello world!
args: ['python', 'hello.py']
returncode 0
stdout None
stderr None

Process finished with exit code 0

可以看到，沒(méi)有設(shè)置subprocess.run()中的參數(shù)stdout和stderr時(shí)，這兩項(xiàng)都是None

例4-代碼有bug的情況

新建fail.py，故意制造一個(gè)bug

# fail.py
a = 

修改main.py

# main.py
import subprocess
res = subprocess.run(['python', 'hello.py'])
res2 = subprocess.run(['python', 'fail.py'])

再運(yùn)行main函數(shù)，得到返回

hello world!
? File "fail.py", line 2
? ? a =
? ? ? ^
SyntaxError: invalid syntax

可以看到，先是正確打印了hello.py的內(nèi)容，然后是fail.py的錯(cuò)誤信息。

例5-捕獲stdout和stderr

修改main.py

# main.py
import subprocess
res = subprocess.run(['python', 'hello.py'], stdout=subprocess.PIPE)
res2 = subprocess.run(['python', 'fail.py'], stderr=subprocess.PIPE)
print('hello.py stdout:', res.stdout)
print('fail.py stderr:', res2.stderr)

結(jié)果

hello.py stdout: b'hello world!\r\n'
fail.py stderr: b' ?File "fail.py", line 2\r\n ? ?a =\r\n ? ? ?^\r\nSyntaxError: invalid syntax\r\n'

可以通過(guò)res.stdout與res2.stderr分別拿到正確print的信息和錯(cuò)誤信息。

同時(shí)可以發(fā)現(xiàn)，子進(jìn)程print和報(bào)錯(cuò)內(nèi)容就不會(huì)在父進(jìn)程打印輸出了。

注意這里的res.stdout是一串二進(jìn)制字符串。如果設(shè)置encoding參數(shù)，拿到的就是字符串。

res = subprocess.run(['python', 'hello.py'], 
                     stdout=subprocess.PIPE,
                     encoding='utf8')

例6-與子進(jìn)程進(jìn)行通信

可以通過(guò)subprocess.run()的input參數(shù)給子進(jìn)程發(fā)送消息。如果不設(shè)置encoding，就要傳入二進(jìn)制串，比如b'hello input'

# main.py
import subprocess
from subprocess import PIPE
res = subprocess.run(['python', 'hello.py'],
                     input='hello input',
                     encoding='utf8')

修改hello.py接收傳進(jìn)來(lái)的字符串。

# hello.py 
import sys
data = sys.stdin.read()
print(data)

結(jié)果

hello input

Process finished with exit code 0