一.subprocess模塊
subprocess是Python 2.4中新增的一個模塊,它允許你生成新的行程,連接到它們的 input/output/error 管道,并獲取它們的回傳(狀態)碼,這個模塊的目的在于替換幾個舊的模塊和方法,如:
os.system
os.spawn*
1.subprocess模塊中的常用函式
| 函式 | 描述 |
|---|---|
| subprocess.run() | Python 3.5中新增的函式,執行指定的命令,等待命令執行完成后回傳一個包含執行結果的CompletedProcess類的實體, |
| subprocess.call() | 執行指定的命令,回傳命令執行狀態,其功能類似于os.system(cmd), |
| subprocess.check_call() | Python 2.5中新增的函式, 執行指定的命令,如果執行成功則回傳狀態碼,否則拋出例外,其功能等價于subprocess.run(…, check=True), |
| subprocess.check_output() | Python 2.7中新增的的函式,執行指定的命令,如果執行狀態碼為0則回傳命令執行結果,否則拋出例外, |
| subprocess.getoutput(cmd) | 接收字串格式的命令,執行命令并回傳執行結果,其功能類似于os.popen(cmd).read()和commands.getoutput(cmd), |
| subprocess.getstatusoutput(cmd) | 執行cmd命令,回傳一個元組(命令執行狀態, 命令執行結果輸出),其功能類似于commands.getstatusoutput(), |
說明:
-
在Python 3.5之后的版本中,官方檔案中提倡通過subprocess.run()函式替代其他函式來使用subproccess模塊的功能;
-
在Python 3.5之前的版本中,我們可以通過subprocess.call(),subprocess.getoutput()等上面列出的其他函式來使用subprocess模塊的功能;
-
subprocess.run()、subprocess.call()、subprocess.check_call()和subprocess.check_output()都是通過對subprocess.Popen的封裝來實作的高級函式,因此如果我們需要更復雜功能時,可以通過subprocess.Popen來完成,
-
subprocess.getoutput()和subprocess.getstatusoutput()函式是來自Python 2.x的commands模塊的兩個遺留函式,它們隱式的呼叫系統shell,并且不保證其他函式所具有的安全性和例外處理的一致性,另外,它們從Python 3.3.4開始才支持Windows平臺,
2.上面各函式的定義及引數說明
函式引數串列:
subprocess.run(args, *, stdin=None, input=None, stdout=None, stderr=None, shell=False, timeout=None, check=False, universal_newlines=False)
subprocess.call(args, *, stdin=None, stdout=None, stderr=None, shell=False, timeout=None)
subprocess.check_call(args, *, stdin=None, stdout=None, stderr=None, shell=False, timeout=None)
subprocess.check_output(args, *, stdin=None, stderr=None, shell=False, universal_newlines=False, timeout=None)
subprocess.getstatusoutput(cmd)
subprocess.getoutput(cmd)
引數說明:
-
args: 要執行的shell命令,默認應該是一個字串序列,如[‘df’, ‘-Th’]或(‘df’, ‘-Th’),也可以是一個字串,如’df -Th’,但是此時需要把shell引數的值置為True,
-
shell: 如果shell為True,那么指定的命令將通過shell執行,如果我們需要訪問某些shell的特性,如管道、檔案名通配符、環境變數擴展功能,這將是非常有用的,當然,python本身也提供了許多類似shell的特性的實作,如glob、fnmatch、os.walk()、os.path.expandvars()、os.expanduser()和shutil等,
-
check: 如果check引數的值是True,且執行命令的行程以非0狀態碼退出,則會拋出一個CalledProcessError的例外,且該例外物件會包含 引數、退出狀態碼、以及stdout和stderr(如果它們有被捕獲的話),
-
stdout, stderr:input: 該引數是傳遞給Popen.communicate(),通常該引數的值必須是一個位元組序列,如果universal_newlines=True,則其值應該是一個字串,
(1.run()函式默認不會捕獲命令執行結果的正常輸出和錯誤輸出,如果我們向獲取這些內容需要傳遞subprocess.PIPE,然后可以通過回傳的CompletedProcess類實體的stdout和stderr屬性或捕獲相應的內容;
(2.call()和check_call()函式回傳的是命令執行的狀態碼,而不是CompletedProcess類實體,所以對于它們而言,stdout和stderr不適合賦值為subprocess.PIPE;
(3.check_output()函式默認就會回傳命令執行結果,所以不用設定stdout的值,如果我們希望在結果中捕獲錯誤資訊,可以執行stderr=subprocess.STDOUT, -
universal_newlines: 該引數影響的是輸入與輸出的資料格式,比如它的值默認為False,此時stdout和stderr的輸出是位元組序列;當該引數的值設定為True時,stdout和stderr的輸出是字串,
3.subprocess.CompletedProcess類介紹
需要說明的是,subprocess.run()函式是Python3.5中新增的一個高級函式,其回傳值是一個subprocess.CompletedPorcess類的實體,因此,subprocess.completedPorcess類也是Python 3.5中才存在的,它表示的是一個已結束行程的狀態資訊,它所包含的屬性如下:
- args: 用于加載該行程的引數,這可能是一個串列或一個字串
- returncode: 子行程的退出狀態碼,通常情況下,退出狀態碼為0則表示行程成功運行了;一個負值-N表示這個子行程被信號N終止了
- stdout: 從子行程捕獲的stdout,這通常是一個位元組序列,如果run()函式被呼叫時指定universal_newlines=True,則該屬性值是一個字串,如果run()函式被呼叫時指定stderr=subprocess.STDOUT,那么stdout和stderr將會被整合到這一個屬性中,且stderr將會為None
- stderr: 從子行程捕獲的stderr,它的值與stdout一樣,是一個位元組序列或一個字串,如果stderr滅有被捕獲的話,它的值就為None
- check_returncode(): 如果returncode是一個非0值,則該方法會拋出一個CalledProcessError例外,
4.實體
subprocess.run()
>>> subprocess.run(["ls", "-l"]) # doesn't capture output
CompletedProcess(args=['ls', '-l'], returncode=0)
>>> subprocess.run("exit 1", shell=True, check=True)
Traceback (most recent call last):
...
subprocess.CalledProcessError: Command 'exit 1' returned non-zero exit status 1
>>> subprocess.run(["ls", "-l", "/dev/null"], stdout=subprocess.PIPE)
CompletedProcess(args=['ls', '-l', '/dev/null'], returncode=0,
stdout=b'crw-rw-rw- 1 root root 1, 3 Jan 23 16:23 /dev/null\n')
subprocess.call()
>>> subprocess.call(['ls', '-l'])
總用量 160
drwxr-xr-x 2 wader wader 4096 12月 7 2015 公共的
drwxr-xr-x 2 wader wader 4096 12月 7 2015 模板
drwxr-xr-x 2 wader wader 4096 12月 7 2015 視頻
drwxr-xr-x 2 wader wader 4096 12月 7 2015 圖片
drwxr-xr-x 2 wader wader 4096 12月 7 2015 檔案
drwxr-xr-x 2 wader wader 4096 4月 13 2016 下載
drwxr-xr-x 2 wader wader 4096 12月 7 2015 音樂
drwxr-xr-x 7 wader wader 4096 5月 26 2016 桌面
0
>>> subprocess.call('ls -l', shell=True)
總用量 160
drwxr-xr-x 2 wader wader 4096 12月 7 2015 公共的
drwxr-xr-x 2 wader wader 4096 12月 7 2015 模板
drwxr-xr-x 2 wader wader 4096 12月 7 2015 視頻
drwxr-xr-x 2 wader wader 4096 12月 7 2015 圖片
drwxr-xr-x 2 wader wader 4096 12月 7 2015 檔案
drwxr-xr-x 2 wader wader 4096 4月 13 2016 下載
drwxr-xr-x 2 wader wader 4096 12月 7 2015 音樂
drwxr-xr-x 7 wader wader 4096 5月 26 2016 桌面
0
>>> subprocess.call(['ls', '-l'], stdout=subprocess.DEVNULL)
0
>>> subprocess.call(['ls', '-l', '/test'])
ls: 無法訪問/test: 沒有那個檔案或目錄
2
suprocess.check_call()
>>> subprocess.check_call(['ls', '-l'])
總用量 160
drwxr-xr-x 2 wader wader 4096 12月 7 2015 公共的
drwxr-xr-x 2 wader wader 4096 12月 7 2015 模板
drwxr-xr-x 2 wader wader 4096 12月 7 2015 視頻
drwxr-xr-x 2 wader wader 4096 12月 7 2015 圖片
drwxr-xr-x 2 wader wader 4096 12月 7 2015 檔案
drwxr-xr-x 2 wader wader 4096 4月 13 2016 下載
drwxr-xr-x 2 wader wader 4096 12月 7 2015 音樂
drwxr-xr-x 7 wader wader 4096 5月 26 2016 桌面
0
>>> subprocess.check_call('ls -l', shell=True)
總用量 160
drwxr-xr-x 2 wader wader 4096 12月 7 2015 公共的
drwxr-xr-x 2 wader wader 4096 12月 7 2015 模板
drwxr-xr-x 2 wader wader 4096 12月 7 2015 視頻
drwxr-xr-x 2 wader wader 4096 12月 7 2015 圖片
drwxr-xr-x 2 wader wader 4096 12月 7 2015 檔案
drwxr-xr-x 2 wader wader 4096 4月 13 2016 下載
drwxr-xr-x 2 wader wader 4096 12月 7 2015 音樂
drwxr-xr-x 7 wader wader 4096 5月 26 2016 桌面
0
>>> subprocess.check_call('ls -l /test', shell=True)
ls: 無法訪問/test: 沒有那個檔案或目錄
Traceback (most recent call last):
File "<stdin>", line 1, in <module>
File "/usr/lib/python3.4/subprocess.py", line 557, in check_call
raise CalledProcessError(retcode, cmd)
subprocess.CalledProcessError: Command 'ls -l /test' returned non-zero exit status 2
subprocess.check_output()
>>> ret = subprocess.check_output(['ls', '-l'])
>>> print(ret)
b' \xe5\x85\xac\xe5\x85\xb1\xe7\x9a\x84\ndrwxr-xr-x 2 wader wader 4096 12\xe6\x9c\x88 7 2015 \xe6\xa8\xa1\xe6\x9d\xbf\ndrwxr-xr-x 2 wader wader 4096 12\xe6\x9c\x88 7 2015 \xe8\xa7\x86\xe9\xa2\x91\ndrwxr-xr-x 2 wader wader 4096 12\xe6\x9c\x88 7 2015 \xe5\x9b\xbe\xe7\x89\x87\ndrwxr-xr-x 2 wader wader 4096 12\xe6\x9c\x88 7 2015 \xe6\x96\x87\xe6\xa1\xa3\ndrwxr-xr-x 2 wader wader 4096 4\xe6\x9c\x88 13 2016 \xe4\xb8\x8b\xe8\xbd\xbd\ndrwxr-xr-x 2 wader wader 4096 12\xe6\x9c\x88 7 2015 \xe9\x9f\xb3\xe4\xb9\x90\ndrwxr-xr-x 7 wader wader 4096 5\xe6\x9c\x88 26 2016 \xe6\xa1\x8c\xe9\x9d\xa2\n'
>>> ret = subprocess.check_output(['ls', '-l'], universal_newlines=True)
>>> print(ret)
總用量 160
drwxr-xr-x 2 wader wader 4096 12月 7 2015 公共的
drwxr-xr-x 2 wader wader 4096 12月 7 2015 模板
drwxr-xr-x 2 wader wader 4096 12月 7 2015 視頻
drwxr-xr-x 2 wader wader 4096 12月 7 2015 圖片
drwxr-xr-x 2 wader wader 4096 12月 7 2015 檔案
drwxr-xr-x 2 wader wader 4096 4月 13 2016 下載
drwxr-xr-x 2 wader wader 4096 12月 7 2015 音樂
drwxr-xr-x 7 wader wader 4096 5月 26 2016 桌面
subprocess.getoutput()與subprocess.getstatusoutput()
>>> ret = subprocess.getoutput('ls -l')
>>> print(ret)
總用量 160
drwxr-xr-x 2 wader wader 4096 12月 7 2015 公共的
drwxr-xr-x 2 wader wader 4096 12月 7 2015 模板
drwxr-xr-x 2 wader wader 4096 12月 7 2015 視頻
drwxr-xr-x 2 wader wader 4096 12月 7 2015 圖片
drwxr-xr-x 2 wader wader 4096 12月 7 2015 檔案
drwxr-xr-x 2 wader wader 4096 4月 13 2016 下載
drwxr-xr-x 2 wader wader 4096 12月 7 2015 音樂
drwxr-xr-x 7 wader wader 4096 5月 26 2016 桌面
>>> retcode, output = subprocess.getstatusoutput('ls -l')
>>> print(retcode)
0
>>> print(output)
總用量 160
drwxr-xr-x 2 wader wader 4096 12月 7 2015 公共的
drwxr-xr-x 2 wader wader 4096 12月 7 2015 模板
drwxr-xr-x 2 wader wader 4096 12月 7 2015 視頻
drwxr-xr-x 2 wader wader 4096 12月 7 2015 圖片
drwxr-xr-x 2 wader wader 4096 12月 7 2015 檔案
drwxr-xr-x 2 wader wader 4096 4月 13 2016 下載
drwxr-xr-x 2 wader wader 4096 12月 7 2015 音樂
drwxr-xr-x 7 wader wader 4096 5月 26 2016 桌面
>>> retcode, output = subprocess.getstatusoutput('ls -l /test')
>>> print(retcode)
2
>>> print(output)
ls: 無法訪問/test: 沒有那個檔案或目錄
二.subprocess.Popen介紹
該類用于在一個新的行程中執行一個子程式,前面我們提到過,上面介紹的這些函式都是基于subprocess.Popen類實作的,通過使用這些被封裝后的高級函式可以很方面的完成一些常見的需求,由于subprocess模塊底層的行程創建和管理是由Popen類來處理的,因此,當我們無法通過上面哪些高級函式來實作一些不太常見的功能時就可以通過subprocess.Popen類提供的靈活的api來完成,
1.subprocess.Popen的建構式
class subprocess.Popen(args, bufsize=-1, executable=None, stdin=None, stdout=None, stderr=None,
preexec_fn=None, close_fds=True, shell=False, cwd=None, env=None, universal_newlines=False,
startup_info=None, creationflags=0, restore_signals=True, start_new_session=False, pass_fds=())
引數說明:
- args: 要執行的shell命令,可以是字串,也可以是命令各個引陣列成的序列,當該引數的值是一個字串時,該命令的解釋程序是與平臺相關的,因此通常建議將args引數作為一個序列傳遞,
- bufsize: 指定快取策略,0表示不緩沖,1表示行緩沖,其他大于1的數字表示緩沖區大小,負數 表示使用系統默認緩沖策略,stdin, stdout, stderr: 分別表示程式標準輸入、輸出、錯誤句柄,
- preexec_fn: 用于指定一個將在子行程運行之前被呼叫的可執行物件,只在Unix平臺下有效,
- close_fds: 如果該引數的值為True,則除了0,1和2之外的所有檔案描述符都將會在子行程執行之前被關閉,
- shell: 該引數用于標識是否使用shell作為要執行的程式,如果shell值為True,則建議將args引數作為一個字串傳遞而不要作為一個序列傳遞,
- cwd: 如果該引數值不是None,則該函式將會在執行這個子行程之前改變當前作業目錄,
- env: 用于指定子行程的環境變數,如果env=None,那么子行程的環境變數將從父行程中繼承,如果env!=None,它的值必須是一個映射物件,
- universal_newlines: 如果該引數值為True,則該檔案物件的stdin,stdout和stderr將會作為文本流被打開,否則他們將會被作為二進制流被打開,
- startupinfo和creationflags: 這兩個引數只在Windows下有效,它們將被傳遞給底層的CreateProcess()函式,用于設定子行程的一些屬性,如主視窗的外觀,行程優先級等,
2.subprocess.Popen類的實體可呼叫的方法
| 方法 | 描述 |
|---|---|
| Popen.poll() | 用于檢查子行程(命令)是否已經執行結束,沒結束回傳None,結束后回傳狀態碼, |
| Popen.wait(timeout=None) | 等待子行程結束,并回傳狀態碼;如果在timeout指定的秒數之后行程還沒有結束,將會拋出一個TimeoutExpired例外, |
| Popen.communicate(input=None, timeout=None) | 該方法可用來與行程進行互動,比如發送資料到stdin,從stdout和stderr讀取資料,直到到達檔案末尾, |
| Popen.send_signal(signal) | 發送指定的信號給這個子行程, |
| Popen.terminate() | 停止該子行程, |
| Popen.kill() | 殺死該子行程, |
關于communicate()方法的說明:
- 該方法中的可選引數 input 應該是將被發送給子行程的資料,或者如沒有資料發送給子行程,該引數應該是None,input引數的資料型別必須是位元組串,如果universal_newlines引數值為True,則input引數的資料型別必須是字串,
- 該方法回傳一個元組(stdout_data, stderr_data),這些資料將會是位元組穿或字串(如果universal_newlines的值為True),
- 如果在timeout指定的秒數后該行程還沒有結束,將會拋出一個TimeoutExpired例外,捕獲這個例外,然后重新嘗試通信不會丟失任何輸出的資料,但是超時之后子行程并沒有被殺死,為了合理的清除相應的內容,一個好的應用應該手動殺死這個子行程來結束通信,
- 需要注意的是,這里讀取的資料是緩沖在記憶體中的,所以,如果資料大小非常大或者是無限的,就不應該使用這個方法
3.subprocess.Popen使用實體
實體1
>>> import subprocess
>>>
>>> p = subprocess.Popen('df -Th', stdout=subprocess.PIPE, shell=True)
>>> print(p.stdout.read())
Filesystem Type Size Used Avail Use% Mounted on
/dev/vda1 ext4 40G 12G 26G 31% /
devtmpfs devtmpfs 3.9G 0 3.9G 0% /dev
tmpfs tmpfs 3.9G 0 3.9G 0% /dev/shm
tmpfs tmpfs 3.9G 386M 3.5G 10% /run
tmpfs tmpfs 3.9G 0 3.9G 0% /sys/fs/cgroup
tmpfs tmpfs 783M 0 783M 0% /run/user/0
tmpfs tmpfs 783M 0 783M 0% /run/user/1000
實體2
>>> obj = subprocess.Popen(["python"], stdin=subprocess.PIPE, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
>>> obj.stdin.write('print(1) \n')
>>> obj.stdin.write('print(2) \n')
>>> obj.stdin.write('print(3) \n')
>>> out,err = obj.communicate()
>>> print(out)
1
2
3
>>> print(err)
實體3
'''
學習中遇到問題沒人解答?小編創建了一個Python學習交流群:489111204
尋找有志同道合的小伙伴,互幫互助,群里還有不錯的視頻學習教程和PDF電子書!
'''
>>> obj = subprocess.Popen(["python"], stdin=subprocess.PIPE, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
>>> out,err = obj.communicate(input='print(1) \n')
>>> print(out)
1
>>> print(err)
實體4
實作類似df -Th | grep data命令的功能,實際上就是實作shell中管道的共功能,
>>>
>>> p1 = subprocess.Popen(['df', '-Th'], stdout=subprocess.PIPE)
>>> p2 = subprocess.Popen(['grep', 'data'], stdin=p1.stdout, stdout=subprocess.PIPE)
>>> out,err = p2.communicate()
>>> print(out)
/dev/vdb1 ext4 493G 4.8G 463G 2% /data
/dev/vdd1 ext4 1008G 420G 537G 44% /data1
/dev/vde1 ext4 985G 503G 432G 54% /data2
>>> print(err)
None
三.總結
那么我們到底該用哪個模塊、哪個函式來執行命令與系統及系統進行互動呢?下面我們來做個總結:
- 首先應該知道的是,Python2.4版本引入了subprocess模塊用來替換os.system()、os.popen()、os.spawn*()等函式以及commands模塊;也就是說如果你使用的是Python 2.4及以上的版本就應該使用subprocess模塊了,
- 如果你的應用使用的Python 2.4以上,但是是Python 3.5以下的版本,Python官方給出的建議是使用subprocess.call()函式,Python 2.5中新增了一個subprocess.check_call()函式,Python 2.7中新增了一個subprocess.check_output()函式,這兩個函式也可以按照需求進行使用,
- 如果你的應用使用的是Python 3.5及以上的版本(目前應該還很少),Python官方給出的建議是盡量使用subprocess.run()函式,
- 當subprocess.call()、subprocess.check_call()、subprocess.check_output()和subprocess.run()這些高級函式無法滿足需求時,我們可以使用subprocess.Popen類來實作我們需要的復雜功能,
轉載請註明出處,本文鏈接:https://www.uj5u.com/houduan/527786.html
標籤:Python
下一篇:重溫Python基礎——變數
