Python從subprocess運行的子進程中如何實時獲取輸出的例子

這篇文章主要介紹Python從subprocess運行的子進程中如何實時獲取輸出的例子，文中介紹的非常詳細，具有一定的參考價值，感興趣的小伙伴們一定要看完！

有些時候，我們需要將某些程序放到子進程中去運行，以達到整合系統的目的。在Python中，一個非常好的選擇就是使用subprocess模塊，本模塊為開辟子進程去執行子程序提供了統一的接口，更加便于學習和使用。

同時，對于在子進程里的程序，我們希望能夠實時獲取其輸出，以在主進程中打印相關信息，使我們能夠了解當前子程序的執行進度。對此，subprocess模塊也提供了相應的參數，能夠將子程序的標準輸出和標準錯誤輸出返回給主程序。

下面，我們就通過一個例子來說明這個功能。首先，我們需要一個用于模擬標準輸出和標準錯誤輸出的“子程序”——subprogram.py：

import sys
import time
 
 
for i in range(5):
  sys.stdout.write('Processing {}\n'.format(i))
  time.sleep(1)
 
for i in range(5):
  sys.stderr.write('Error {}\n'.format(i))
  time.sleep(1)

可以看到這個程序非常簡單，分別向標準輸出和標準錯誤寫入了5條信息，并且輸出之間有1秒的間隔。下面是驅動這個“子程序”運行的“主程序”——main.py：

import shlex
import subprocess
 
if __name__ == '__main__':
  shell_cmd = 'python3 subprogram.py'
  cmd = shlex.split(shell_cmd)
  p = subprocess.Popen(cmd, shell=False, stdout=subprocess.PIPE, stderr=subprocess.STDOUT)
  while p.poll() is None:
    line = p.stdout.readline()
    line = line.strip()
    if line:
      print('Subprogram output: [{}]'.format(line))
  if p.returncode == 0:
    print('Subprogram success')
  else:
    print('Subprogram failed')

可以看到，我們通過指定stderr=subprocess.STDOUT，將子程序的標準錯誤輸出重定向到了標準輸出，以使我們可以直接從標準輸出中同時獲取標準輸出和標準錯誤的信息。運行這個程序，我們會期待main.py會每秒輸出一次信息到控制臺，但是事實上，我們直到等了10秒之后才一次性看到所有的10條輸出。

產生這種現象的原因也非常簡單，就是標準輸出和標準錯誤有一個緩存的概念，它不會立即將程序的標準輸出內容返回，而是會做一定的緩存，直到緩存滿或者程序結束強制清空緩存時才輸出。了解到問題的原因，解決問題的方法也就一目了然了，我們只需要在子程序中，每次輸出后去手動清空一下緩存即可，以下是修改過的subprogram.py：

import sys
import time
 
 
for i in range(5):
  sys.stdout.write('Processing {}\n'.format(i))
  sys.stdout.flush()
  time.sleep(1)
 
for i in range(5):
  sys.stderr.write('Error {}\n'.format(i))
  sys.stderr.flush()
  time.sleep(1)

經過上述的修改之后，再次運行main.py程序，我們會看到，每秒會輸出一條信息，達到了我們在主程序中，去追蹤子程序執行過程的目的。

Python主要用來做什么

Python主要應用于：1、Web開發；2、數據科學研究；3、網絡爬蟲；4、嵌入式應用開發；5、游戲開發；6、桌面應用開發。

以上是“Python從subprocess運行的子進程中如何實時獲取輸出的例子”這篇文章的所有內容，感謝各位的閱讀！希望分享的內容對大家有幫助，更多相關知識，歡迎關注億速云行業資訊頻道！