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

[blc (Anemos)]

我在推文給了錯誤方法，只好再想一個

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

--
※ 發信站: 批踢踢實業坊(ptt.cc), 來自: 158.39.201.239 (挪威)
※ 文章網址: https://www.ptt.cc/bbs/Python/M.1648550140.A.25A.html
推 VivianAnn: 感謝分享喔。有沒有人知道怎麼去掉module? 還有我在本
→ VivianAnn: 地測試ok，用公司的機器就是出現找不到containers的錯
→ VivianAnn: 誤訊息，不知問題在哪。晚點補充
推 lycantrope: https://reurl.cc/qOE4dg
→ lycantrope: if inspect.isclass可以改動
→ blc: 原來還有globals()這個東西，學到了。
→ kenduest: 或是根本改架構，設計一個可以讓模組自動註冊的功能