命令行工具在日常开发、运维、数据处理中的作用无需赘言。但随着功能复杂化,如何组织好命令结构、模块划分、便捷扩展 成了痛点:
-
❌ 各种脚本散落无法统一调用
-
❌ 想增加子命令必须修改主逻辑
-
❌ 不易模块化拆分和分组
-
❌ 命令帮助信息难维护
于是我们设计了一个 插件式命令行引擎(CLI Plugin Engine),具备以下特性:
✅ 支持插件注册与分组展示
✅ 插件为独立
.py
文件或目录模块✅ 支持自动加载、无需手动引入
✅ 支持动态参数解析、自动 help
✅ 主程序不动,插件任意增删
一、功能目标与适用场景
✨ 功能亮点:
特性 | 说明 |
---|---|
插件注册机制 | 每个命令为独立模块,注册方式统一 |
动态命令加载 | 支持按目录加载所有命令插件 |
参数解析 | 使用 argparse 或 click 自动解析 |
分组调用 | 支持 tool data:clean 或 tool dev:run 风格 |
自动帮助 | 自动生成 help 信息与参数说明 |
热插拔 | 插件增删不影响主程序 |
✅ 适用场景:
-
团队内部 CLI 工具管理平台
-
Python 自动化命令脚本集成
-
私有云/容器/部署命令工具
-
数据流程构建工具的命令入口
二、项目结构设计
csharp
复制编辑
cli_engine/ ├── cli.py # 主入口脚本 ├── engine/ │ ├── loader.py # 插件加载器 │ └── base.py # 插件基类定义 └── plugins/ ├── data_clean.py # 示例插件1 ├── dev_server.py # 示例插件2 └── utils_convert.py # 示例插件3
三、插件定义规范(基类)
我们定义所有插件需继承 PluginCommand
:
python
复制编辑
# engine/base.py class PluginCommand: name = "unnamed" group = "default" description = "" def add_arguments(self, parser): pass def run(self, args): raise NotImplementedError
四、插件加载器 loader.py
python
复制编辑
# engine/loader.py import importlib.util import os from engine.base import PluginCommand def discover_plugins(plugin_dir): commands = [] for fname in os.listdir(plugin_dir): if fname.endswith(".py") and not fname.startswith("__"): path = os.path.join(plugin_dir, fname) spec = importlib.util.spec_from_file_location(fname[:-3], path) mod = importlib.util.module_from_spec(spec) spec.loader.exec_module(mod) for attr in dir(mod): cls = getattr(mod, attr) if isinstance(cls, type) and issubclass(cls, PluginCommand) and cls != PluginCommand: commands.append(cls()) return commands
五、主入口 cli.py
python
复制编辑
# cli.py import argparse from engine.loader import discover_plugins plugin_dir = "plugins" commands = discover_plugins(plugin_dir) def group_by(commands): groups = {} for cmd in commands: groups.setdefault(cmd.group, []).append(cmd) return groups def main(): parser = argparse.ArgumentParser(description="🔧 CLI 插件引擎") subparsers = parser.add_subparsers(dest="command") cmd_map = {} for cmd in commands: subparser = subparsers.add_parser(f"{cmd.group}:{cmd.name}", help=cmd.description) cmd.add_arguments(subparser) cmd_map[f"{cmd.group}:{cmd.name}"] = cmd args = parser.parse_args() if args.command in cmd_map: cmd_map[args.command].run(args) else: parser.print_help() if __name__ == "__main__": main()
六、示例插件 plugins/data_clean.py
python
复制编辑
from engine.base import PluginCommand import pandas as pd class DataCleanCommand(PluginCommand): name = "clean" group = "data" description = "清洗 CSV 文件(去空行、去重)" def add_arguments(self, parser): parser.add_argument("infile", help="输入 CSV 文件") parser.add_argument("--dropna", action="store_true", help="是否去除空行") parser.add_argument("--dedup", action="store_true", help="是否去除重复行") def run(self, args): df = pd.read_csv(args.infile) if args.dropna: df = df.dropna() if args.dedup: df = df.drop_duplicates() outfile = args.infile.replace(".csv", "_cleaned.csv") df.to_csv(outfile, index=False) print(f"✅ 已输出清洗后文件:{outfile}")
七、示例插件 plugins/dev_server.py
python
复制编辑
from engine.base import PluginCommand import os class DevRunCommand(PluginCommand): name = "run" group = "dev" description = "启动开发服务器" def add_arguments(self, parser): parser.add_argument("--port", default=8080, type=int, help="端口号") def run(self, args): os.system(f"python -m http.server {args.port}")
八、使用示例
运行帮助:
bash
复制编辑
$ python cli.py --help usage: cli.py [-h] {data:clean,dev:run} ... options: -h, --help show this help message and exit commands: data:clean 清洗 CSV 文件(去空行、去重) dev:run 启动开发服务器
执行命令:
bash
复制编辑
# 执行 CSV 清洗任务 python cli.py data:clean data.csv --dropna --dedup # 启动服务器 python cli.py dev:run --port 8888
九、功能增强建议
模块 | 扩展功能 |
---|---|
自动帮助生成 | 支持 tool list 显示所有插件 |
参数提示 | 集成自动补全(支持 argcomplete ) |
插件分组折叠 | 显示分组及子命令层级结构 |
插件热加载 | 实时监听插件目录,新增插件即生效 |
任务装饰器 | 插件中通过装饰器注册命令参数及元信息 |
插件元数据文件 | 支持插件携带 meta.json 提供描述、作者、版本等信息 |
十、适用场景
场景 | 说明 |
---|---|
内部运维命令平台 | 每位成员开发模块后可独立提交命令插件 |
数据自动化流水线 | 各阶段流程(导入 → 清洗 → 统计)可插件化组织 |
脚本分发平台 | 项目脚本封装成插件,避免脚本文件散落 |
工具平台脚手架 | 类似 vue-cli / ng-cli 的 Python 命令平台基础架构 |