From ea1c7e8a001be39202badce750328e5c33701c9c Mon Sep 17 00:00:00 2001
From: yuyx <237899745@qq.com>
Date: Wed, 31 Dec 2025 20:05:39 +0800
Subject: [PATCH] =?UTF-8?q?feat:=20wkhtmltoimage=E6=94=AF=E6=8C=81?=
 =?UTF-8?q?=E8=87=AA=E5=AE=9A=E4=B9=89=E9=AB=98=E5=BA=A6?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 services/screenshots.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/services/screenshots.py b/services/screenshots.py
index b1cd7d0..1d037cf 100644
--- a/services/screenshots.py
+++ b/services/screenshots.py
@@ -28,6 +28,7 @@ os.makedirs(SCREENSHOTS_DIR, exist_ok=True)
 _WKHTMLTOIMAGE_TIMEOUT_SECONDS = int(os.environ.get("WKHTMLTOIMAGE_TIMEOUT_SECONDS", "60"))
 _WKHTMLTOIMAGE_JS_DELAY_MS = int(os.environ.get("WKHTMLTOIMAGE_JS_DELAY_MS", "3000"))
 _WKHTMLTOIMAGE_WIDTH = int(os.environ.get("WKHTMLTOIMAGE_WIDTH", "1920"))
+_WKHTMLTOIMAGE_HEIGHT = int(os.environ.get("WKHTMLTOIMAGE_HEIGHT", "2000"))
 _WKHTMLTOIMAGE_QUALITY = int(os.environ.get("WKHTMLTOIMAGE_QUALITY", "95"))
 _WKHTMLTOIMAGE_UA = os.environ.get(
     "WKHTMLTOIMAGE_USER_AGENT",
@@ -118,6 +119,9 @@ def take_screenshot_wkhtmltoimage(
     if image_format in ("jpg", "jpeg"):
         cmd.extend(["--quality", str(_WKHTMLTOIMAGE_QUALITY)])
 
+    if _WKHTMLTOIMAGE_HEIGHT > 0:
+        cmd.extend(["--height", str(_WKHTMLTOIMAGE_HEIGHT)])
+
     if cookies_path:
         cookie_pairs = _select_cookie_pairs(_read_cookie_pairs(cookies_path))
         if cookie_pairs: