Re: [問題] 如何import所有來自指定目錄的module

作者: blc (Anemos)   2022-03-29 18:35:37
我在推文給了錯誤方法,只好再想一個
containers/__init__.py:
import os
__all__ = [i.replace(".py","") for i in os.listdir(__path__[0]) ]
from . import *
main.py:
import containers
t1 = containers.module1.Test1(6)
t1.test1()
如果要去掉module1那層大概得找其他方法...
※ 引述《VivianAnn (薇薇安安)》之銘言:
: 各位好,以下是本人要處理的檔案
: 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
作者: VivianAnn (薇薇安安)   2022-03-30 02:30:00
感謝分享喔。有沒有人知道怎麼去掉module? 還有我在本地測試ok,用公司的機器就是出現找不到containers的錯誤訊息,不知問題在哪。晚點補充
作者: lycantrope (阿寬)   2022-03-30 12:34:00
https://reurl.cc/qOE4dgif inspect.isclass可以改動
作者: blc (Anemos)   2022-03-30 15:27:00
原來還有globals()這個東西,學到了。
作者: kenduest (小州)   2022-03-30 15:37:00
或是根本改架構,設計一個可以讓模組自動註冊的功能

Links booklink

Contact Us: admin [ a t ] ucptt.com