Python標準庫之Sys模組使用詳解

sys 模組提供了許多函數和變數來處理 Python 運行時環境的不同部分.

處理命令列參數

在解譯器啟動後, argv 列表包含了傳遞給指令碼的所有參數, 列表的第一個元素為指令碼自身的名稱.

使用sys模組獲得指令碼的參數

代碼如下:

print "script name is", sys.argv[0] # 使用sys.argv[0]採集指令碼名稱

if len(sys.argv) > 1:
print "there are", len(sys.argv)-1, "arguments:" # 使用len(sys.argv)-1採集參數個數-1為減去[0]指令碼名稱
for arg in sys.argv[1:]: #輸出除了[0]外所有參數
print arg
else:
print "there are no arguments!"

如果是從標準輸入讀入指令碼 (比如 "python < sys-argv-example-1.py"), 指令碼的名稱將被設定為空白串.

如果把指令碼作為字串傳遞給python (使用 -c 選項), 指令碼名會被設定為 "-c".

處理模組

path 列表是一個由目錄名構成的列表, Python 從中尋找擴充模組( Python 源模組, 編譯模組,或者二進位擴充).

啟動 Python 時,這個列表從根據內建規則, PYTHONPATH 環境變數的內容, 以及註冊表( Windows 系統)等進行初始化.

由於它只是一個普通的列表, 你可以在程式中對它進行操作,

使用sys模組操作模組搜尋路徑

代碼如下:

print "path has", len(sys.path), "members"

sys.path.insert(0, "samples") #將路徑插入到path,[0]中
import sample

sys.path = [] #刪除path中所有路徑
import random

使用sys模組尋找內建模組

builtin_module_names 列表包含 Python 解譯器中所有內建模組的名稱

代碼如下:

def dump(module):
print module, "=>",
if module in sys.builtin_module_names: #尋找內建模組是否存在
print ""
else:
module = _ _import_ _(module) #非內建模組輸出模組路徑
print module._ _file_ _

dump("os")
dump("sys")
dump("string")
dump("strop")
dump("zlib")

os => C:\python\lib\os.pyc
sys =>
string => C:\python\lib\string.pyc
strop =>
zlib => C:\python\zlib.pyd

使用sys模組尋找已匯入的模組

modules 字典包含所有載入的模組. import 語句在從磁碟匯入內容之前會先檢查這個字典.

Python 在處理你的指令碼之前就已經匯入了很多模組.

代碼如下:

print sys.modules.keys()

['os.path', 'os', 'exceptions', '_ _main_ _', 'ntpath', 'strop', 'nt',
'sys', '_ _builtin_ _', 'site', 'signal', 'UserDict', 'string', 'stat']

使用sys模組獲得當前平台

sys.platform 返回當前平台 出現如： "win32" "linux2" 等

處理標準輸出/輸入

標準輸入和標準錯誤 (通常縮寫為 stdout 和 stderr) 是內建在每一個 UNIX 系統中的管道。

當你 print 某些東西時，結果前往 stdout 管道；

當你的程式崩潰並列印出調試資訊 (例如 Python 中的 traceback (錯誤跟蹤)) 的時候，資訊前往 stderr 管道

代碼如下:

>>> for i in range(3):
... print'Dive in'

Dive in
Dive in
Dive in
>>> import sys
>>> for i in range(3):
... sys.stdout.write('Dive in')

Dive inDive inDive in
>>> for i in range(3):
... sys.stderr.write('Dive in')

Dive inDive inDive in

stdout 是一個類檔案對象；調用它的 write 函數可以列印出你給定的任何字串。

實際上，這就是 print 函數真正做的事情；它在你列印的字串後面加上一個硬斷行符號，然後調用 sys.stdout.write 函數。

在最簡單的例子中，stdout 和 stderr 把它們的輸出發送到相同的地方

和 stdout 一樣，stderr 並不為你添加硬斷行符號；如果需要，要自己加上。

stdout 和 stderr 都是類檔案對象，但是它們都是唯寫的。

它們都沒有 read 方法，只有 write 方法。然而，它們仍然是類檔案對象，因此你可以將其它任何 (類) 檔案對象賦值給它們來重新導向其輸出。

使用sys重新導向輸出

代碼如下:

print 'Dive in' # 標準輸出
saveout = sys.stdout # 終在重新導向前儲存stdout，這樣的話之後你還可以將其設回正常
fsock = open('out.log', 'w') # 開啟一個新檔案用於寫入。如果檔案不存在，將會被建立。如果檔案存在，將被覆蓋。
sys.stdout = fsock # 所有後續的輸出都會被重新導向到剛才開啟的新檔案上。

print 'This message will be logged instead of displayed' # 這樣只會將輸出結果“列印”到記錄檔中；螢幕上不會看到輸出

sys.stdout = saveout # 在我們將 stdout 搞亂之前，讓我們把它設回原來的方式。

fsock.close() # 關閉記錄檔。

重新導向錯誤資訊

fsock = open('error.log', 'w') # 開啟你要儲存調試資訊的記錄檔。
sys.stderr = fsock # 將新開啟的記錄檔的檔案對象賦值給stderr以重新導向標準錯誤。
raise Exception, 'this error will be logged' # 引發一個異常,沒有在螢幕上列印出任何東西,所有正常的跟蹤資訊已經寫進error.log

還要注意你既沒有顯式關閉記錄檔，也沒有將 stderr 設回最初的值。

這樣挺好，因為一旦程式崩潰 (由於引發的異常)，Python 將替我們清理並關閉檔案

列印到 stderr

向標準錯誤寫入錯誤資訊是很常見的，所以有一種較快的文法可以立刻匯出資訊

代碼如下:

>>> print 'entering function'
entering function
>>> import sys
>>> print >> sys.stderr, 'entering function'

entering function

print 語句的快捷文法可以用於寫入任何開啟的檔案 (或者是類檔案對象)。

在這裡，你可以將單個print語句重新導向到stderr而且不用影響後面的print語句。

使用sys模組退出程式

代碼如下:

import sys
sys.exit(1)

注意 sys.exit 並不是立即退出. 而是引發一個 SystemExit 異常. 這意味著你可以在主程式中捕獲對 sys.exit 的調用

捕獲sys.exit調用

代碼如下:

import sys
print "hello"
try:
sys.exit(1)
except SystemExit: # 捕獲退出的異常
pass # 捕獲後不做任何操作
print "there"

hello
there

如果準備在退出前自己清理一些東西(比如刪除臨時檔案), 你可以配置一個 "退出處理函數"(exit handler), 它將在程式退出的時候自動被調用

另一種捕獲sys.exit調用的方法

代碼如下:

def exitfunc():
print "world"

sys.exitfunc = exitfunc # 設定捕獲時調用的函數

print "hello"
sys.exit(1) # 退出自動調用exitfunc()後，程式依然退出了
print "there" # 不會被 print

hello
world

