如果我們要操作文件、目錄,可以在命令行下面輸入操作系統(tǒng)提供的各種命令來(lái)完成。比如dir、cp等命令。
如果要在Python程序中執(zhí)行這些目錄和文件的操作怎么辦?其實(shí)操作系統(tǒng)提供的命令只是簡(jiǎn)單地調(diào)用了操作系統(tǒng)提供的接口函數(shù),Python內(nèi)置的os模塊也可以直接調(diào)用操作系統(tǒng)提供的接口函數(shù)。
打開(kāi)Python交互式命令行,我們來(lái)看看如何使用os模塊的基本功能:
>>> import os
>>> os.name # 操作系統(tǒng)類型
'posix'
如果是posix,說(shuō)明系統(tǒng)是Linux、Unix或Mac OS X,如果是nt,就是Windows系統(tǒng)。
要獲取詳細(xì)的系統(tǒng)信息,可以調(diào)用uname()函數(shù):
>>> os.uname()
posix.uname_result(sysname='Darwin', nodename='MichaelMacPro.local', release='14.3.0', version='Darwin Kernel Version 14.3.0: Mon Mar 23 11:59:05 PDT 2015; root:xnu-2782.20.48~5/RELEASE_X86_64', machine='x86_64')
注意uname()函數(shù)在Windows上不提供,也就是說(shuō),os模塊的某些函數(shù)是跟操作系統(tǒng)相關(guān)的。
環(huán)境變量
在操作系統(tǒng)中定義的環(huán)境變量,全部保存在os.environ這個(gè)變量中,可以直接查看:
>>> os.environ
environ({'VERSIONER_PYTHON_PREFER_32_BIT': 'no', 'TERM_PROGRAM_VERSION': '326', 'LOGNAME': 'michael', 'USER': 'michael', 'PATH': '/usr/bin:/bin:/usr/sbin:/sbin:/usr/local/bin:/opt/X11/bin:/usr/local/mysql/bin', ...})
要獲取某個(gè)環(huán)境變量的值,可以調(diào)用os.environ.get('key'):
>>> os.environ.get('PATH')
'/usr/bin:/bin:/usr/sbin:/sbin:/usr/local/bin:/opt/X11/bin:/usr/local/mysql/bin'
>>> os.environ.get('x', 'default')?
'default'
操作文件和目錄
操作文件和目錄的函數(shù)一部分放在os模塊中,一部分放在os.path模塊中,這一點(diǎn)要注意一下。查看、創(chuàng)建和刪除目錄可以這么調(diào)用:
# 查看當(dāng)前目錄的絕對(duì)路徑:
>>> os.path.abspath('.')
'/Users/michael'
# 在某個(gè)目錄下創(chuàng)建一個(gè)新目錄,首先把新目錄的完整路徑表示出來(lái):
>>> os.path.join('/Users/michael', 'testdir')
'/Users/michael/testdir'
# 然后創(chuàng)建一個(gè)目錄:
>>> os.mkdir('/Users/michael/testdir')
# 刪掉一個(gè)目錄:
>>> os.rmdir('/Users/michael/testdir')
把兩個(gè)路徑合成一個(gè)時(shí),不要直接拼字符串,而要通過(guò)os.path.join()函數(shù),這樣可以正確處理不同操作系統(tǒng)的路徑分隔符。在Linux/Unix/Mac下,os.path.join()返回這樣的字符串:
part-1/part-2
而Windows下會(huì)返回這樣的字符串:
part-1\part-2
同樣的道理,要拆分路徑時(shí),也不要直接去拆字符串,而要通過(guò)os.path.split()函數(shù),這樣可以把一個(gè)路徑拆分為兩部分,后一部分總是最后級(jí)別的目錄或文件名:
>>> os.path.split('/Users/michael/testdir/file.txt')
('/Users/michael/testdir', 'file.txt')
os.path.splitext()可以直接讓你得到文件擴(kuò)展名,很多時(shí)候非常方便:
>>> os.path.splitext('/path/to/file.txt')
('/path/to/file', '.txt')
這些合并、拆分路徑的函數(shù)并不要求目錄和文件要真實(shí)存在,它們只對(duì)字符串進(jìn)行操作。
文件操作使用下面的函數(shù)。假定當(dāng)前目錄下有一個(gè)test.txt文件:
# 對(duì)文件重命名:
>>> os.rename('test.txt', 'test.py')
# 刪掉文件:
>>> os.remove('test.py')
但是復(fù)制文件的函數(shù)居然在os模塊中不存在!原因是復(fù)制文件并非由操作系統(tǒng)提供的系統(tǒng)調(diào)用。理論上講,我們通過(guò)上一節(jié)的讀寫文件可以完成文件復(fù)制,只不過(guò)要多寫很多代碼。
幸運(yùn)的是shutil模塊提供了copyfile()的函數(shù),你還可以在shutil模塊中找到很多實(shí)用函數(shù),它們可以看做是os模塊的補(bǔ)充。
最后看看如何利用Python的特性來(lái)過(guò)濾文件。比如我們要列出當(dāng)前目錄下的所有目錄,只需要一行代碼:
>>> [x for x in os.listdir('.') if os.path.isdir(x)]
['.lein', '.local', '.m2', '.npm', '.ssh', '.Trash', '.vim', 'Applications', 'Desktop', ...]
要列出所有的.py文件,也只需一行代碼:
>>> [x for x in os.listdir('.') if os.path.isfile(x) and os.path.splitext(x)[1]=='.py']
['apis.py', 'config.py', 'models.py', 'pymonitor.py', 'test_db.py', 'urls.py', 'wsgiapp.py']