Screenshot parameter

This commit is contained in:
estebanthi 2025-01-24 11:08:41 +01:00
parent d1f19405a1
commit 7ff4e4f44f
2 changed files with 5 additions and 0 deletions

View File

@ -10,6 +10,7 @@ class ChallengeResolutionResultT:
response: str = None response: str = None
cookies: list = None cookies: list = None
userAgent: str = None userAgent: str = None
screenshot: str | None = None
def __init__(self, _dict): def __init__(self, _dict):
self.__dict__.update(_dict) self.__dict__.update(_dict)
@ -41,6 +42,7 @@ class V1RequestBase(object):
url: str = None url: str = None
postData: str = None postData: str = None
returnOnlyCookies: bool = None returnOnlyCookies: bool = None
returnScreenshot: bool = None
download: bool = None # deprecated v2.0.0, not used download: bool = None # deprecated v2.0.0, not used
returnRawHtml: bool = None # deprecated v2.0.0, not used returnRawHtml: bool = None # deprecated v2.0.0, not used

View File

@ -418,6 +418,9 @@ def _evil_logic(req: V1RequestBase, driver: WebDriver, method: str) -> Challenge
challenge_res.headers = {} # todo: fix, selenium not provides this info challenge_res.headers = {} # todo: fix, selenium not provides this info
challenge_res.response = driver.page_source challenge_res.response = driver.page_source
if req.returnScreenshot:
challenge_res.screenshot = driver.get_screenshot_as_base64()
res.result = challenge_res res.result = challenge_res
return res return res