一、原理

1)sys.path 添加环境变量目录

sys.path.append(<dir_name>)

2)__import__ 函数导入模块

_module = __import__(<module_name>)

3)getattr 获取类

getattr(<module_name>, <class_name>)

注:如果需要实例化类的话,后面加个括号:getattr(<module_name>, <class_name>)()

 

二、实例

说明:加载 tasks 目录里的所以类,并执行 run 方法

import sys
import pathlib
from tools import istring

sys.path.append('tasks')


def run(task):
    items = []
    if task != '':
        _module = __import__(task)
        _class = istring.ucwords(task)
        items.append(getattr(_module, _class)())
    else:
        for x in pathlib.Path('tasks').iterdir():
            if x.name == '__pycache__':
                continue
            _module = __import__(x.stem)
            _class = istring.ucwords(x.stem)
            items.append(getattr(_module, _class)())

    for task in items:
        task.run()


if __name__ == '__main__':
    task = sys.argv[1] if len(sys.argv) == 2 else ''
    run(task)

注:istring 是我自己写的助手类,这个需要按自己的类命名规则处理一下 ~



完。