通常在讀寫檔案之前,需要判斷檔案或目錄是否存在,不然某些處理方法可能會使程式出錯。所以最好在做任何操作之前,先判斷檔案是否存在。
這裡將介紹三種判斷檔案或資料夾是否存在的方法,分別使用os模組、Try語句、pathlib模組。
1.使用os模組
os模組中的os.path.exists()方法用於檢驗檔案是否存在。
判斷檔案是否存在
import os
os.path.exists(test_file.txt)
#True
os.path.exists(no_exist_file.txt)
#False
判斷資料夾是否存在
import os
os.path.exists(test_dir)
#True
os.path.exists(no_exist_dir)
#False
可以看出用os.path.exists()
方法,判斷檔案和資料夾是一樣。
其實這種方法還是有個問題,假設你想檢查檔案“test_data”是否存在,但是當前路徑下有個叫“test_data”的資料夾,這樣就可能出現誤判。為了避免這樣的情況,可以這樣:
只檢查檔案
import os
os.path.isfile("test-data")
通過這個方法,如果檔案”test-data”不存在將返回False,反之返回True。
即是檔案存在,你可能還需要判斷檔案是否可進行讀寫操作。
判斷檔案是否可做讀寫操作
使用os.access()方法判斷檔案是否可進行讀寫操作。
語法:
os.access(path, mode)
path為檔案路徑,mode為操作模式,有這麼幾種:
os.F_OK: 檢查檔案是否存在;
os.R_OK: 檢查檔案是否可讀;
os.W_OK: 檢查檔案是否可以寫入;
os.X_OK: 檢查檔案是否可以執行
該方法通過判斷檔案路徑是否存在和各種訪問模式的許可權返回True或者False。
import os
if os.access("/file/path/foo.txt", os.F_OK):
print "Given file path is exist."
if os.access("/file/path/foo.txt", os.R_OK):
print "File is accessible to read"
if os.access("/file/path/foo.txt", os.W_OK):
print "File is accessible to write"
if os.access("/file/path/foo.txt", os.X_OK):
print "File is accessible to execute"
2.使用Try語句
可以在程式中直接使用open()方法來檢查檔案是否存在和可讀寫。
語法:
open(<file/path>)
如果你open的檔案不存在,程式會丟擲錯誤,使用try語句來捕獲這個錯誤。
程式無法訪問檔案,可能有很多原因:
如果你open的檔案不存在,將丟擲一個FileNotFoundError的異常;
檔案存在,但是沒有許可權訪問,會丟擲一個PersmissionError的異常。
所以可以使用下面的程式碼來判斷檔案是否存在:
try:
f =open()
f.close()
except FileNotFoundError:
print "File is not found."
except PersmissionError:
print "You don`t have permission to access this file."
其實沒有必要去這麼細緻的處理每個異常,上面的這兩個異常都是IOError的子類。所以可以將程式簡化一下:
try:
f =open()
f.close()
except IOError:
print "File is not accessible."
使用try語句進行判斷,處理所有異常非常簡單和優雅的。而且相比其他不需要引入其他外部模組。
3. 使用pathlib模組
pathlib模組在Python3版本中是內建模組,但是在Python2中是需要單獨安裝三方模組。
使用pathlib需要先使用檔案路徑來建立path物件。此路徑可以是檔名或目錄路徑。
檢查路徑是否存在
path = pathlib.Path("path/file")
path.exist()
檢查路徑是否是檔案
path = pathlib.Path("path/file")
path.is_file()