各位好,以下是本人要處理的檔案
main.py
containers
/module1.py
/module2.py
/module3.py
module1, 2, 3中的程式碼類似,皆如下:
class Test1(self):
def __init__(self, num):
self.num = num
def test1(self):
print("test1")
我現在想在main.py中import所有在module1,2,3中的class,但是如果寫成
from module1 import *
from module2 import *
from module3 import *
好像不太好,之後在containers裡還會增加很多module
不知道各位版友有沒有什麼好方法?
我查過一些資料,像這篇裡有些人去修改__all__,或是使用pkgutil
https://reurl.cc/AKNxqj
但我都沒有成功
我嘗試過用importlib
for name in os.listdir("containers"):
if name.endswith(".py"):
module = name[:-3]
importlib.import_module(module)
但還是沒有辦法將所有module匯入
import_module 好像只能這樣用:
x = importlib.import_module(module1)
x.test1.test1()
還有看過一個方法是:
for name in os.listdir("containers"):
if name.endswith(".py"):
module = name[:-3]
test_spec = importlib.util.spec_from_file_location(module, "containers/" + module + ".py")
test_module = importlib.util.module_from_spec(test_spec)
test_spec.loader.exec_module(test_module)
我以為執行完上面這幾行後,可以直接用Test1.test1(),程式裡還是找不到Test1