導(dǎo)入父文件夾中的模塊并讀取當(dāng)前文件夾內(nèi)的資源
在某些特殊情況下,我們的 Python 腳本需要調(diào)用父目錄下的其他模塊。例如:
在編寫 GNE 的測試用例時,有一個腳本 generate_new_cases.py放在 tests文件夾中。而 tests 文件夾與 gne 文件夾放在同一個位置。其中 gne 文件夾是一個包。我現(xiàn)在需要從generate_new_cases.py 文件中導(dǎo)入 gne 里面的一個類GeneralNewsExtractor。
為了簡化問題,我單獨寫了一個演示的樣例。它的文件結(jié)構(gòu)與每個文件中的內(nèi)容如下:
現(xiàn)在,我直接在 scripts 文件夾里面運行run.py會報錯,提示從包的最頂層之外相對導(dǎo)入。
現(xiàn)在,我們改一下代碼,嘗試在scripts 文件夾的父文件夾中運行代碼,發(fā)現(xiàn)還是會報錯:
我們再來改一下代碼,在 scripts 文件夾中,把父文件夾加入到sys.path中:
- import sys
- sys.path.append('..')
運行效果如下圖所示:
這種寫法,當(dāng)我們在scripts 文件夾里面運行run.py時,確實已經(jīng)正常了。但是如果我們在scripts文件夾的父文件夾再運行代碼,又不對了,如下圖所示:
為了檢查這個原因,我們打印一下sys.path:
大家有沒有發(fā)現(xiàn)一個很不和諧的東西:其他路徑都是絕對路徑,就最后我們加了兩個點看起來很違和。那如果我們把這兩個點換成絕對路徑呢?于是嘗試獲取當(dāng)前正在運行的這個文件的絕對路徑:
- import sys
- from pathlib import Path
- current_folder = Path(__file__).absolute().parent
- father_folder = str(current_folder.parent)
- sys.path.append(father_folder)
運行效果如下圖所示:
導(dǎo)入模塊已經(jīng)正常了,但是讀取資源文件又異常了。
這是因為,import導(dǎo)入模塊時,是根據(jù)sys.path中的路徑來尋找的。但是讀取資源文件的時候,相對文件路徑是相對于工作區(qū)來尋找的。
現(xiàn)在我們在~/test_import_father_module執(zhí)行python3 scripts/run.py,那么當(dāng)前的工作區(qū)就是~/test_import_father_module。而由于資源文件是在scripts文件夾中的,所以就找不到。
所以我們還需要修改一下工作區(qū):
- import os
- from pathlib import Path
- current_folder = Path(__file__).absolute().parent
- os.chdir(str(current_folder))
運行效果如下圖所示:
現(xiàn)在無論是讀取資源文件還是導(dǎo)入模塊,都已經(jīng)正常了。
我們再回到 scripts 文件夾中執(zhí)行看看:
發(fā)現(xiàn)也能正常執(zhí)行。
總結(jié)涉及到模塊導(dǎo)入相關(guān)的環(huán)境,可以通過在sys.path添加絕對路徑來解決。涉及到讀取資源文件的相關(guān)環(huán)境,可以通過使用os.chdir修改工作區(qū)為另一個絕對路徑來解決。