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