78 lines
3.5 KiB
Python
78 lines
3.5 KiB
Python
"""微信机器人框架的统一入口：python main.py。"""
|
|
from __future__ import annotations
|
|
|
|
import argparse
|
|
import json
|
|
import logging
|
|
import math
|
|
from logging.handlers import RotatingFileHandler
|
|
from pathlib import Path
|
|
import sys
|
|
|
|
from wxbot_framework.config import ConfigStore
|
|
from wxbot_framework.locking import InstanceLock
|
|
from wxbot_framework.runtime import BotApplication
|
|
|
|
ROOT = Path(__file__).resolve().parent
|
|
|
|
|
|
def configure_logging(config):
|
|
directory = ROOT / "logs"
|
|
directory.mkdir(exist_ok=True)
|
|
formatter = logging.Formatter("%(asctime)s %(levelname)s [%(name)s] %(message)s")
|
|
console = logging.StreamHandler()
|
|
console.setFormatter(formatter)
|
|
file = RotatingFileHandler(directory / "bot.log", maxBytes=2 * 1024 * 1024,
|
|
backupCount=3, encoding="utf-8")
|
|
file.setFormatter(formatter)
|
|
logging.basicConfig(level=config.get("log_level", "INFO").upper(), handlers=[console, file], force=True)
|
|
|
|
|
|
def main():
|
|
if hasattr(sys.stdout, "reconfigure"):
|
|
sys.stdout.reconfigure(encoding="utf-8")
|
|
sys.stderr.reconfigure(encoding="utf-8")
|
|
parser = argparse.ArgumentParser(description=__doc__)
|
|
parser.add_argument("--config", type=Path, default=ROOT / "bot_config.json")
|
|
parser.add_argument("--check", action="store_true", help="检查配置、清单及启用插件语法，不注入微信")
|
|
parser.add_argument("--list-plugins", action="store_true", help="列出可用插件，不注入微信")
|
|
parser.add_argument("--run-seconds", type=float, default=0, help="有限时长检查；默认 0 持续运行")
|
|
args = parser.parse_args()
|
|
if not math.isfinite(args.run_seconds) or args.run_seconds < 0:
|
|
parser.error("运行时长必须是非负有限数值")
|
|
try:
|
|
store = ConfigStore(args.config)
|
|
config = store.load()
|
|
if not (args.check or args.list_plugins) and config.get("wechat_account") == "wxid_example_robot":
|
|
parser.error("请先在本地 bot_config.json 中将 wechat_account 占位符改为自己的机器人账号 ID")
|
|
app = BotApplication(ROOT, store)
|
|
if args.check or args.list_plugins:
|
|
specs = app.plugins.discover()
|
|
if app.plugins.load_errors:
|
|
raise ValueError(str(app.plugins.load_errors))
|
|
for spec in specs.values():
|
|
enabled = config.get("plugins", {}).get(spec.name, {}).get("enabled", True)
|
|
if args.check and enabled:
|
|
for source in spec.path.parent.rglob("*.py"):
|
|
compile(source.read_bytes(), str(source), "exec")
|
|
if spec.name in ("idiom_chain", "chat_summary", "mention_chat"):
|
|
from wxbot_framework.llm import LLMClient
|
|
LLMClient.from_file(store.path.parent / config.get("llm_config_file", "llm_config.json"))
|
|
print(json.dumps({"name": spec.name, "enabled": enabled, "api": spec.api,
|
|
"description": spec.description,
|
|
"missing_capabilities": sorted(spec.requires - app.plugins.capabilities)}, ensure_ascii=False))
|
|
if args.check:
|
|
print("配置与插件语法检查通过")
|
|
return 0
|
|
configure_logging(config)
|
|
with InstanceLock(ROOT / ".runtime/bot.lock"):
|
|
app.run(seconds=args.run_seconds)
|
|
return 0
|
|
except Exception:
|
|
logging.getLogger("Main").exception("启动或运行失败")
|
|
return 1
|
|
|
|
|
|
if __name__ == "__main__":
|
|
raise SystemExit(main())
|