
    ]jY                    @   d dl mZ d dlmZmZ d dlmZ d dlmZ  G d d      Z	e G d d             Z
e G d	 d
             Ze G d d             Ze G d d             Ze G d d             Ze G d d             Ze G d d             Ze G d d             Ze G d d             Ze G d d             Ze G d d             Ze G d d             Ze G d d              Ze G d! d"             Ze G d# d$             Ze G d% d&             Z G d' d(      Zy))*    )annotations)	dataclassfield)Any)command_builderc                      e Zd ZdZdZdZdZy)UserPromptHandlerTypezUserPromptHandlerType.acceptdismissignoreN)__name__
__module____qualname____doc__ACCEPTDISMISSIGNORE     Z/home/ubuntu/.local/lib/python3.12/site-packages/selenium/webdriver/common/bidi/session.pyr	   r	      s     FGFr   r	   c                  <    e Zd ZU dZdZded<    ee      Zded<   y)CapabilitiesRequestzCapabilitiesRequest.N
Any | Nonealways_matchdefault_factory	list[Any]first_match)	r   r   r   r   r   __annotations__r   listr   r   r   r   r   r      s    #L*#"48K8r   r   c                  f    e Zd ZU dZdZded<   dZded<   dZded<   dZded<   dZ	d	ed
<   dZ
d	ed<   y)CapabilityRequestzCapabilityRequest.Nbool | Noneaccept_insecure_certs
str | Nonebrowser_namebrowser_versionplatform_namer   proxyunhandled_prompt_behavior)r   r   r   r   r$   r   r&   r'   r(   r)   r*   r   r   r   r"   r"      sC    )-;-#L*#"&OZ& $M:$E:,0z0r   r"   c                  0    e Zd ZU dZ edd      Zded<   y)AutodetectProxyConfigurationzAutodetectProxyConfiguration.
autodetectFdefaultinitstr
proxy_typeNr   r   r   r   r   r2   r   r   r   r   r,   r,   *   s    'Lu=J=r   r,   c                  0    e Zd ZU dZ edd      Zded<   y)DirectProxyConfigurationzDirectProxyConfiguration.directFr.   r1   r2   Nr3   r   r   r   r5   r5   1       #H59J9r   r5   c                  h    e Zd ZU dZ edd      Zded<   dZded	<   dZded
<    ee	      Z
ded<   y)ManualProxyConfigurationzManualProxyConfiguration.manualFr.   r1   r2   Nr%   
http_proxy	ssl_proxyr   r   no_proxy)r   r   r   r   r   r2   r   r;   r<   r    r=   r   r   r   r9   r9   8   s9    #H59J9!J
! Iz 5Hi5r   r9   c                  .    e Zd ZU dZdZded<   dZded<   y)SocksProxyConfigurationzSocksProxyConfiguration.Nr%   socks_proxyr   socks_version)r   r   r   r   r@   r   rA   r   r   r   r?   r?   B   s    ""K" $M:$r   r?   c                  >    e Zd ZU dZ edd      Zded<   dZded	<   y)
PacProxyConfigurationzPacProxyConfiguration.pacFr.   r1   r2   Nr%   proxy_autoconfig_url)r   r   r   r   r   r2   r   rE   r   r   r   rC   rC   J   s!     E6J6'+*+r   rC   c                  0    e Zd ZU dZ edd      Zded<   y)SystemProxyConfigurationzSystemProxyConfiguration.systemFr.   r1   r2   Nr3   r   r   r   rG   rG   R   r7   r   rG   c                  f    e Zd ZU dZ ee      Zded<    ee      Zded<    ee      Z	ded<   y)	SubscribeParameterszSubscribeParameters.r   	list[str]eventsr   contextsuser_contextsN)
r   r   r   r   r   r    rL   r   rM   rN   r   r   r   rJ   rJ   Y   s2    d3FI35Hi5$T:M9:r   rJ   c                  .    e Zd ZU dZ ee      Zded<   y)UnsubscribeByIDRequestzUnsubscribeByIDRequest.r   r   subscriptionsN)r   r   r   r   r   r    rQ   r   r   r   r   rP   rP   b   s    !$T:M9:r   rP   c                  .    e Zd ZU dZ ee      Zded<   y)UnsubscribeByAttributesRequestzUnsubscribeByAttributesRequest.r   rK   rL   N)r   r   r   r   r   r    rL   r   r   r   r   rS   rS   i   s    )d3FI3r   rS   c                  .    e Zd ZU dZdZded<   dZded<   y)StatusResultzStatusResult.Nr#   readyr%   message)r   r   r   r   rV   r   rW   r   r   r   rU   rU   p   s    E;GZr   rU   c                       e Zd ZU dZdZded<   y)NewParameterszNewParameters.Nr   capabilities)r   r   r   r   rZ   r   r   r   r   rY   rY   x   s    #L*#r   rY   c                      e Zd ZU dZdZded<   dZded<   dZded<   dZded<   dZ	ded	<   dZ
ded
<   dZded<   dZded<   dZded<   dZded<   y)	NewResultz
NewResult.Nr%   
session_idr#   r$   r&   r'   r(   set_window_rect
user_agentr   r)   r*   web_socket_url)r   r   r   r   r]   r   r$   r&   r'   r(   r^   r_   r)   r*   r`   r   r   r   r\   r\      sk    !J
!)-;-#L*#"&OZ& $M:$#'O['!J
!E:,0z0!%NJ%r   r\   c                       e Zd ZU dZdZded<   y)SubscribeResultzSubscribeResult.Nr   subscription)r   r   r   r   rc   r   r   r   r   rb   rb      s    #L*#r   rb   c                  v    e Zd ZU dZdZded<   dZded<   dZded<   dZded<   dZ	ded<   dZ
ded	<   dd
ZddZy)UserPromptHandlerzUserPromptHandler.Nr   alertbefore_unloadconfirmr/   filepromptc                N   i }| j                   | j                   |d<   | j                  | j                  |d<   | j                  | j                  |d<   | j                  | j                  |d<   | j                  | j                  |d<   | j
                  | j
                  |d<   |S )z2Convert to BiDi protocol dict with camelCase keys.rf   beforeUnloadrh   r/   ri   rj   )rf   rg   rh   r/   ri   rj   )selfresults     r   to_bidi_dictzUserPromptHandler.to_bidi_dict   s    ::!"jjF7O)%)%7%7F>"<<# $F9<<# $F999 !YYF6N;;"#{{F8r   c                "    | j                         S )z-Backward-compatible alias for to_bidi_dict().)ro   )rm   s    r   to_dictzUserPromptHandler.to_dict   s      ""r   )returndict)r   r   r   r   rf   r   rg   rh   r/   ri   rj   ro   rq   r   r   r   re   re      sK    E: $M:$GZGZD*FJ"#r   re   c                  R    e Zd ZdZd	dZd Zd
