wangchuanli
23799b4ea5
* LICENSE —— MIT
* THIRD-PARTY-NOTICES —— 依赖清单、再分发合规说明（含随仓库分发的
Apache ECharts 5.6.0 / Apache-2.0）与自查清单
* CONTRIBUTING.md —— 开发环境、五层验证、必须遵守的不变量、提交规范
* SECURITY.md —— 漏洞私有报告渠道、已有措施、已知非目标
* CODE_OF_CONDUCT.md —— 改编自 Contributor Covenant 2.1
* .github/ —— Bug 报告 / 功能建议表单 + PR 模板
* .editorconfig —— 与 .gitattributes 保持一致
* 全部 Python / Shell 源文件加 SPDX-License-Identifier: MIT 头
* README 增加「开源与许可」章节与许可标识
156 行
6.3 KiB
Python
156 行
6.3 KiB
Python
#!/usr/bin/env python
|
|
# -*- coding: utf-8 -*-
|
|
# SPDX-License-Identifier: MIT
|
|
# Copyright (c) 2026 Wang Chuanli
|
|
|
|
"""界面实检：登录后逐页截图，用来目视确认「统一美化」是否真的落地。
|
|
|
|
用法：
|
|
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())
|