a
>xd, @ s d Z ddlZddlZddlZddlmZmZmZmZm Z ddl
m Z ddl Z ddl
mZ e rhddlmZ eeZeed Zeed ZG d d
d
e ZG d d d e Zeeeed
ddZeeedddZdS )z%Connection/Session management module. N) AwaitableCallableDictUnion
TYPE_CHECKING) EventEmitter) NetworkError)Optionalz .Connectionz .CDPSessionc s e Zd ZdZd%eejedd fdd
Ze eddd Z
ddd
d Z eedd d
dZ d&ee
edddZe
ddddZe
ddddZeg df ddddZedddd Zddd d Zddd d Zed!d"d#d$Z ZS )'
Connectionz Connection management class.r N)urlloopdelayreturnc st t || _d| _t | _|d | _|| _t | _| d| _ t
j j | jd| jd| _
| j| | _d| _dS )zMake connection.
:arg str url: WebSocket url to connect devtool.
:arg int delay: delay to wait before processing received messages.
r i FN)max_sizer )super__init___url_lastIddict
_callbacks_delay_loop _sessions
_connected
websocketsclientconnect_ws create_task
_recv_loop _recv_fut_closeCallback)selfr r r
__class__ L/var/www/html/Ranjet/env/lib/python3.9/site-packages/pyppeteer/connection.pyr s
zConnection.__init__r c C s | j S )z Get connected WebSocket url.)r r" r% r% r& r 0 s zConnection.urlc
s | j 4 I d H }d| _|| _| jrz(| j I d H }|rH| |I d H W n& tjtfyp t d Y qY n0 t
dI d H q W d I d H q1 I d H s0 Y | jr| j
| d S )NTzconnection closedr )r r
connectionrecv _on_messager ConnectionClosedConnectionResetErrorloggerinfoasynciosleepr r dispose)r" r) respr% r% r& r 5 s
<zConnection._recv_loop)msg callback_idr c s | j st| jI d H q z| j|I d H W nP tjy t d | j
|d }|r|| s||
d | I d H Y n0 d S )Nz connection unexpectedly closed)r r0 r1 r r) sendr r, r. errorr getdone
set_resultr2 )r" r4 r5 callbackr% r% r& _async_sendE s
zConnection._async_sendmethodparamsr c C s | j r| jstd|du r"t }| j d7 _ | j }tt|||d}td| | j |
|| | j }|| j |< t
|_||_|S )z Send message via the connection.zConnection is closedN idr> r? SEND: )r r ConnectionErrorr jsondumpslogger_connectiondebugr r r<
create_futurer r r7 r> )r" r> r? _idr4 r; r% r% r& r6 Q s$
zConnection.sendr4 r c C sJ | j |dd}|dr6|t|j|j| n||d d S )NrB r7 result)r popr8
set_exception_createProtocolErrorr7 r> r: )r" r4 r; r% r% r& _on_responseg s
zConnection._on_responsec C s | di }| dd}| d}|dkrL| j |}|r|| d n6|dkrv| j |}|r| | j|= n | || d S )Nr? r> sessionId Target.receivedMessageFromTargetmessagezTarget.detachedFromTarget)r8 r r+
_on_closedemit)r" r4 r? r> rS sessionr% r% r& _on_queryt s
zConnection._on_query)r; r c C s
|| _ dS )zSet closed callback.N)r! )r" r; r% r% r& setClosedCallback s z Connection.setClosedCallback)rU r c sV t | jI d H td| t|}|d| jv rH| | n
|
| d S )NRECV: rB ) r0 r1 r rG rH rE loadsr8 r rQ rY )r" rU r4 r% r% r& r+ s
zConnection._on_messagec s | j r| d | _ | j D ] }|t|jd|j d q | j | j D ] }| qT| j t
| dr| j I d H | j
s| j
d S )NProtocol error : Target closed.r) )r! r valuesrO
_rewriteErrorr7 r> clearr rV hasattrr) closer r9 cancel)r" cbrX r% r% r& _on_close s
zConnection._on_closec s d| _ | I dH dS )zClose all connection.FN)r rf r( r% r% r& r2 s zConnection.dispose
CDPSession)
targetInfor c sF | dd|d iI dH }|d}t| |d || j}|| j|< |S )zCreate new session.zTarget.attachToTargetZtargetIdNrS type)r6 r8 rg r r )r" rh r3 rS rX r% r% r&
createSession s
zConnection.createSession)r )N)__name__
__module__ __qualname____doc__strr0 AbstractEventLoopintr propertyr r r< r r r6 rQ rY r rZ r+ rf r2 r rj
__classcell__r% r% r# r& r
s"
r
c s e Zd ZdZeed f eeejdd fdd Z dee
e dddZ edd d
d Z
dd d
dZdd ddZeed dddZ ZS )rg a{ Chrome Devtools Protocol Session.
The :class:`CDPSession` instances are used to talk raw Chrome Devtools
Protocol:
* protocol methods can be called with :meth:`send` method.
* protocol events can be subscribed to with :meth:`on` method.
Documentation on DevTools Protocol can be found
`here `_.
N)r)
targetTyperS r r c s: t d| _i | _|| _|| _|| _t | _|| _ dS )zMake new session.r N)
r r r r _connection _targetType
_sessionIdr r r )r" r) rt rS r r# r% r& r s
zCDPSession.__init__r= c
C s | j s td| d| j d| jd7 _| j}tt|||d}td| | j
}|| j |< t |_ ||_
z| j d| j|d W nX ty } z@|| j v r| j |= | j | }|t|j |jd W Y d
}~n
d
}~0 0 |S ) zSend message to the connected session.
:arg str method: Protocol method name.
:arg dict params: Optional method parameters.
zProtocol Error (z#): Session closed. Most likely the z has been closed.r@ rA rC zTarget.sendMessageToTarget)rS rU r N)ru r rv r rE rF r logger_sessionrH r rI r r7 r> r6 rw ExceptionrO r` args)r" r> r? rJ r4 r; e _callbackr% r% r& r6 s8
zCDPSession.sendrK c C s t d| t|}|d}|r| j|}|r| j|= |drb|t|j|j | n |d}|r|
s| | n|di }|ddkr| j |d}|r|
|d n8|dd
kr|d}| j |}|r| | j |= | |d|d d S ) Nr[ rB r7 rM r? r> rT rS rU Target.detachFromTarget)rx rH rE r\ r8 r rO rP r7 r> r9 r: r r+ rV rW ) r" r4 objrJ r; rM r? rX rS r% r% r& r+ s8
zCDPSession._on_messager' c s, | j std| j dd| jiI dH dS )zDetach session from target.
Once detached, session won't emit any events and can't be used to send
messages.
zConnection already closed.r} rS N)ru r r6 rw r( r% r% r& detach s
zCDPSession.detachc C s@ | j D ] }|t|jd|j d q
| j d | _d S )Nr] r^ )r r_ rO r` r7 r> ra ru )r" re r% r% r& rV ! s
zCDPSession._on_closed)rt rS r c C s t | ||| j}|| j|< |S N)rg r r )r" rt rS rX r% r% r& _createSession* s
zCDPSession._createSession)N)rk rl rm rn r r
ro r0 rp r r r r6 r+ r rV r rs r% r% r# r& rg s
$ rg )r7 r> r~ r c C sD d| d|d d }d|d v r:|d|d d 7 }t | |S )NzProtocol error (z): r7 rU data )r` )r7 r> r~ rU r% r% r& rP 0 s rP )r7 rU r c C s |f| _ | S r )rz )r7 rU r% r% r& r` 8 s r` )rn r0 rE loggingtypingr r r r r Zpyeer r Zpyppeteer.errorsr r getLoggerrk r. rG rx r
rg ry ro rP r` r% r% r% r& s&
!x