workbuddy-portal/tools/shots.py

#!/usr/bin/env python
# -*- coding: utf-8 -*-
"""界面实检：登录后逐页截图，用来目视确认「统一美化」是否真的落地。

用法：
    python manage.py serve --port 8849 --no-scheduler   # 另开一个终端
    python tools/shots.py --base http://127.0.0.1:8849
    python tools/shots.py --full                        # 整页长图（默认只截首屏）

产物：data/shots/*.png（已被 .gitignore 之外的目录，可直接删）。

为什么不用 headless chrome 直出：本项目的页面都要登录态，
`--screenshot` 无法注入会话 Cookie，所以必须用 Playwright 走一次真实登录。
"""
from __future__ import annotations

import argparse
import os
import sys

BASE = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
sys.path.insert(0, BASE)

PAGES = [
    ("login", "/login", "登录页"),
    ("overview", "/", "概览"),
    ("records", "/records", "数据明细"),
    ("tasks", "/tasks", "任务管理"),
    ("config", "/config", "配置管理"),
    ("logs", "/logs", "日志管理"),
    ("users", "/users", "用户管理"),
    ("dashboard", "/dashboard", "用量大屏"),
]


def _find_browser() -> str | None:
    """找一个可用的 Chromium 可执行文件。

    Playwright 的驱动版本与本地已下载的浏览器版本常常错位（例如驱动要
    chromium_headless_shell-1234 而本机只有 -1228），此时 launch() 会直接报
    「Executable doesn't exist」。这里按优先级探测，命中就显式传 executable_path。
    """
    import glob
    home = os.environ.get("LOCALAPPDATA") or os.path.expanduser("~")
    pats = [
        os.path.join(home, "ms-playwright", "chromium-*", "chrome-win64", "chrome.exe"),
        os.path.join(home, "ms-playwright", "chromium_headless_shell-*",
                     "chrome-headless-shell-win64", "chrome-headless-shell.exe"),
        r"C:\Program Files\Google\Chrome\Application\chrome.exe",
        r"C:\Program Files (x86)\Google\Chrome\Application\chrome.exe",
        r"C:\Program Files (x86)\Microsoft\Edge\Application\msedge.exe",
        r"C:\Program Files\Microsoft\Edge\Application\msedge.exe",
    ]
    for p in pats:
        hits = sorted(glob.glob(p))
        if hits:
            return hits[-1]
    return None


def main() -> int:
    ap = argparse.ArgumentParser()
    ap.add_argument("--base", default="http://127.0.0.1:8849")
    ap.add_argument("-u", "--user", default="admin")
    ap.add_argument("-p", "--password", default="admin123")
    ap.add_argument("--out", default=os.path.join(BASE, "data", "shots"))
    ap.add_argument("--full", action="store_true", help="截整页长图")
    ap.add_argument("--browser", default="", help="显式指定 chrome/msedge 可执行文件")
    ap.add_argument("--width", type=int, default=1440)
    ap.add_argument("--height", type=int, default=900)
    a = ap.parse_args()

    from playwright.sync_api import sync_playwright

    exe = a.browser or _find_browser()
    os.makedirs(a.out, exist_ok=True)
    problems = []
    with sync_playwright() as pw:
        try:
            br = pw.chromium.launch(executable_path=exe) if exe else pw.chromium.launch()
        except Exception as e:                  # noqa: BLE001
            print("[FAIL] 启动浏览器失败：%s" % e)
            print("       可用 --browser 显式指定，或用 `playwright install chromium` 装齐。")
            return 1
        print("浏览器：%s" % (exe or "playwright 默认"))
        ctx = br.new_context(viewport={"width": a.width, "height": a.height},
                             device_scale_factor=2, locale="zh-CN")
        page = ctx.new_page()
        errors = []
        page.on("console", lambda m: errors.append(m.text) if m.type == "error" else None)
        page.on("pageerror", lambda e: errors.append(str(e)))

        # 1) 先截未登录的登录页
        page.goto(a.base + "/login", wait_until="networkidle")
        page.screenshot(path=os.path.join(a.out, "00-login.png"), full_page=a.full)
        print("[ok] 00-login.png")

        # 2) 登录
        page.fill('input[name=username]', a.user)
        page.fill('input[name=password]', a.password)
        page.click('button[type=submit]')
        page.wait_for_load_state("networkidle")
        if "/login" in page.url:
            print("[FAIL] 登录失败，后续截图无意义")
            br.close()
            return 1

        # 3) 逐页截图
        for i, (slug, path, label) in enumerate(PAGES[1:], start=1):
            errors.clear()
            page.goto(a.base + path, wait_until="networkidle")
            page.wait_for_timeout(900)          # 等 ECharts / 表格渲染稳下来
            page.screenshot(path=os.path.join(a.out, "%02d-%s.png" % (i, slug)),
                            full_page=a.full)
            js_err = [e for e in errors if "favicon" not in e.lower()]
            flag = "" if not js_err else "  [JS错误] " + " | ".join(js_err[:3])
            if js_err:
                problems.append("%s: %s" % (label, js_err[:3]))
            print("[ok] %02d-%s.png  %s%s" % (i, slug, label, flag))

        # 4) 大屏页再点几个交互，确认控件联动不炸
        page.goto(a.base + "/dashboard", wait_until="networkidle")
        page.wait_for_timeout(1200)
        for sel in ["#segRange button", ".seg button"]:
            btns = page.query_selector_all(sel)
            if len(btns) > 1:
                errors.clear()
                btns[1].click()
                page.wait_for_timeout(900)
                page.screenshot(path=os.path.join(a.out, "08-dashboard-interact.png"),
                                full_page=a.full)
                js_err = [e for e in errors if "favicon" not in e.lower()]
                print("[ok] 08-dashboard-interact.png  点击 %s 第 2 项%s"
                      % (sel, "" if not js_err else "  [JS错误] %s" % js_err[:2]))
                if js_err:
                    problems.append("大屏交互: %s" % js_err[:2])
                break

        br.close()

    print("\n截图目录：%s" % a.out)
    if problems:
        print("发现问题：")
        for p in problems:
            print("  - " + p)
        return 1
    print("RESULT: 全页面截图完成，无 JS 报错")
    return 0


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