utils.py

import sys
import importlib
import os
from . import logger


def load_module(module, path):
    files = os.listdir(path)
    for file in files:
        try:
            if file.endswith('.py') or file.endswith('.pyc'):
                filename = file.replace('.pyc', '').replace('.py', '')
                # if 'login.py' == file:
                #     logger.info(f"加载成功  {filename}")
                #     continue
                name = "jbot.{}.{}".format(module, filename)
                spec = importlib.util.spec_from_file_location(name, path+file)
                load = importlib.util.module_from_spec(spec)
                spec.loader.exec_module(load)
                sys.modules[f"jbot.{module}.{filename}"] = load
                logger.info(f"加载成功  {filename}")
        except Exception as e:
            logger.info(f"加载失败  {file}  {str(e)}")
            continue