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: