46 lines
1.2 KiB
Python
46 lines
1.2 KiB
Python
import os
|
|
import sys
|
|
|
|
from dotbot.plugin import Plugin
|
|
|
|
# We keep references to loaded modules so they don't get garbage collected.
|
|
loaded_modules = []
|
|
|
|
|
|
def load(path):
|
|
basename = os.path.basename(path)
|
|
module_name, extension = os.path.splitext(basename)
|
|
loaded_module = load_module(module_name, path)
|
|
plugins = []
|
|
for name in dir(loaded_module):
|
|
possible_plugin = getattr(loaded_module, name)
|
|
try:
|
|
if issubclass(possible_plugin, Plugin) and possible_plugin is not Plugin:
|
|
plugins.append(possible_plugin)
|
|
except TypeError:
|
|
pass
|
|
loaded_modules.append(loaded_module)
|
|
return plugins
|
|
|
|
|
|
if sys.version_info >= (3, 5):
|
|
import importlib.util
|
|
|
|
def load_module(module_name, path):
|
|
spec = importlib.util.spec_from_file_location(module_name, path)
|
|
module = importlib.util.module_from_spec(spec)
|
|
spec.loader.exec_module(module)
|
|
return module
|
|
|
|
elif sys.version_info >= (3, 3):
|
|
from importlib.machinery import SourceFileLoader
|
|
|
|
def load_module(module_name, path):
|
|
return SourceFileLoader(module_name, path).load_module()
|
|
|
|
else:
|
|
import imp
|
|
|
|
def load_module(module_name, path):
|
|
return imp.load_source(module_name, path)
|