ykxmz/_build/fix_brute_order.py

# -*- coding: utf-8 -*-
"""把爆破测试真正挪到成功配对之前。

上一版插错了位置 —— 插在 check("配对成功") 之后，
那时配对窗口已经因为首台设备配对成功而自动关闭，
所以每次都是 "被控端已关闭新设备配对"，根本测不到限速。
"""
from __future__ import annotations

import sys
from pathlib import Path

E2E = Path(r"C:\Users\shu\Desktop\deepseekagent\LanRemote\_tests"
           r"\test_e2e_features.py")

BLOCK_START = '        print()\n        print("一b、★ 配对防爆破'
BLOCK_END = '        print()\n        print("二、★ 功能列表'


def main() -> int:
    text = E2E.read_text(encoding="utf-8")

    # 先把已经插错位置的那段摘出来
    start = text.find(BLOCK_START)
    if start < 0:
        print("  [失败] 找不到爆破测试段")
        return 1
    end = text.find(BLOCK_END, start)
    if end < 0:
        print("  [失败] 找不到爆破测试段的结尾")
        return 1
    block = text[start:end]
    text = text[:start] + text[end:]

    # 再插到"配对"之前 —— 找到 client = Client(...) 那一行
    anchor = "        client = Client(\"127.0.0.1\", PORT)"
    if anchor not in text:
        print("  [失败] 找不到 client 创建处")
        return 1
    insert_at = text.index(anchor)
    # 插在 client = Client(...) 这行的后面
    line_end = text.index("\n", insert_at) + 1
    text = text[:line_end] + "\n" + block + text[line_end:]

    E2E.write_text(text, encoding="utf-8")
    print("  爆破测试已挪到配对之前")

    import ast
    try:
        ast.parse(E2E.read_text(encoding="utf-8"))
        print("  语法 OK")
    except SyntaxError as exc:
        print(f"  语法错误: {exc}")
        return 1

    # 打印新的阶段顺序
    print()
    for i, line in enumerate(E2E.read_text(encoding="utf-8").splitlines()):
        if 'print("' in line and ("、" in line):
            print(f"  {line.strip()}")
    return 0


if __name__ == "__main__":
    sys.exit(main())