From 9a7da1ef1b81736a3c72251ec456083d00ed26cd Mon Sep 17 00:00:00 2001 From: HsiangNianian Date: Sat, 2 Aug 2025 11:38:42 +0800 Subject: chore(examples): add rule packs examples --- examples/main.py | 77 ++++++++++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 77 insertions(+) create mode 100644 examples/main.py (limited to 'examples/main.py') diff --git a/examples/main.py b/examples/main.py new file mode 100644 index 00000000..e7dffc1c --- /dev/null +++ b/examples/main.py @@ -0,0 +1,77 @@ +# main.py +import os +import importlib + +# --- 配置区 --- +# 在这里指定你想要加载的规则包文件名 (不含 .py) +# 你可以随时修改这个列表来开启或关闭不同的规则包 +ENABLED_RULE_PACKS = [ + "basic_roller", + "coc", +] + + +class DiceBotCore: + def __init__(self, rule_pack_names): + self.loaded_packs = [] + self.load_rule_packs(rule_pack_names) + + def load_rule_packs(self, rule_pack_names): + """动态加载指定的规则包""" + print("正在加载规则包...") + for name in rule_pack_names: + try: + # 动态导入模块 + module = importlib.import_module(f"rule_packs.{name}") + # 检查模块是否符合我们的结构要求 + if hasattr(module, "handle_command"): + self.loaded_packs.append(module) + print(f" - 规则包 '{name}' 加载成功。") + else: + print( + f" - 警告: 规则包 '{name}'缺少 'handle_command' 函数,已跳过。" + ) + except ImportError: + print(f" - 错误: 找不到名为 '{name}' 的规则包。") + print("-" * 20) + + def process_command(self, command): + """将指令传递给所有加载的规则包""" + for pack in self.loaded_packs: + result = pack.handle_command(command) + if result is not None: + return result # 一旦有规则包处理了指令,就立即返回结果 + return "无法识别的指令。输入 'help' 查看可用指令。" + + def get_all_help(self): + """收集所有已加载规则包的帮助信息""" + help_texts = ["--- 欢迎使用模块化骰子机器人 ---"] + for pack in self.loaded_packs: + if hasattr(pack, "get_help"): + help_texts.append(pack.get_help()) + return "\n".join(help_texts) + + +def main(): + """程序主循环""" + bot = DiceBotCore(ENABLED_RULE_PACKS) + + print("机器人已启动。输入 'exit' 退出。") + print(bot.get_all_help()) # 启动时显示帮助信息 + + while True: + user_input = input("\n> ") + + if user_input.lower() == "exit": + print("感谢使用,再见!") + break + elif user_input.lower() == "help": + print(bot.get_all_help()) + continue + + result = bot.process_command(user_input) + print(result) + + +if __name__ == "__main__": + main() -- cgit v1.2.3-70-g09d2