U
|
[ ±dfE ã @ s² U d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlmZ d dl m
|
Z
|
d dl m Z d dl m Z d dlZd dlmZ d dlmZ e d¡Ze d ¡Zd
|
Zdadad d Ze d ¡a e je d< e d¡a e je d< dd Z dd Z e dd Z!e dd Z"dd Z#dd Z$G dd d e%Z&G d d d eZ'G d d d e%Z(e G d!d" d"Z)G d#d$ d$Z*G d%d& d&e*Z+G d'd( d(e*ej,j-Z.e
|
ej/e. d)d*d+Z0e.d)d,d-Z1dS ).é N)Ú defaultdict)Úasynccontextmanager)Úcontextmanager)Ú dataclass)ÚConnectionClosed)Úconnect_websocket_urlZtrio_cdpÚTi c C s¨ | a d}zt | | ¡atW S tk
|
r¢ t t¡jd d¡}t
|
dd | ¡ D }t dd |D }t t¡}| d| ¡ t | | ¡at Y S X dS ) z^Attempt to load the current latest available devtools into the module
|
cache for use later.z$selenium.webdriver.common.devtools.vé Údevtoolsc s s | ]}| ¡ r|jV qd S ©N)Úis_dirÚname)Ú.0Úf© r úYd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\selenium/webdriver/common/bidi/cdp.pyÚ <genexpr>A s z"import_devtools.<locals>.<genexpr>c s s | ]}t |d d
V qdS )r N)Úint)r Úxr r r r B s z%Falling back to loading `devtools`: vN)ÚversionÚ importlibÚ import_moduler
|
ÚModuleNotFoundErrorÚpathlibÚPathÚ__file__ÚparentsÚjoinpathÚtupleÚiterdirÚmaxÚloggingÚ getLoggerÚ__name__Údebug)ÚverÚbaseZ devtools_pathÚversionsZlatestZselenium_loggerr r r Úimport_devtools3 s
|
r( Úconnection_contextÚ_connection_contextÚsession_contextÚ_session_contextc C s2 z
|
t ¡ W S tk
|
r, t| dY nX dS )zLook up the current connection.
|
|
If there is no current connection, raise a ``RuntimeError`` with a
|
helpful message.
|
z*() must be called in a connection context.N)r* ÚgetÚ LookupErrorÚ RuntimeError©Úfn_namer r r Úget_connection_contextM s
|
r2 c C s2 z
|
t ¡ W S tk
|
r, t| dY nX dS )z{Look up the current session.
|
|
If there is no current session, raise a ``RuntimeError`` with a
|
helpful message.
|
z'() must be called in a session context.N)r, r- r. r/ r0 r r r Úget_session_contextY s
|
r3 c c s& t | ¡}z
|
dV W 5 t |¡ X dS )zbThis context manager installs ``connection`` as the session context for
|
the current Trio task.N)r* ÚsetÚreset)Ú
|
connectionÚtokenr r r r) e s
|
|
c c s& t | ¡}z
|
dV W 5 t |¡ X dS )z_This context manager installs ``session`` as the session context for the
|
current Trio task.N)r, r4 r5 )Úsessionr7 r r r r+ p s
|
|
c C s t jd| dadS )zõInstall ``connection`` in the root context so that it will become the
|
default connection for all tasks.
|
|
This is generally not recommended, except it may be necessary in
|
certain use cases such as running inside Jupyter notebook.
|
r* ©ÚdefaultN)Ú contextvarsÚ
|
ContextVarr* )r6 r r r Úset_global_connection{ s r= c C s t jd| dadS )zïInstall ``session`` in the root context so that it will become the
|
default session for all tasks.
|
|
This is generally not recommended, except it may be necessary in
|
certain use cases such as running inside Jupyter notebook.
|
r, r9 N)r; r< r, )r8 r r r Úset_global_session s r> c @ s e Zd ZdZdd Zdd ZdS )Ú BrowserErrorzgThis exception is raised when the browser's response to a command
|
indicates that an error occurred.c C s$ |d | _ |d | _| d¡| _d S )NÚcodeÚmessageÚdata)r@ rA r- Údetail)ÚselfÚobjr r r Ú__init__ s
|
|
zBrowserError.__init__c C s d| j d| j d| j S )NzBrowserError<code=z message=z> )r@ rA rC ©rD r r r Ú__str__ s zBrowserError.__str__N)r# Ú
|
__module__Ú __qualname__Ú__doc__rF rH r r r r r? s r? c @ s e Zd ZdZdd Zdd ZdS )ÚCdpConnectionClosedzARaised when a public method is called on a closed CDP connection.c C s
|
|| _ dS )zfConstructor.
|
|
:param reason:
|
:type reason: wsproto.frame_protocol.CloseReason
|
N)Úreason)rD rM r r r rF ¡ s z CdpConnectionClosed.__init__c C s | j j d| j dS )zReturn representation.ú<ú>)Ú __class__r# rM rG r r r Ú__repr__© s z CdpConnectionClosed.__repr__N)r# rI rJ rK rF rQ r r r r rL s rL c @ s e Zd ZdZdS )Ú InternalErrorzfThis exception is only raised when there is faulty logic in TrioCDP or
|
the integration with PyCDP.N)r# rI rJ rK r r r r rR ® s rR c @ s e Zd ZU dZdZejed< dS )Ú CmEventProxyz³A proxy object returned by :meth:`CdpBase.wait_for()``.
|
|
After the context manager executes, this proxy object will have a
|
value set that contains the returned event.
|
NÚvalue)r# rI rJ rK rT ÚtypingÚAnyÚ__annotations__r r r r rS ³ s
|
rS c @ s~ e Zd Zdd Zejeeejf edddZ dddd Z
|
e dej e ej ed
|
f d d d Zdd Zdd Zdd Zd
|
S )ÚCdpBasec C s6 || _ || _|| _tt| _t ¡ | _i | _ i | _
|
d S r ) ÚwsÚ
|
session_idÚ target_idr r4 ÚchannelsÚ itertoolsÚcountÚid_iterÚ inflight_cmdÚinflight_result©rD rY rZ r[ r r r rF ¿ s
|
|
zCdpBase.__init__)ÚcmdÚreturnc
|
à s¼ t | j}t ¡ }||f| j|< t |}||d< | jr@| j|d< t |¡}z| j |¡I dH W n. t
|
k
|
r } zt |j dW 5 d}~X Y nX | ¡ I dH | j |¡}t|tr¸||S )z
Execute a command on the server and wait for the result.
|
|
:param cmd: any CDP command
|
:returns: a CDP result
|
ÚidÚ sessionIdN)Únextr_ ÚtrioÚEventr` rZ ÚjsonÚdumpsrY Ú send_messageÚWsConnectionClosedrL rM Úwaitra ÚpopÚ
|
isinstanceÚ Exception)rD rc Úcmd_idZ cmd_eventÚrequestZ request_strZwccÚresponser r r ÚexecuteÈ s"
|
|
|
|
zCdpBase.executeé
|
)Ú buffer_sizec G s, t |¡\}}|D ]}| j| |¡ q|S )zXReturn an async iterator that iterates over events matching the
|
indicated types.)rh Úopen_memory_channelr\ Úadd)rD rw Z event_typesÚsenderÚreceiverÚ
|
event_typer r r Úlistenà s zCdpBase.listenN)r| rd c
|
C s` t |¡\}}| j| |¡ t }|V |4 I dH | ¡ I dH }W 5 Q I dH R X ||_dS )zçWait for an event of the given type and return it.
|
|
This is an async context manager, so you should open it inside
|
an async with block. The block will not exit until the indicated
|
event is received.
|
N)rh rx r\ ry rS ZreceiverT )rD r| rw rz r{ ÚproxyÚeventr r r Úwait_forè s zCdpBase.wait_forc C s" d|kr| |¡ n
|
| |¡ dS )zUHandle incoming WebSocket data.
|
|
:param dict data: a JSON dictionary
|
re N)Ú_handle_cmd_responseÚ _handle_event)rD rB r r r Ú _handle_dataø s zCdpBase._handle_datac
|
C sº |d }z| j |¡\}}W n& tk
|
rB t d| ¡ Y dS X d|kr`t|d | j|< nNz| |d ¡}tdW n( t k
|
r¢ } z
|
|j
|
}W 5 d}~X Y nX || j|< | ¡ dS )zÃHandle a response to a command. This will set an event flag that
|
will return control to the task that called the command.
|
|
:param dict data: response as a JSON dictionary
|
re z5Got a message with a command ID that does not exist: NÚerrorÚresultz<The command's generator function did not exit when expected!) r` ro ÚKeyErrorÚloggerÚwarningr? ra ÚsendrR Ú StopIterationrT r4 )rD rB rr rc r rt ÚexitZreturn_r r r r s
|
z CdpBase._handle_cmd_responsec C s¢ t j |¡}t d|¡ t }| jt| D ]V}z| |¡ W q, t j
|
k
|
rb t d||¡ Y q, t j k
|
r | |¡ Y q,X q,|r| jt| |8 < dS )zOHandle an event.
|
|
:param dict data: event as a JSON dictionary
|
zReceived event: %sz0Unable to send event "%r" due to full channel %sN)r
|
ÚutilZparse_json_eventr r$ r4 r\ ÚtypeZ send_nowaitrh Z
|
WouldBlockr ZBrokenResourceErrorry )rD rB r Z to_removerz r r r r s zCdpBase._handle_event)rv )r# rI rJ rF rU Ú GeneratorÚdictr rV ru r} r ÚTypeÚAsyncGeneratorrS r r r r r r r r rX ¾ s $
|
rX c s8 e Zd ZdZ fddZedd Zedd Z ZS )Ú
|
CdpSessionz¢Contains the state for a CDP session.
|
|
Generally you should not instantiate this object yourself; you should call
|
:meth:`CdpConnection.open_session`.
|
c s4 t |||¡ d| _t ¡ | _d| _t ¡ | _dS )z´Constructor.
|
|
:param trio_websocket.WebSocketConnection ws:
|
:param devtools.target.SessionID session_id:
|
:param devtools.target.TargetID target_id:
|
r N)ÚsuperrF Ú_dom_enable_countrh ÚLockÚ_dom_enable_lockÚ_page_enable_countÚ_page_enable_lockrb ©rP r r rF 8 s
|
|
zCdpSession.__init__c
|
C s¦ | j 4 I dH 4 | jd7 _| jdkr>| tj ¡ ¡I dH W 5 Q I dH R X dV | j 4 I dH 4 | jd8 _| jdkr| tj ¡ ¡I dH W 5 Q I dH R X dS )zåA context manager that executes ``dom.enable()`` when it enters and
|
then calls ``dom.disable()``.
|
|
This keeps track of concurrent callers and only disables DOM
|
events when all callers have exited.
|
Nr r )r r ru r
|
ÚdomÚenableÚdisablerG r r r Ú
|
dom_enableF s
|
&
|
zCdpSession.dom_enablec
|
C s¦ | j 4 I dH 4 | jd7 _| jdkr>| tj ¡ ¡I dH W 5 Q I dH R X dV | j 4 I dH 4 | jd8 _| jdkr| tj ¡ ¡I dH W 5 Q I dH R X dS )zöA context manager that executes ``page.enable()`` when it enters and
|
then calls ``page.disable()`` when it exits.
|
|
This keeps track of concurrent callers and only disables page
|
events when all callers have exited.
|
Nr r )r r ru r
|
Úpager r rG r r r Ú page_enable[ s
|
&
|
zCdpSession.page_enable) r# rI rJ rK rF r r r Ú __classcell__r r r r r 1 s
|
r c sV e Zd ZdZ fddZdd Zeeje dddZ
|
d dd
|
d Z d d Z Z S )Ú CdpConnectiona% Contains the connection state for a Chrome DevTools Protocol server.
|
|
CDP can multiplex multiple "sessions" over a single connection. This
|
class corresponds to the "root" session, i.e. the implicitly created
|
session that has no session ID. This class is responsible for
|
reading incoming WebSocket messages and forwarding them to the
|
corresponding session, as well as handling messages targeted at the
|
root session itself. You should generally call the
|
:func:`open_cdp()` instead of instantiating this class directly.
|
c s t j|ddd i | _dS )zLConstructor.
|
|
:param trio_websocket.WebSocketConnection ws:
|
N)rZ r[ )r rF Úsessions)rD rY r r r rF } s zCdpConnection.__init__c à s | j ¡ I dH dS )at Close the underlying WebSocket connection.
|
|
This will cause the reader task to gracefully exit when it tries
|
to read the next message from the WebSocket. All of the public
|
APIs (``execute()``, ``listen()``, etc.) will raise
|
``CdpConnectionClosed`` after the CDP connection is closed. It
|
is safe to call this multiple times.
|
N)rY ÚacloserG r r r r£
s zCdpConnection.aclose©rd c C s. | |¡I dH }t| |V W 5 Q R X dS )a This context manager opens a session and enables the "simple" style
|
of calling CDP APIs.
|
|
For example, inside a session context, you can call ``await
|
dom.get_document()`` and it will execute on the current session
|
automatically.
|
N)Úconnect_sessionr+ )rD r[ r8 r r r Ú open_session s
|
zCdpConnection.open_sessionr c à s6 | tj |d¡¡I dH }t| j||}|| j|< |S )zLReturns a new :class:`CdpSession` connected to the specified
|
target.TN)ru r
|
ÚtargetZattach_to_targetr rY r¢ )rD r[ rZ r8 r r r r¥ s
|
z CdpConnection.connect_sessionc à sÖ z| j ¡ I dH }W n tk
|
r, Y qÒY nX zt |¡}W n& tjk
|
rb tdd|dY nX t d|¡ d|krÆt j
|
|d ¡}z| j | }W n" t k
|
r¸ td|Y nX | |¡ q | |¡ q dS )zxRuns in the background and handles incoming messages: dispatching
|
responses to commands and events to listeners.NiDÿÿz Client received invalid JSON)r@ rA rB zReceived message %rrf z/Browser sent a message for an invalid session: )rY Z get_messagerm rj ÚloadsÚJSONDecodeErrorr? r r$ r
|
r§ Z SessionIDr¢ r r )rD rA rB rZ r8 r r r Ú _reader_task¦ s"
|
zCdpConnection._reader_task)r# rI rJ rK rF r£ r rU Ú AsyncIteratorr r¦ r¥ rª r r r r r r¡ q s r¡ r¤ c
|
C sf t ¡ 4 I dH F}t|| I dH }z t| |V W 5 Q R X W 5 | ¡ I dH X W 5 Q I dH R X dS )aè This async context manager opens a connection to the browser specified
|
by ``url`` before entering the block, then closes the connection when the
|
block exits.
|
|
The context manager also sets the connection as the default
|
connection for the current task, so that commands like ``await
|
target.get_targets()`` will run on this connection automatically. If
|
you want to use multiple connections concurrently, it is recommended
|
to open each on in a separate task.
|
N)rh Z open_nurseryÚ connect_cdpr£ r) )ÚurlÚnurseryÚconnr r r Úopen_cdpà s
|
r° c à s, t | |tdI dH }t|}| |j¡ |S )a¯ Connect to the browser specified by ``url`` and spawn a background task
|
in the specified nursery.
|
|
The ``open_cdp()`` context manager is preferred in most situations.
|
You should only use this function if you need to specify a custom
|
nursery. This connection is not automatically closed! You can either
|
use the connection object as a context manager (``async with
|
conn:``) or else call ``await conn.aclose()`` on it when you are
|
done with it. If ``set_context`` is True, then the returned
|
connection will be installed as the default connection for the
|
current task. This argument is for unusual use cases, such as
|
running inside of a notebook.
|
)Zmax_message_sizeN)r ÚMAX_WS_MESSAGE_SIZEr¡ Z
|
start_soonrª )r® r rY Zcdp_connr r r r¬ Ù s r¬ )2r; r r] rj r! r rU Ú collectionsr Ú
|
contextlibr r Z dataclassesr rh Ztrio_websocketr rm r r" r ÚTypeVarr r± r
|
r r( r< r* rW r, r2 r3 r) r+ r= r> rq r? rL rR rS rX r ÚabcZ AsyncResourcer¡ r« r° r¬ r r r r Ú<module> sR
|
|
|
|
|
|
|
s@R
|
