python中使用subprocess批量執行linux下命令

君惜發表於2019-02-16

可以執行shell命令的相關模組和函式有:

  • os.system
  • os.spawn
  • os.popen –廢棄
  • popen –廢棄
  • commands –廢棄,3.x中被移除

以上執行shell命令的相關的模組和函式的功能均在 subprocess 模組中實現,並提供了更豐富的功能。

subprocess

call

執行命令,返回狀態碼

>>> import subprocess
>>> ret = subprocess.call(["ls", "-l"], shell=False)
total 4684
-rw-r--r-- 1 root root     454 May  5 12:20 aa.py
-rw-r--r-- 1 root root       0 May  8 16:51 aa.txt
-rw-r--r-- 1 root root 4783286 Apr 11 16:39 DockerToolbox.exe
-rw-r--r-- 1 root root     422 May  5 12:20 ip_info.txt
-rw-r--r-- 1 root root     718 Apr 19 10:52 my.cnf
>>> ret = subprocess.call("ls -l", shell=True)
total 4684
-rw-r--r-- 1 root root     454 May  5 12:20 aa.py
-rw-r--r-- 1 root root       0 May  8 16:51 aa.txt
-rw-r--r-- 1 root root 4783286 Apr 11 16:39 DockerToolbox.exe
-rw-r--r-- 1 root root     422 May  5 12:20 ip_info.txt
-rw-r--r-- 1 root root     718 Apr 19 10:52 my.cnf
>>> print(ret)
0

check_call

執行命令,如果執行狀態碼是 0 ,則返回0,否則拋異常

>>> subprocess.check_call(["ls", "-l"])
total 4684
-rw-r--r-- 1 root root     454 May  5 12:20 aa.py
-rw-r--r-- 1 root root       0 May  8 16:51 aa.txt
-rw-r--r-- 1 root root 4783286 Apr 11 16:39 DockerToolbox.exe
-rw-r--r-- 1 root root     422 May  5 12:20 ip_info.txt
-rw-r--r-- 1 root root     718 Apr 19 10:52 my.cnf
0
>>> subprocess.check_call("exit 1", shell=True)
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "/usr/local/python3.5/lib/python3.5/subprocess.py", line 581, in check_call
    raise CalledProcessError(retcode, cmd)
subprocess.CalledProcessError: Command `exit 1` returned non-zero exit status 1

check_output

執行命令,如果狀態碼是 0 ,則返回執行結果,否則拋異常

>>> subprocess.check_output(["echo", "Hello World!"])
b`Hello World!
`
>>> subprocess.check_output("exit 1", shell=True)
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "/usr/local/python3.5/lib/python3.5/subprocess.py", line 626, in check_output
    **kwargs).stdout
  File "/usr/local/python3.5/lib/python3.5/subprocess.py", line 708, in run
    output=stdout, stderr=stderr)
subprocess.CalledProcessError: Command `exit 1` returned non-zero exit status 1

subprocess.Popen(…)

用於執行復雜的系統命令
引數:

args:shell命令,可以是字串或者序列型別(如:list,元組)
bufsize:指定緩衝。0 無緩衝,1 行緩衝,其他 緩衝區大小,負值 系統緩衝
stdin, stdout, stderr:分別表示程式的標準輸入、輸出、錯誤控制程式碼
preexec_fn:只在Unix平臺下有效,用於指定一個可執行物件(callable object),它將在子程式執行之前被呼叫
close_sfs:在windows平臺下,如果close_fds被設定為True,則新建立的子程式將不會繼承父程式的輸入、輸出、錯誤管道。
所以不能將close_fds設定為True同時重定向子程式的標準輸入、輸出與錯誤(stdin, stdout, stderr)。
shell:同上
cwd:用於設定子程式的當前目錄
env:用於指定子程式的環境變數。如果env = None,子程式的環境變數將從父程式中繼承。
universal_newlines:不同系統的換行符不同,True -> 同意使用 n
startupinfo與createionflags只在windows下有效
將被傳遞給底層的CreateProcess()函式,用於設定子程式的一些屬性,如:主視窗的外觀,程式的優先順序等等

執行普通命令

>>> import subprocess
>>> ret1 = subprocess.Popen(["mkdir","t1"])
>>> ret2 = subprocess.Popen("mkdir t2", shell=True)
>>> print(ret1)
<subprocess.Popen object at 0x7f4d7609dd30>
>>> print(ret2)
<subprocess.Popen object at 0x7f4d7609dc18>

終端輸入的命令分為兩種:

  • 輸入即可得到輸出,如:ifconfig
  • 輸入進行某環境,依賴再輸入,如:python
>>> import subprocess
>>> obj = subprocess.Popen("mkdir t3", shell=True, cwd=`/tmp/`,)
>>> import subprocess
>>> obj = subprocess.Popen(["python"], stdin=subprocess.PIPE, stdout=subprocess.PIPE, stderr=subprocess.PIPE, universal_newlines=True)
>>> obj.stdin.write("print(1)
")
9
>>> obj.stdin.write("print(2)")
8
>>> obj.stdin.close()
>>> cmd_out = obj.stdout.read()
>>> obj.stdout.close()
>>> cmd_error = obj.stderr.read()
>>> obj.stderr.close()
>>> print(cmd_out)
1
2
>>> print(cmd_error)
>>> import subprocess
>>> 
>>> obj = subprocess.Popen(["python"], stdin=subprocess.PIPE, stdout=subprocess.PIPE, stderr=subprocess.PIPE, universal_newlines=True)
>>> obj.stdin.write("print(1)
")
9
>>> obj.stdin.write("print(2)")
8
>>> 
>>> out_error_list = obj.communicate()
>>> print(out_error_list)
(`1
2
`, ``)
>>> obj = subprocess.Popen(["python"], stdin=subprocess.PIPE, stdout=subprocess.PIPE, stderr=subprocess.PIPE, universal_newlines=True)
>>> out_error_list = obj.communicate(`print("hello")`)
>>> print(out_error_list)
(`hello
`, ``)

相關文章