Làm thế nào để tìm tất cả các mô-đun con trong Python?

Question

Mặc dù nó khá tầm thường trong Python để nhập mô-đun "con" vào mô-đun khác và liệt kê các thuộc tính của nó, nó trở nên khó khăn hơn khi bạn muốn nhập tất cả các mô-đun con.

Tôi đang tạo thư viện công cụ cho ứng dụng 3D hiện có. Mỗi công cụ có mục menu và menu phụ riêng. Tôi muốn công cụ chịu trách nhiệm tạo các menu của riêng mình vì nhiều công cụ thay đổi dựa trên ngữ cảnh và các mẫu. Tôi muốn mô-đun cơ bản của tôi có thể tìm thấy tất cả các mô-đun con và kiểm tra chức năng create_menu() và gọi nó nếu nó tìm thấy nó.

Cách dễ nhất để khám phá tất cả các mô-đun con là gì?

Answer 1

sử dụng dir() và imp module

Answer 2

Khi tôi còn là một loại và lập trình chỉ mới bắt đầu bằng Python Tôi đã viết này cho bot IRC mô-đun của tôi:

    # 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] 

Nó không an toàn hay "đúng" cách, nhưng có lẽ nó sẽ hữu ích. Đó là rất mã cũ vì vậy xin đừng đánh tôi.

Answer 3

Bạn có thể thử glob bing thư mục:

import os 
import glob 

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

Sau đó, bạn có thể thử nhập họ:

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

Answer 4

Tôi nghĩ rằng cách tốt nhất để làm điều này loại plugin được sử dụng entry_points và API for querying them.

Answer 5

Giải pháp trên vượt qua hệ thống tệp để tìm mô-đun con là ok miễn là bạn triển khai mọi plugin làm mô-đun dựa trên hệ thống tệp.

Một cách linh hoạt hơn sẽ là danh sách plugin rõ ràng trong mô-đun chính của bạn và có mọi plugin (cho dù mô-đun được tạo bởi tệp, động hoặc thậm chí là một lớp) tự thêm vào danh sách đó một cách rõ ràng. Có thể thông qua chức năng registerPlugin.

Hãy nhớ rằng: "rõ ràng là tốt hơn là ẩn" là một phần của zen của python.