ddZd Z	 	 	 d	 	 	 	 	 ddZdddZ	y)SessionzWebDriver BiDi session module.c                    || _         y N)_conn)rm   conns     r   __init__zSession.__init__   s	    
r   c                    i }|j                         D ci c]  \  }}|	|| }}}t        d|      }| j                  j                  |      }|S c c}}w )zExecute session.statuszsession.statusitemsr   rx   executerm   paramskvcmdrn   s         r   statuszSession.status   sZ    
#)<<>C41aQ]!Q$CC.7##C( D
   
AANc                    |t        d      d|i}|j                         D ci c]  \  }}|	|| }}}t        d|      }| j                  j	                  |      }|S c c}}w )zExecute session.newz/new() missing required argument: 'capabilities'rZ   zsession.new	TypeErrorr}   r   rx   r~   )rm   rZ   r   r   r   r   rn   s          r   newzSession.new   st    MNN L
 $*<<>C41aQ]!Q$CCmV4##C( Ds
   
A!A!c                    i }|j                         D ci c]  \  }}|	|| }}}t        d|      }| j                  j                  |      }|S c c}}w )zExecute session.endzsession.endr|   r   s         r   endzSession.end   sY    
#)<<>C41aQ]!Q$CCmV4##C( Dr   c                    |t        d      |||d}|j                         D ci c]  \  }}|	|| }}}t        d|      }| j                  j	                  |      }|S c c}}w )zExecute session.subscribez/subscribe() missing required argument: 'events')rL   rM   userContextszsession.subscriber   )	rm   rL   rM   rN   r   r   r   r   rn   s	            r   	subscribezSession.subscribe   sz     >MNN  )

 $*<<>C41aQ]!Q$CC16:##C( Ds
   
A#A#c                    ||d}|j                         D ci c]  \  }}|	|| }}}t        d|      }| j                  j                  |      }|S c c}}w )zExecute session.unsubscribe)rL   rQ   zsession.unsubscriber|   )rm   rL   rQ   r   r   r   r   rn   s           r   unsubscribezSession.unsubscribe   sd     *
 $*<<>C41aQ]!Q$CC3V<##C( Ds
   
AA)rr   Nonerw   )rZ   r   )NNN)rL   list[Any] | NonerM   r   rN   r   )NN)rL   r   rQ   r   )
r   r   r   r   rz   r   r   r   r   r   r   r   r   ru   ru      sJ    ( $(%)*.	  # (	(	r   ru   N)
__future__r   dataclassesr   r   typingr   %selenium.webdriver.common.bidi.commonr   r	   r   r"   r,   r5   r9   r?   rC   rG   rJ   rP   rS   rU   rY   r\   rb   re   ru   r   r   r   <module>r      s   # (  A  9 9 9 1 1 1 > > > : : : 6 6 6 % % % , , , : : : ; ; ; ; ; ; 4 4 4    $ $ $ & & & $ $ $ # # #>B Br   