chongming-core/public/utils/launch.py

from .config import ConfigLoader
import os
from module_bank import PythonToSQLite
from typing import Callable
from pathlib import Path

# module_bank 依赖在每次加载 PyArmor 加密模块时都会 print 一行
# "已部署 PyArmor 运行时"，启动期会刷屏（约 50 行）。依赖包不便改动，
# 这里用模块级 print 遮蔽（模块内名字解析 globals 优先于 builtins）静默掉。
import module_bank.sqlite_module_loader as _sqlite_module_loader

_sqlite_module_loader.print = lambda *args, **kwargs: None


def init_module_bank(config):
    """
    初始化模块银行系统，将指定路径下的Python模块打包并安装导入器。

    该函数会验证所有模块路径的有效性，创建PythonToSQLite打包器实例，
    并为每个打包器安装导入器以支持模块的动态加载。

    Args:
        config (dict): 配置字典，必须包含 'production' -> 'module_system' -> 'path' 键，
                      其值为模块路径列表

    Raises:
        ValueError: 当配置的模块路径中有任何一个不存在时抛出异常
    """
    MODULE_BANK_PATH_LIST = config["production"]["module_system"]["path"]
    for module_path in MODULE_BANK_PATH_LIST:
        if not os.path.exists(module_path):
            raise ValueError(f"模块路径不存在: {module_path}")

    # 为每个模块路径创建PythonToSQLite打包器实例
    packer_list = [
        PythonToSQLite(module_path) for module_path in MODULE_BANK_PATH_LIST
    ]
    # 遍历所有打包器，安装导入器（模块清单不再打印，避免启动刷屏）
    for packer in packer_list:
        key_file_path = Path(packer.db_path).with_suffix(".key")
        with open(key_file_path, "r", encoding="utf-8") as f:
            key = f.read().strip()
        packer.install_importer(
            key=key,
            signature_key="my_key",
            signature_key_id="ci-2026",
            allow_unsigned=False,
        )


def launch(run_fun: Callable[[dict, dict], None], run_mode: str):
    """
    启动应用程序，加载配置并执行指定的运行函数。

    该函数从config.toml文件中加载配置，根据运行模式获取对应的应用配置和默认配置，
    然后调用传入的运行函数执行应用程序逻辑。

    Args:
        run_fun (Callable[[dict, dict], None]): 运行函数，接收两个参数：
                                               - app_config (dict): 当前运行模式的配置
                                               - default_config (dict): 默认配置
        run_mode (str): 运行模式，用于从配置文件中获取对应模式的配置项

    Returns:
        None
    """
    config = ConfigLoader("config.toml")

    # 获取对应运行模式的配置
    default_config = config.get("default", {})
    app_config = config.get(run_mode, {})

    run_fun(app_config, default_config)