@taloswork/ose-shared 0.1.10 → 0.1.11

package/skills/cozy-static-svg/scripts/to_png.py

@@ -1,333 +0,0 @@
-#!/usr/bin/env python3
-"""SVG → PNG（跨平台：macOS / Windows / Linux 同一条命令，纯标准库 + 系统已有工具）。
-
-仅当用户明确要 PNG 时才用，默认工作流仍是「直出原生 SVG，不转 PNG」。
-
-用法（两个平台写法一致）：
-    macOS / Linux   python3 scripts/to_png.py 作品.svg
-    Windows         py -3 scripts\\to_png.py 作品.svg
-    python3 scripts/to_png.py --scale 2 .          # 目录下全部 SVG，2 倍图
-    py -3 scripts\\to_png.py -o out --width 2400 作品目录
-    python3 scripts/to_png.py --backend inkscape 作品.svg
-
-后端自动探测，找到哪个用哪个（也可用 --backend 指定）：
-    1. rsvg-convert       与手绘流程同源，最推荐
-                          macOS: brew install librsvg ｜ Windows: MSYS2 / scoop 自行安装
-    2. inkscape           Inkscape 1.x，装了图形界面就自带命令行
-    3. cairosvg           pip install cairosvg（纯 Python 依赖）
-    4. chrome / edge      系统自带浏览器无头截图，零安装兜底
-
-其它参数：
-    -o DIR        输出目录（默认与源文件同目录）
-    --scale N     相对 SVG 原始尺寸的倍数（默认 1）
-    --width / --height   指定像素尺寸（只给一边则按比例算另一边）
-    --background  RRGGBB 铺不透明底色（默认保留透明，圆角外仍是透明的）
-"""
-from __future__ import annotations
-
-import argparse
-import glob
-import os
-import re
-import shutil
-import subprocess
-import sys
-import tempfile
-import xml.etree.ElementTree as ET
-from pathlib import Path
-
-if sys.version_info < (3, 9):
-    sys.exit(
-        "需要 Python 3.9 或更新版本（当前 %d.%d）；"
-        "Windows 请用 py -3 运行，或到 python.org 安装新版。" % sys.version_info[:2]
-    )
-
-if hasattr(sys.stdout, "reconfigure"):  # Windows 控制台默认 GBK，强制 UTF-8 输出
-    try:
-        sys.stdout.reconfigure(encoding="utf-8", errors="replace")
-    except (AttributeError, ValueError):
-        pass
-
-CHROME_CANDIDATES = {
-    "darwin": [
-        "/Applications/Google Chrome.app/Contents/MacOS/Google Chrome",
-        "/Applications/Microsoft Edge.app/Contents/MacOS/Microsoft Edge",
-        "/Applications/Chromium.app/Contents/MacOS/Chromium",
-    ],
-    "win32": [
-        r"%PROGRAMFILES%\Google\Chrome\Application\chrome.exe",
-        r"%PROGRAMFILES(X86)%\Google\Chrome\Application\chrome.exe",
-        r"%LOCALAPPDATA%\Google\Chrome\Application\chrome.exe",
-        r"%PROGRAMFILES(X86)%\Microsoft\Edge\Application\msedge.exe",
-        r"%PROGRAMFILES%\Microsoft\Edge\Application\msedge.exe",
-    ],
-}
-CHROME_ON_PATH = ["chrome", "google-chrome", "chromium", "chromium-browser", "msedge"]
-
-
-# ---------- 输入展开（与 artkit.py 同规则：文件 / 目录 / 通配符） ----------
-
-def collect_sources(items: list[str]) -> list[Path]:
-    found: list[Path] = []
-    for item in items:
-        text = str(item)
-        path = Path(text)
-        if path.is_dir():
-            found.extend(
-                sorted(
-                    p
-                    for p in path.rglob("*")
-                    if p.is_file() and p.suffix.lower() == ".svg"
-                )
-            )
-            continue
-        if any(ch in text for ch in "*?["):
-            found.extend(
-                sorted(
-                    Path(m)
-                    for m in glob.glob(text, recursive=True)
-                    if Path(m).is_file() and Path(m).suffix.lower() == ".svg"
-                )
-            )
-            continue
-        found.append(path)
-
-    unique: list[Path] = []
-    seen: set[str] = set()
-    for p in found:
-        try:
-            key = str(p.resolve())
-        except OSError:
-            key = str(p)
-        if key not in seen:
-            seen.add(key)
-            unique.append(p)
-    return unique
-
-
-# ---------- 后端探测 ----------
-
-def _which(*names: str) -> str | None:
-    for name in names:
-        hit = shutil.which(name)
-        if hit:
-            return hit
-    return None
-
-
-def find_chrome() -> str | None:
-    for raw in CHROME_CANDIDATES.get(sys.platform, []):
-        path = os.path.expandvars(raw)
-        if Path(path).is_file():
-            return path
-    return _which(*CHROME_ON_PATH)
-
-
-def find_backends() -> dict[str, str]:
-    """返回 {后端名: 可执行路径或说明}；cairosvg 用 'python' 表示模块可用。"""
-    backends: dict[str, str] = {}
-    if (rsvg := _which("rsvg-convert")):
-        backends["rsvg"] = rsvg
-    if (ink := _which("inkscape")) or Path(
-        "/Applications/Inkscape.app/Contents/MacOS/inkscape"
-    ).is_file():
-        backends["inkscape"] = ink or "/Applications/Inkscape.app/Contents/MacOS/inkscape"
-    try:
-        import cairosvg  # noqa: F401
-
-        backends["cairosvg"] = "python-module"
-    except ImportError:
-        pass
-    if (chrome := find_chrome()):
-        backends["chrome"] = chrome
-    return backends
-
-
-# ---------- 尺寸 ----------
-
-def svg_size(src: Path) -> tuple[float, float] | None:
-    try:
-        root = ET.parse(src).getroot()
-    except (ET.ParseError, OSError):
-        return None
-    size = re.compile(r"^\s*(\d+(?:\.\d+)?)")
-    out = []
-    for attr in ("width", "height"):
-        match = size.match(root.get(attr) or "")
-        if not match:
-            return None
-        out.append(float(match.group(1)))
-    return out[0], out[1]
-
-
-def target_size(src: Path, scale: float, width: float | None, height: float | None):
-    base = svg_size(src) or (1200.0, 900.0)
-    w, h = base[0] * scale, base[1] * scale
-    if width and height:
-        return width, height
-    if width:
-        return width, round(w and h * width / w)
-    if height:
-        return round(w * height / h), height
-    return round(w), round(h)
-
-
-# ---------- 各后端转换 ----------
-
-def convert_rsvg(exe: str, src: Path, out: Path, size, background: str | None) -> None:
-    cmd = [exe, "-w", str(size[0]), "-h", str(size[1])]
-    if background:
-        cmd += ["-b", f"#{background.lstrip('#')}"]
-    cmd += [str(src), "-o", str(out)]
-    subprocess.run(cmd, check=True, capture_output=True)
-
-
-def convert_inkscape(exe: str, src: Path, out: Path, size, background: str | None) -> None:
-    cmd = [
-        exe,
-        "--export-type=png",
-        f"--export-filename={out}",
-        f"--export-width={size[0]}",
-        f"--export-height={size[1]}",
-    ]
-    if background:
-        cmd += [f"--export-background=#{background.lstrip('#')}", "--export-background-opacity=255"]
-    cmd.append(str(src))
-    subprocess.run(cmd, check=True, capture_output=True)
-
-
-def convert_cairosvg(src: Path, out: Path, size, background: str | None) -> None:
-    import cairosvg
-
-    cairosvg.svg2png(
-        url=str(src),
-        write_to=str(out),
-        output_width=size[0],
-        output_height=size[1],
-        background_color=f"#{background.lstrip('#')}" if background else None,
-    )
-
-
-def convert_chrome(exe: str, src: Path, out: Path, size, background: str | None) -> None:
-    markup = src.read_text(encoding="utf-8")
-    markup = re.sub(r"<\?xml[^>]*\?>", "", markup, count=1).strip()
-    if background:
-        markup = markup.replace(
-            "<svg ", f'<svg style="background:#{background.lstrip("#")}" ', 1
-        )
-    html = (
-        "<!doctype html><html><head><meta charset=\"utf-8\"><style>"
-        "html,body{margin:0;padding:0;overflow:hidden;background:transparent}"
-        "svg{display:block;width:100vw;height:100vh}</style></head><body>"
-        + markup
-        + "</body></html>"
-    )
-    with tempfile.TemporaryDirectory() as tmp:
-        page = Path(tmp) / "page.html"
-        page.write_text(html, encoding="utf-8")
-        common = [
-            f"--screenshot={out}",
-            f"--window-size={size[0]},{size[1]}",
-            "--hide-scrollbars",
-            "--force-device-scale-factor=1",
-            "--virtual-time-budget=2000",
-            "--default-background-color=00000000",
-            page.resolve().as_uri(),
-        ]
-        for headless in ("--headless=new", "--headless"):
-            try:
-                subprocess.run(
-                    [exe, headless, *common], check=True, capture_output=True, timeout=120
-                )
-                if out.is_file():
-                    return
-            except (subprocess.CalledProcessError, subprocess.TimeoutExpired):
-                continue
-    raise RuntimeError("无头浏览器没有产出截图（可换 --backend inkscape 或 rsvg）")
-
-
-# ---------- 主流程 ----------
-
-def main() -> int:
-    parser = argparse.ArgumentParser(
-        description="SVG → PNG（跨平台，自动选择可用后端）",
-        epilog="参数可以是文件、目录或通配符。默认保留透明底，输出与源文件同目录。",
-    )
-    parser.add_argument("svgs", nargs="+", help="待转换的 SVG（文件 / 目录 / 通配符）")
-    parser.add_argument("-o", "--outdir", type=Path, help="输出目录")
-    parser.add_argument("--scale", type=float, default=1.0, help="相对原始尺寸的倍数（默认 1）")
-    parser.add_argument("--width", type=float, help="输出宽度（像素）")
-    parser.add_argument("--height", type=float, help="输出高度（像素）")
-    parser.add_argument("--background", help="铺底色，如 FFFFFF（默认透明）")
-    parser.add_argument(
-        "--backend",
-        choices=["auto", "rsvg", "inkscape", "cairosvg", "chrome"],
-        default="auto",
-        help="指定后端（默认 auto：rsvg → inkscape → cairosvg → chrome）",
-    )
-    args = parser.parse_args()
-
-    if args.outdir:
-        args.outdir.mkdir(parents=True, exist_ok=True)
-
-    sources = collect_sources(args.svgs)
-    if not sources:
-        print("没有找到任何 .svg —— 检查路径，或用目录 / 通配符批量指定。")
-        return 1
-
-    backends = find_backends()
-    order = ["rsvg", "inkscape", "cairosvg", "chrome"]
-    if args.backend != "auto":
-        if args.backend not in backends:
-            print(f"指定的后端 {args.backend} 在本机没找到。可用：{', '.join(backends) or '无'}")
-            print(_install_hint())
-            return 1
-        chosen = args.backend
-    else:
-        chosen = next((name for name in order if name in backends), None)
-    if not chosen:
-        print("没有找到任何可用的转换后端。")
-        print(_install_hint())
-        return 1
-
-    print(f"后端：{chosen}（{backends[chosen]}）")
-    failed = 0
-    for index, src in enumerate(sources, 1):
-        name = src.name
-        if not src.is_file():
-            print(f"[{index}/{len(sources)}] {name} ✗ 文件不存在")
-            failed += 1
-            continue
-        size = target_size(src, args.scale, args.width, args.height)
-        out = (args.outdir or src.parent) / (src.stem + ".png")
-        try:
-            if chosen == "rsvg":
-                convert_rsvg(backends["rsvg"], src, out, size, args.background)
-            elif chosen == "inkscape":
-                convert_inkscape(backends["inkscape"], src, out, size, args.background)
-            elif chosen == "cairosvg":
-                convert_cairosvg(src, out, size, args.background)
-            else:
-                convert_chrome(backends["chrome"], src, out, size, args.background)
-            print(f"[{index}/{len(sources)}] {name} → {out.name}（{size[0]:g}×{size[1]:g}）")
-        except (subprocess.CalledProcessError, RuntimeError, OSError, ValueError) as exc:
-            print(f"[{index}/{len(sources)}] {name} ✗ 转换失败：{exc}")
-            failed += 1
-
-    ok = len(sources) - failed
-    print(f"总计：{ok}/{len(sources)} 张转成 PNG。")
-    return 1 if failed else 0
-
-
-def _install_hint() -> str:
-    return (
-        "装一个就行：\n"
-        "  macOS   brew install librsvg   （或 brew install --cask inkscape）\n"
-        "  Windows choco/scoop 装 librsvg，或装 Inkscape，或 pip install cairosvg\n"
-        "     —— 其实系统自带的 Edge 就能当后端，装不了工具时用 --backend chrome\n"
-        "  Linux   apt install librsvg2-bin / dnf install librsvg2-tools"
-    )
-
-
-if __name__ == "__main__":
-    raise SystemExit(main())