12.3 操作文件和目錄
如果我們要操作文件、目錄,可以在命令行下面輸入操作系統(tǒng)提供的各種命令來完成。比如dir
、cp
等命令。
如果要在Python程序中執(zhí)行這些目錄和文件的操作怎么辦?其實操作系統(tǒng)提供的命令只是簡單地調(diào)用了操作系統(tǒng)提供的接口函數(shù),Python內(nèi)置的os
模塊也可以直接調(diào)用操作系統(tǒng)提供的接口函數(shù)。
打開Python交互式命令行,我們來看看如何使用os
模塊的基本功能:
如果是posix
,說明系統(tǒng)是Linux
、Unix
或Mac OS X
,如果是nt
,就是Windows
系統(tǒng)。
要獲取詳細的系統(tǒng)信息,可以調(diào)用uname()
函數(shù):
注意uname()
函數(shù)在Windows上不提供,也就是說,os
模塊的某些函數(shù)是跟操作系統(tǒng)相關的。
環(huán)境變量
在操作系統(tǒng)中定義的環(huán)境變量,全部保存在os.environ
這個變量中,可以直接查看:
要獲取某個環(huán)境變量的值,可以調(diào)用os.environ.get('key')
:
操作文件和目錄
操作文件和目錄的函數(shù)一部分放在os
模塊中,一部分放在os.path
模塊中,這一點要注意一下。查看、創(chuàng)建和刪除目錄可以這么調(diào)用:
把兩個路徑合成一個時,不要直接拼字符串,而要通過os.path.join()
函數(shù),這樣可以正確處理不同操作系統(tǒng)的路徑分隔符。在Linux/Unix/Mac下,os.path.join()
返回這樣的字符串:
而Windows下會返回這樣的字符串:
同樣的道理,要拆分路徑時,也不要直接去拆字符串,而要通過os.path.split()
函數(shù),這樣可以把一個路徑拆分為兩部分,后一部分總是最后級別的目錄或文件名:
os.path.splitext()
可以直接讓你得到文件擴展名,很多時候非常方便:
這些合并、拆分路徑的函數(shù)并不要求目錄和文件要真實存在,它們只對字符串進行操作。
文件操作使用下面的函數(shù)。假定當前目錄下有一個test.txt
文件:
但是復制文件的函數(shù)居然在os
模塊中不存在!原因是復制文件并非由操作系統(tǒng)提供的系統(tǒng)調(diào)用。理論上講,我們通過上一節(jié)的讀寫文件可以完成文件復制,只不過要多寫很多代碼。
幸運的是shutil
模塊提供了copyfile()
的函數(shù),你還可以在shutil
模塊中找到很多實用函數(shù),它們可以看做是os
模塊的補充。
最后看看如何利用Python的特性來過濾文件。比如我們要列出當前目錄下的所有目錄,只需要一行代碼:
要列出所有的.py
文件,也只需一行代碼:
是不是非常簡潔?
小結(jié)
Python的os
模塊封裝了操作系統(tǒng)的目錄和文件操作,要注意這些函數(shù)有的在os
模塊中,有的在os.path
模塊中。