@taloswork/ose-shared 0.1.8 → 0.1.9

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

@@ -0,0 +1,333 @@
+#!/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())