snowy-yang
f8e093c98f
三件事凑成一次不兼容变更：WebSocket 通道、Hypercorn 服务器、「手动推进一天」都消失了。 实时通道：WebSocket -> SSE - hub 改为 SSE 订阅池，GET /api/stream 单向推送 welcome/state/day/notice/bye， 250ms 节流 + 15s 心跳注释；写操作一律走 HTTP POST /api/actions，不再有 WS 回执。 - 停机三处一起修才干净：close_all() 发 bye + timeout_graceful_shutdown 兜底 + 客户端 stop() 直接 shutdown 底层 socket（光 close() 会等读线程的锁，最长卡到下次心跳）。 - 网页版改 EventSource + fetch；tools/ws_demo.py 换成 tools/sse_demo.py。 服务器：Hypercorn -> uvicorn + zttp 的 HTTP/2 - 同一端口同时说 HTTP/1.1 与 HTTP/2。uvicorn 0.53 不再内置 h2，必须装 zttp 并显式 http="zttp"，否则 Config.load() 直接 raise。 - 信号接管改成子类覆盖 Server.handle_exit（0.53 没有 install_signal_handlers 可覆盖）， 不调 super()、不写 _captured_signals，退出时不重放信号；第二次信号强制退出。 - 入口启动前用 daemon.port_busy() 自检端口：Windows 上 SO_REUSEADDR 会让第二个实例 「绑上」已监听端口并抢走新连接，所以判断「能不能连上」而不是「能不能 bind」。 - 修掉 clear_pid_file 的 Windows 真 bug：文件还开着就 os.remove 会失败且被 except OSError 吞掉，导致每次正常退出都留 stale pid 文件。 节奏：一天 15 秒 -> 40 分钟，并整体移除「休息」 - DEFAULT_TICK_SECONDS=2400，MAX_TICK_SECONDS 放宽到 86400（旧上限会把 40 分钟夹回去）。 - engine.sleep / SLEEP_COOLDOWN / /api/sleep / TUI 的 [N]休息 / 网页版按钮全部删掉， 时间只能自然流动；机器人改为 wait_for_next_day(timeout)，CLI 加 bot --wait。 - 时长显示统一走 models.format_duration()，不再出现 2399.6s 这种写法。 键盘：修复方向键/空格/Tab/ESC 全部静默失效 - keys.py 的具名键统一小写并成为唯一来源，tui.py 一律引用 keys.XXX 而不手写字符串 （历史 bug 就是 Windows 侧产大写 UP、TUI 里写小写 up，字母键正常所以很难发现）。 - 顺带修 POSIX 路径一次 os.read 只取首字节、其余按键静默丢弃的老毛病。 测试：重建 500 项 - test_engine 110 / test_actions 35 / test_server 131(skip 35) / test_tui_keys 75 / test_bot 36 / test_client 113。未安装的第三方依赖用 skipUnless 自动跳过，其余照跑， 所以是全绿而不是靠跳过伪装。 - 被测试逼出两处真隐患并修掉：RemoteFarm._normalize() 必须「先补 scheme 再削尾斜杠」 （顺序反了畸形端点会溜到系统代理、报一个和真实原因无关的 502）；StateView 空视图要 预置 plots/animals，否则连不上服务器时读 state.plots 直接 AttributeError。 优化：先量后改，依据与前后数字都在 tools/bench.py - 新增 engine.brief_source()：brief/summary/动作回执只要十几个标量，不再为它们序列化 整局；to_dict 里 net_worth 只算一次。brief -76%、summary -77%、leaderboard(32) -78%、 GET /api/world -31%、POST /api/actions?full=1 -16%。 - 拆掉三个巨型函数：create_app 475->32 行（_Services + _register_*）、 build_parser 与 available_actions 分组、__main__.main 160->7 行。 - 消除重复：等级门槛 5 处 -> _is_locked/_require_level/_newly_unlocked；hub 的 broadcast/broadcast_all/close_all -> _push_many；daemon.probe/world -> _get_json； main.py 的 _add_endpoint_args/_leaderboard_lines。 - tools/bench.py 作为长期性能守卫（含调用计数器），谁把轻量路径改回 snapshot() 会立刻报警。 门禁：ruff format/check 全绿、pyright 0 错误、500 项测试 OK（skip 35）。
90 lines
3.2 KiB
Python
90 lines
3.2 KiB
Python
#!/usr/bin/env python3
|
|
"""外挂式 AI/机器人玩家——完全通过 HTTP + SSE 接口参与游戏。
|
|
|
|
这个脚本**不**导入任何游戏规则代码，只用接口。换句话说：
|
|
它对服务器来说和一个远程 LLM Agent 没有任何区别。
|
|
|
|
用法
|
|
----
|
|
uv run ai_player.py # 连本机 default 牧场
|
|
uv run ai_player.py --farm ai-1 --minutes 5 # 经营 ai-1 五分钟
|
|
uv run ai_player.py --url http://10.0.0.5:8765 --farm my-bot
|
|
|
|
策略在哪里
|
|
----------
|
|
决策逻辑在 :mod:`farm.bot` 的 ``RuleBot.score()``：那是一张"什么该先做"的
|
|
打分表。想换成真 LLM，只需要把「读 /api/actions 之后选哪个」这一步换成模型
|
|
调用——接口调用序列完全一样：
|
|
|
|
读 GET /api/actions → 挑一个 → POST /api/actions → 读回执里的 state
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
import argparse
|
|
import json
|
|
import sys
|
|
|
|
from farm.bot import RuleBot, random_bot
|
|
from farm.client import RemoteFarm
|
|
from farm.engine import GameError
|
|
|
|
|
|
def build_parser() -> argparse.ArgumentParser:
|
|
parser = argparse.ArgumentParser(
|
|
prog="uv run ai_player.py",
|
|
description="通过 HTTP + SSE 接口参与 SimpleFarm 的机器人玩家",
|
|
)
|
|
parser.add_argument("--url", default="http://127.0.0.1:8765", help="服务器地址")
|
|
parser.add_argument("--farm", default="ai-1", help="经营哪个牧场，默认 ai-1")
|
|
parser.add_argument("--token", default=None, help="访问令牌（服务器启用时必填）")
|
|
parser.add_argument("--rounds", type=int, default=0, help="最多执行多少轮，0 表示不限")
|
|
parser.add_argument("--minutes", type=float, default=0.0, help="最多跑多少分钟，0 表示不限")
|
|
parser.add_argument("--pause", type=float, default=0.25, help="动作间隔（秒）")
|
|
parser.add_argument("--random", action="store_true", help="用随机策略（对照组）")
|
|
parser.add_argument("--quiet", action="store_true", help="只输出结果，不打过程")
|
|
return parser
|
|
|
|
|
|
def main(argv=None) -> int:
|
|
args = build_parser().parse_args(argv)
|
|
|
|
try:
|
|
client = RemoteFarm(
|
|
endpoint=args.url,
|
|
farm=args.farm,
|
|
token=args.token,
|
|
name=f"bot-{args.farm}",
|
|
realtime=False,
|
|
)
|
|
except GameError as exc:
|
|
print(f"✗ 连不上服务器：{exc.message}", file=sys.stderr)
|
|
print(" 先启动：uv run main.py serve --daemon", file=sys.stderr)
|
|
return 2
|
|
|
|
snapshot = client.snapshot()
|
|
print(
|
|
f"已连接 {args.url} · 牧场「{args.farm}」· 第 {snapshot.get('day')} 天 · "
|
|
f"金币 {snapshot.get('gold')}"
|
|
)
|
|
|
|
factory = random_bot if args.random else RuleBot
|
|
bot = factory(client, verbose=not args.quiet, pause=args.pause, name=args.farm)
|
|
try:
|
|
summary = bot.run(rounds=args.rounds, max_minutes=args.minutes)
|
|
except KeyboardInterrupt:
|
|
bot.stop()
|
|
summary = {
|
|
"name": args.farm,
|
|
"actions": bot.actions_executed,
|
|
"failures": bot.failures,
|
|
"interrupted": True,
|
|
}
|
|
|
|
print(json.dumps(summary, ensure_ascii=False, indent=2))
|
|
return 0
|
|
|
|
|
|
if __name__ == "__main__":
|
|
sys.exit(main())
|