subprocess

官方中文文檔

介紹參考文檔，我的直觀感受和實際用法是：subprocess可以開啟一個子進程來運行cmd命令。那就意味著可以在一個py文件里運行另一個py文件

例1-快速使用subprocess

新建一個目錄，目錄下有兩個文件

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

在hello.py中

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

在main.py中

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

執行main.py文件得到如下結果

hello world!

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

修改代碼如下：

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

運行后

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

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

例3-全面的返回值介紹

• args：被用作啟動進程的參數，可能是列表或字符串
• returncode：子進程的退出狀態碼
• stdout：從子進程捕獲到的標準輸出，但是沒設置subprocess.run()中的stdout參數時，這一項是None。
• stderr：捕獲到的子進程標準錯誤，沒設置subprocess.run()中的stderr參數時，這一項是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)

結果：

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

Process finished with exit code 0

可以看到，沒有設置subprocess.run()中的參數stdout和stderr時，這兩項都是None

例4-代碼有bug的情況

新建fail.py，故意制造一個bug

# fail.py
a = 

修改main.py

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

再運行main函數，得到返回

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

可以看到，先是正確打印了hello.py的內容，然后是fail.py的錯誤信息。

例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)

結果

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'

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

同時可以發現，子進程print和報錯內容就不會在父進程打印輸出了。

注意這里的res.stdout是一串二進制字符串。如果設置encoding參數，拿到的就是字符串。

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

例6-與子進程進行通信

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

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

修改hello.py接收傳進來的字符串。

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

結果

hello input

Process finished with exit code 0