Jak znaleźć wszystkie moduły potomne w Pythonie?

Question

Podczas gdy w Pythonie dość banalne jest importowanie modułu "dziecko" do innego modułu i wyświetlanie jego atrybutów, staje się nieco trudniejsze, gdy chcemy zaimportować wszystkie moduły potomne.

Buduję bibliotekę narzędzi do istniejącej aplikacji 3D. Każde narzędzie ma swój własny element menu i podmenu. Chciałbym, aby narzędzie było odpowiedzialne za tworzenie własnych menu, ponieważ wiele z nich zmienia się w zależności od kontekstu i szablonów. Chciałbym, żeby mój moduł bazowy mógł znaleźć wszystkie moduły potomne i sprawdzić funkcję create_menu() i wywołać ją, jeśli ją znajdzie.

Jaki jest najprostszy sposób na odkrycie wszystkich modułów potomnych?

Answer 1

użyciu dir() i imp module

Answer 2

Kiedy byłem miły i programowania dopiero zaczynają w Pythonie Pisałem to dla mojego modułowym bot IRC:

    # Load plugins 

    _plugins = [] 

    def ifName(name): 
     try: 
      return re.match('([^_.].+)\.[^.]+', a).group(1) 
     except: 
      return None 

    def isValidPlugin(obj): 
     from common.base import PluginBase 
     try: 
      if obj.__base__ == PluginBase: 
       return True 
      else: 
       return False 
     except: 
      return False 

    plugin_names = set(ifilter(lambda a: a!=None, [ifName(a) for a in os.listdir(os.path.join(os.getcwd(), 'plugins'))])) 
    for plugin_name in plugin_names: 
     try: 
      plugin = __import__('plugins.'+plugin_name, fromlist=['plugins']) 
      valid_plugins = filter(lambda a: isValidPlugin(a), [plugin.__getattribute__(a) for a in dir(plugin)]) 
      _plugins.extend(valid_plugins) 
     except Exception, e: 
      logger.exception('Error loading plugin %s', plugin_name) 

    # Run plugins 

    _plugins = [klass() for klass in _plugins] 

To nie jest bezpieczne lub „prawo” sposób, ale może jednak będziemy przydatni. To jest stary kod very, więc proszę mnie nie bić.

Answer 3

Można spróbować glob bing katalogu:

import os 
import glob 

modules = glob.glob(os.path.join('/some/path/to/modules', '*.py')) 

Następnie można spróbować ich importowania:

checked_modules 
for module in modules: 
    try: 
     __import__(module, globals(), locals()) # returns module object 
    except ImportError: 
     pass 
    else: 
     checked_modules.append(module) 

Answer 4

Myślę, że najlepszym sposobem, aby robić takie rzeczy pluginu korzysta entry_points oraz API for querying them.

Answer 5

Rozwiązanie powyżej przechodzące przez system plików do znajdowania submodułów jest w porządku, o ile zaimplementujesz każdą wtyczkę jako moduł oparty na systemie plików.

Bardziej elastycznym sposobem jest jawna lista wtyczek w głównym module i każda wtyczka (niezależnie czy jest to moduł utworzony przez plik, dynamicznie, czy nawet wystąpienie klasy), dodając się bezpośrednio do tej listy. Może poprzez funkcję registerPlugin.

Pamiętaj: "wyraźna jest lepsza niż niejawna" jest częścią zenu Pythona.