Python：根据用户想要的文件导入文件[重复]

Question

我有以下目录结构

+ code
|
--+ plugins
  |
  -- __init__.py
  -- test_plugin.py (has a class TestPlugin)
  -- another_test_plugin.py (has a class AnotherTestPlugin)
--+ load.py
--+ __init__.py

满载。py，我希望能够仅初始化用户指定的那些类。例如，假设我做了如下事情

$ python load.py -c test_plugin # Should only import test_plugin.py and initialize an object of the TestPlugin class

我在尝试使用“imp”模块时遇到问题。它不断地说“没有这样的文件或目录”。我的理解是，它不知何故没有正确理解路径。有人能帮我解决这个问题吗？

Answer 1

exec('import ' + sys.argv[2])
obj = test_plugin.TestPlugin()

这里sys.argv[2]是来自命令行参数的test_plugin字符串。

编辑：避免使用exec的另一种方法：

import importlib
mod = importlib.import_module(sys.argv[2])

Answer 2

另一个解决方案，如果你想要一个“插件”发现工具：

import imp, os
import glob

def load_plugins(path):
  """
  Assuming `path` is the only directory in which you store your plugins,
  and assuming each name follows the syntax:
  plugin_file.py -> PluginFile
  Please note that we don't import files starting with an underscore.

  """
  plugins = {}
  plugin_files = glob.glob(path + os.sep + r'[!_]*.py')
  for plugin_path in plugin_files:
    module_name, ext = os.path.splitext(plugin_path)
    module_name = os.path.basename(module_name)
    class_name = module_name.title().replace('_', '')
    loaded_module = imp.load_source(class_name, plugin_path) # we import the plugin
    plugins[module_name] = getattr(loaded_module, class_name)
  return plugins

plugins = load_plugins(your_path_here)
plugin_name = sys.argv[3]
plugin = plugins.get(plugin_name)
if not plugin:
    # manage a not existing plugin
else:
    plugin_instance = plugin() # creates an instance of your plugin

这样，您还可以通过更改密钥来指定不同的名称，例如“test_plugins”=

Answer 3

你要做的事情是：

import imp, os, plugins

path = os.path.dirname(plugins.__file__)
imp.load_source('TestPlugin', os.path.join(path, 'test_plugin.py')

插件是包含所有插件的模块（即空的__init__. py），这将帮助您获得插件模块文件的完整路径。