diff --git a/src/urlscan/pro/livescan.py b/src/urlscan/pro/livescan.py index df5e757..88eb106 100644 --- a/src/urlscan/pro/livescan.py +++ b/src/urlscan/pro/livescan.py @@ -73,6 +73,8 @@ def task( extra_headers: dict[str, str] | None = None, enable_features: list[str] | None = None, disable_features: list[str] | None = None, + user_agent: str | None = None, + eval_script: str | None = None, **kwargs: Any, ) -> dict: """Task a URL to be scanned. @@ -88,6 +90,8 @@ def task( extra_headers (dict[str, str] | None, optional): Extra HTTP headers. Defaults to None. enable_features (list[str] | None, optional): Features to enable. Defaults to None. disable_features (list[str] | None, optional): Features to disable. Defaults to None. + user_agent (str | None, optional): Custom User-Agent to use for the scan. Defaults to None. + eval_script (str | None, optional): This Javascript snippet will be executed as the first item in the main page context. Defaults to None. **kwargs: Additional parameters to include in the request payload. Returns: @@ -105,6 +109,8 @@ def task( extra_headers=extra_headers, enable_features=enable_features, disable_features=disable_features, + user_agent=user_agent, + eval_script=eval_script, **kwargs, ) res = self._post(f"/api/v1/livescan/{scanner_id}/task/", json=data) @@ -121,6 +127,8 @@ def scan( extra_headers: dict[str, str] | None = None, enable_features: list[str] | None = None, disable_features: list[str] | None = None, + user_agent: str | None = None, + eval_script: str | None = None, **kwargs: Any, ) -> dict: """Task a URL to be scanned. The HTTP request will block until the scan has finished. @@ -134,6 +142,8 @@ def scan( extra_headers (dict[str, str] | None, optional): Extra HTTP headers. Defaults to None. enable_features (list[str] | None, optional): Features to enable. Defaults to None. disable_features (list[str] | None, optional): Features to disable. Defaults to None. + user_agent (str | None, optional): Custom User-Agent to use for the scan. Defaults to None. + eval_script (str | None, optional): This Javascript snippet will be executed as the first item in the main page context. Defaults to None. **kwargs: Additional parameters to include in the request payload. Returns: @@ -151,6 +161,8 @@ def scan( extra_headers=extra_headers, enable_features=enable_features, disable_features=disable_features, + user_agent=user_agent, + eval_script=eval_script, **kwargs, ) res = self._post(f"/api/v1/livescan/{scanner_id}/scan/", json=data)