o ѷ6i$@sfddlZddlmZddlmZddlmZddlmZGdddZ Gdd d Z Gd d d Z dS) N)Any)command_builder)UserPromptHandler)Proxyc@s,eZdZdZdZdZdZdZeeeehZdS)ClientWindowStatezRepresents a window state. fullscreen maximized minimizednormalN) __name__ __module__ __qualname____doc__ FULLSCREEN MAXIMIZED MINIMIZEDNORMAL VALID_STATESrrk/var/www/hoanhtaovolam_webdjango/env/lib/python3.10/site-packages/selenium/webdriver/common/bidi/browser.pyrsrc@seZdZdZdededededededefd d Zd efd d Zd efddZ d efddZ d efddZ d efddZ d efddZ d efddZeded dfddZdS)ClientWindowInfoz'Represents a client window information. client_windowstatewidthheightxyactivecCs.||_||_||_||_||_||_||_dSNrrrrrrr)selfrrrrrrrrrr__init__'s  zClientWindowInfo.__init__returncC|jS)zGets the state of the client window. Returns: str: The state of the client window (one of the ClientWindowState constants). )rr rrr get_state9zClientWindowInfo.get_statecCr#)zlGets the client window identifier. Returns: str: The client window identifier. )rr$rrrget_client_windowAr&z"ClientWindowInfo.get_client_windowcCr#)zpGets the width of the client window. Returns: int: The width of the client window. )rr$rrr get_widthIr&zClientWindowInfo.get_widthcCr#)zrGets the height of the client window. Returns: int: The height of the client window. )rr$rrr get_heightQr&zClientWindowInfo.get_heightcCr#)z~Gets the x coordinate of the client window. Returns: int: The x coordinate of the client window. )rr$rrrget_xYr&zClientWindowInfo.get_xcCr#)z~Gets the y coordinate of the client window. Returns: int: The y coordinate of the client window. )rr$rrrget_yar&zClientWindowInfo.get_ycCr#)zChecks if the client window is active. Returns: bool: True if the client window is active, False otherwise. )rr$rrr is_activeir&zClientWindowInfo.is_activedatac CsPz|d}t|tstd|d}t|tstd|tjvr+td|dtj|d}t|tr8|dkr?td ||d }t|trL|dkrStd ||d }t|tsftd t|j|d}t|tsytdt|j|d}t|tstd||||||||dWSt t fy} ztd| | d} ~ ww)aJCreates a ClientWindowInfo instance from a dictionary. Args: data: A dictionary containing the client window information. Returns: ClientWindowInfo: A new instance of ClientWindowInfo. Raises: ValueError: If required fields are missing or have invalid types. clientWindowzclientWindow must be a stringrzstate must be a stringzInvalid state: z. Must be one of rrz*width must be a non-negative integer, got rz+height must be a non-negative integer, got rzx must be an integer, got rzy must be an integer, got rzactive must be a booleanrz*Invalid data format for ClientWindowInfo: N) isinstancestr ValueErrorrrinttyper boolKeyError TypeError) clsr-rrrrrrrerrr from_dictqsJ        zClientWindowInfo.from_dictN)r r r rr0r2r4r!r%r'r(r)r*r+r, classmethoddictr9rrrrr$s4 rc @seZdZdZddZ   ddedBdedBdedBdefd d Z de efd d Z d eddfddZ de e fddZdddddedBdeejBdBde edBddfddZdS)Browserz*BiDi implementation of the browser module.cCs ||_dSr)conn)r r=rrrr!s zBrowser.__init__Naccept_insecure_certsproxyunhandled_prompt_behaviorr"cCsVi}|dur ||d<|dur||d<|dur||d<|jtd|}|dS)aoCreates a new user context. Args: accept_insecure_certs: Optional flag to accept insecure TLS certificates. proxy: Optional proxy configuration for the user context. unhandled_prompt_behavior: Optional configuration for handling user prompts. Returns: str: The ID of the created user context. NacceptInsecureCertsr?unhandledPromptBehaviorzbrowser.createUserContext userContext) to_bidi_dictto_dictr=executer)r r>r?r@paramsresultrrrcreate_user_contexts  zBrowser.create_user_contextcC$|jtdi}dd|dDS)zeGets all user contexts. Returns: List[str]: A list of user context IDs. zbrowser.getUserContextscSsg|]}|dqS)rCr).0 context_inforrr sz-Browser.get_user_contexts.. userContextsr=rFrr rHrrrget_user_contextszBrowser.get_user_contextsuser_context_idcCs.|dkrtdd|i}|jtd|dS)zRemoves a user context. Args: user_context_id: The ID of the user context to remove. Raises: ValueError: If the user context ID is "default" or does not exist. defaultz&Cannot remove the default user contextrCzbrowser.removeUserContextN)r1r=rFr)r rSrGrrrremove_user_contexts zBrowser.remove_user_contextcCrJ)z|Gets all client windows. Returns: List[ClientWindowInfo]: A list of client window information. zbrowser.getClientWindowscSsg|]}t|qSr)rr9)rKwindowrrrrMsz.Browser.get_client_windows.. clientWindowsrOrPrrrget_client_windowsrRzBrowser.get_client_windows)alloweddestination_folder user_contextsrYrZr[cCszi}|dur d|d<n|r|stddt|d|d<n |r$tdddi|d<|dur2||d <|jtd |dS) aSet the download behavior for the browser or specific user contexts. Args: allowed: True to allow downloads, False to deny downloads, or None to clear download behavior (revert to default). destination_folder: Required when allowed is True. Specifies the folder to store downloads in. user_contexts: Optional list of user context IDs to apply this behavior to. If omitted, updates the default behavior. Raises: ValueError: If allowed=True and destination_folder is missing, or if allowed=False and destination_folder is provided. NdownloadBehaviorz1destination_folder is required when allowed=True.rY)r3destinationFolderz=destination_folder should not be provided when allowed=False.r3deniedrNzbrowser.setDownloadBehavior)r1osfspathr=rFr)r rYrZr[rGrrrset_download_behaviors   zBrowser.set_download_behavior)NNN)r r r rr!r4rrr0rIlistrQrUrrXr_PathLikerarrrrr<s<     r<) r_typingr%selenium.webdriver.common.bidi.commonr&selenium.webdriver.common.bidi.sessionrselenium.webdriver.common.proxyrrrr<rrrrs