From 7c3d0a09473c4a477678ad9581eab1ee17cd2fc3 Mon Sep 17 00:00:00 2001
From: yuyx <237899745@qq.com>
Date: Wed, 31 Dec 2025 19:13:20 +0800
Subject: [PATCH] =?UTF-8?q?fix:=20wkhtmltoimage=E5=85=BC=E5=AE=B9UA?=
 =?UTF-8?q?=E5=8F=82=E6=95=B0?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 services/screenshots.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/services/screenshots.py b/services/screenshots.py
index e3c0d70..6923ab7 100644
--- a/services/screenshots.py
+++ b/services/screenshots.py
@@ -80,10 +80,11 @@ def take_screenshot_wkhtmltoimage(
         "--enable-local-file-access",
         "--encoding",
         "utf-8",
-        "--user-agent",
-        _WKHTMLTOIMAGE_UA,
     ]
 
+    if _WKHTMLTOIMAGE_UA:
+        cmd.extend(["--custom-header", "User-Agent", _WKHTMLTOIMAGE_UA, "--custom-header-propagation"])
+
     if image_format in ("jpg", "jpeg"):
         cmd.extend(["--quality", str(_WKHTMLTOIMAGE_QUALITY)])