# -*- coding: utf-8 -*-
r'''让管理页测试对"属性有没有引号"不敏感。

背景：
    旧版管理页的 HTML 是 `class=pick`（无引号），
    重做时改成了标准的 `class="pick"`（带引号）。
    测试里写死了 `"class=pick" in html`，于是集体误报失败。

    HTML 两种写法都合法，测试不该关心这个。
    所以给测试加个 _has()：两种写法都认。
'''
from __future__ import annotations

import ast
import re
import sys
from pathlib import Path

TESTS = Path(r"C:\Users\shu\Desktop\deepseekagent\LanRemote\_tests")

# 要改的测试文件和里面的断言
TARGETS = {
    "test_admin_batch_delete.py": [
        ('check("有复选框（input.pick）", "class=pick" in html)',
         'check("有复选框（input.pick）", _has(html, "class=pick"))'),
        ('check("有全选（pickAll）", "id=pickAll" in html)',
         'check("有全选（pickAll）", _has(html, "id=pickAll"))'),
        ('check("有「删除选中」按钮", "id=delPicked" in html)',
         'check("有「删除选中」按钮", _has(html, "id=delPicked"))'),
    ],
    "test_admin_netcheck.py": [
        ('check("有网络体检面板", \'id=netPanel\' in html, "")',
         'check("有网络体检面板", _has(html, "id=netPanel"), "")'),
        ('check("有「测一下」按钮", \'id=netBtn\' in html, "")',
         'check("有「测一下」按钮", _has(html, "id=netBtn"), "")'),
    ],
}

# 插到 check() 定义附近
HELPER = '''

def _has(html: str, token: str) -> bool:
    """判断 html 里有没有这个属性，**带不带引号都算**。

    旧版页面写的是 class=pick（无引号），
    重做后改成了标准的 class="pick"（带引号）。
    测试不该因为引号差异就报失败 —— HTML 两种都合法。
    """
    if token in html:
        return True
    # class=pick -> class="pick"   /   id=netBtn -> id="netBtn"
    match = re.match(r"^([A-Za-z-]+)=(.+)$", token)
    if not match:
        return False
    attr, value = match.group(1), match.group(2)
    return ('%s="%s"' % (attr, value)) in html
'''


def ensure_import_re(text: str) -> str:
    if re.search(r"^import re$", text, re.M):
        return text
    # 插到第一个 import 之后
    match = re.search(r"^(import \w+|from \w+ import .+)$", text, re.M)
    if match:
        return (text[:match.end()] + "\nimport re" + text[match.end():])
    return "import re\n" + text


def main() -> int:
    total = 0
    for name, replacements in TARGETS.items():
        path = TESTS / name
        if not path.is_file():
            print("  [跳过] %s 不存在" % name)
            continue
        text = path.read_text(encoding="utf-8")

        if "_has(" in text and "def _has" in text:
            print("  %s: 已经是新版" % name)
            continue

        changed = 0
        for old, new in replacements:
            if old in text:
                text = text.replace(old, new, 1)
                changed += 1
        if changed == 0:
            print("  %s: 没找到要改的断言" % name)
            continue

        text = ensure_import_re(text)

        # 把 _has 插到 "def check(" 前面
        marker = "def check("
        index = text.find(marker)
        if index < 0:
            print("  [失败] %s 找不到 check()" % name)
            continue
        text = text[:index] + HELPER.strip() + "\n\n\n" + text[index:]

        path.write_text(text, encoding="utf-8", newline="\n")
        try:
            ast.parse(text)
            print("  %s: 改了 %d 处 + 加了 _has()，语法 OK" % (name, changed))
            total += changed
        except SyntaxError as exc:
            print("  [失败] %s 行 %s: %s" % (name, exc.lineno, exc.msg))

    print()
    print("  共改 %d 处" % total)
    return 0


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