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)])