feat: wkhtmltoimage支持自定义高度

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: