hgoDUddlmZmZddlZddlmZddlmZm Z ddl m Z ddl Z ddl Z ddlZddlZddlZddlZejdZejdZdZdadad Ze jd ae jed <e jd ae jed <dZdZedZ edZ!dZ"dZ#Gdde$Z%GddeZ&Gdde$Z'e GddZ(GddZ)Gdde)Z*Gd d!e)ej+j,Z-e d"ej.e-fd#Z/d"e-fd$Z0dS)%)ConnectionClosedconnect_websocket_urlN) defaultdict)contextmanagerasynccontextmanager) dataclasstrio_cdpTicb|atjdtadS)Nz&selenium.webdriver.common.devtools.v{})version importlib import_moduleformatdevtools)vers [/var/www/fb-scrape/myenv/lib/python3.11/site-packages/selenium/webdriver/common/bidi/cdp.pyimport_devtoolsr3s-G&'O'V'VW^'_'_``HHHconnection_context_connection_contextsession_context_session_contextcv tS#t$rt|dwxYw)z Look 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.)rget LookupError RuntimeErrorfn_names rget_connection_contextr>sP S"&&((( SSSgQQQRRRS8cv tS#t$rt|dwxYw)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.)rrrrrs rget_session_contextr"IsP P##%%% PPPgNNNOOOPr c#Kt|} dVt|dS#t|wxYw)zd This context manager installs ``connection`` as the session context for the current Trio task. N)rsetreset) connectiontokens rrrTs^ # #J / /E) !!%(((((!!%(((( >Ac#Kt|} dVt|dS#t|wxYw)za This context manager installs ``session`` as the session context for the current Trio task. N)rr$r%)sessionr's rrr_s^   ) )E& u%%%%%u%%%%r(c2tjd|adS)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. rdefaultN) contextvars ContextVarr)r&s rset_global_connectionr0js+&01F9CEEErc2tjd|adS)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. rr,N)r.r/r)r*s rset_global_sessionr2us!#-.@'RRRrceZdZdZdZdZdS) BrowserErrorzi This exception is raised when the browser's response to a command indicates that an error occurred. cn|d|_|d|_|d|_dS)Ncodemessagedata)r6r7rdetail)selfobjs r__init__zBrowserError.__init__s-K 9~ ggfoo rcNd|j|j|jS)Nz#BrowserError {})rr6r7r9r:s r__str__zBrowserError.__str__s*4;;DI<@L$+WW WrN)__name__ __module__ __qualname____doc__r<r?rrr4r4sA**&&& WWWWWrr4ceZdZdZdZdZdS)CdpConnectionClosedzC Raised when a public method is called on a closed CDP connection. c||_dS)zn Constructor. :param reason: :type reason: wsproto.frame_protocol.CloseReason N)reason)r:rHs rr<zCdpConnectionClosed.__init__s  rcLd|jj|jS)z Return representation. z{}<{}>)r __class__r@rHr>s r__repr__zCdpConnectionClosed.__repr__st~6 DDDrN)r@rArBrCr<rKrDrrrFrFs=MMEEEEErrFceZdZdZdS) InternalErrorzh This exception is only raised when there is faulty logic in TrioCDP or the integration with PyCDP. N)r@rArBrCrDrrrMrMs$$$$rrMc,eZdZUdZdZejed<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. Nvalue)r@rArBrCrPtypingAny__annotations__rDrrrOrOs0%%E6:rrOceZdZdZdejeeejfdefdZ dddZ e ddej edej ed ffd Zd Zd Zd Zd S)CdpBasec||_||_||_tt|_t j|_t|_ t|_ dS)N) ws session_id target_idrr$channels itertoolscountid_iterdict inflight_cmdinflight_result)r:rWrXrYs rr<zCdpBase.__init__sS$"#C((  ((  FF#vvrcmdreturnc Kt|j}tj}||f|j|<t|}||d<|jr |j|d<t j|} |j |d{Vn'#t$r}t|j dd}~wwxYw| d{V|j|}t!|t"r||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]trioEventr_rXjsondumpsrW send_messageWsConnectionClosedrFrHwaitr`pop isinstance Exception)r:racmd_id cmd_eventrequest request_strwccresponses rexecutezCdpBase.executes! dl##JLL $'N&!s)) ? 3#'?GK j))  <'&&{33 3 3 3 3 3 3 3 3! < < <%cj11t ; <nn'++F33 h * * Ns0 B B5B00B5 ) buffer_sizec~tj|\}}|D]"}|j||#|S)zZ Return an async iterator that iterates over events matching the indicated types. )rgopen_memory_channelrZadd)r:ry event_typessenderreceiver event_types rlistenzCdpBase.listensK 3K@@% 2 2J M* % ) )& 1 1 1 1rrNc8Ktj|\}}|j||t }|WV|4d{V|d{V}dddd{Vn#1d{VswxYwY||_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)rgr{rZr|rOreceiverP)r:rryr~rproxyevents rwait_forzCdpBase.wait_fors/ 3K@@ j!%%f---  - - - - - - - -"**,,,,,,,,E - - - - - - - - - - - - - - - - - - - - - - - - - - - sB B B cfd|vr||dS||dS)z] Handle incoming WebSocket data. :param dict data: a JSON dictionary rdN)_handle_cmd_response _handle_event)r:r8s r _handle_datazCdpBase._handle_datasA 4<<  % %d + + + + +   t $ $ $ $ $rc|d} |j|\}}n>#t$r1td|YdSwxYwd|vrt |d|j|<nS ||d}td#t$r}|j }Yd}~nd}~wwxYw||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 rdz7Got a message with a command ID that does not exist: {}Nerrorresultzs r dom_enablezCdpSession.dom_enable7s<( : : : : : : : :  " "a ' " "%**ll8<#6#6#8#8999999999 : : : : : : : : : : : : : : : : : : : : : : : : : : : ( ; ; ; ; ; ; ; ;  " "a ' " "%**ll8<#7#7#9#9::::::::: ; ; ; ; ; ; ; ; ; ; ; ; ; ; ; ; ; ; ; ; ; ; ; ; ; ; ; ; ; ;%AA55 A?A?AC>> D Dc"K|j4d{V|xjdz c_|jdkr7|tjd{Vdddd{Vn#1d{VswxYwYdWV|j4d{V|xjdzc_|jdkr7|tjd{Vdddd{VdS#1d{VswxYwYdS)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. Nrr)rrrwrpagerrr>s r page_enablezCdpSession.page_enableLs<) ; ; ; ; ; ; ; ;  # #q ( # #&!++ll8=#7#7#9#9::::::::: ; ; ; ; ; ; ; ; ; ; ; ; ; ; ; ; ; ; ; ; ; ; ; ; ; ; ; ) < < < < < < < <  # #q ( # #&!++ll8=#8#8#:#:;;;;;;;;; < < < < < < < < < < < < < < < < < < < < < < < < < < < < < s rrzCdpConnection.aclosevs2gnnrrbcK||d{V}t|5|WVddddS#1swxYwYdS)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)r:rYr*s r open_sessionzCdpConnection.open_sessions,,Y77777777 W % %  MMMM                  sAAArcK|tj|dd{V}t |j||}||j|<|S)zV Returns a new :class:`CdpSession` connected to the specified target. TN)rwrtargetattach_to_targetrrWr)r:rYrXr*s rrzCdpConnection.connect_sessionsv  <<(H(H t)) TWj)<<$+ j!rcDK |jd{V}n#t$rYdSwxYw tj|}n&#tj$rt dd|dwxYwtd|d|vrztj |d} |j |}n0#t$r#t d|wxYw||n||) z Runs in the background and handles incoming messages: dispatching responses to commands and events to listeners. TNiDzClient received invalid JSON)r6r7r8zReceived message %rrez3Browser sent a message for an invalid session: {!r})rW get_messagerlriloadsJSONDecodeErrorr4rrrr SessionIDrrrr)r:r7r8rXr*s r _reader_taskzCdpConnection._reader_tasks|  (  $ 3 3 5 5555555%        z'**'   ""=#$$  LL. 5 5 5d""%_66tK7HII K"mJ7GGKKK&(77=vj7I7IKKKK$$T****!!$'''7 (s#% 33A #A/7 C-C2)rbr)r@rArBrCr<rrrQ AsyncIteratorrrrrrrs@rrrbs         ,        !(!(!(!(!(!(!(rrrbcKtj4d{V}t||d{V} t|5|WVdddn #1swxYwY|d{Vn#|d{VwxYw dddd{VdS#1d{VswxYwYdS)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)rg open_nursery connect_cdprr)urlnurseryconns ropen_cdprs ""       g #........ #D))                  ++--        $++--                                        sLB.A;A A;A A;A A; B.;BB.. B8;B8cKt||td{V}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. )max_message_sizeN)rMAX_WS_MESSAGE_SIZEr start_soonr)rrrWcdp_conns rrrss%Wc6IKKK K K K K K KBR  H x,--- Or)1trio_websocketrrlrrg collectionsr contextlibrr dataclassesrr.r r[riloggingrQ getLoggerrTypeVarr rrr rr/rrSrrr"rrr0r2rpr4rFrMrOrUrabc AsyncResourcerrrrrDrrrsr6 ######<<<<<<<<!!!!!!    : & &FN3  aaa/Ek.DEY.Z.Z[+ZZZ+A;+ABS+T+T+(TTTSSSPPP)))&&&EEESSS W W W W W9 W W W E E E E E, E E E $$$$$I$$$   u4u4u4u4u4u4u4u4p=<=<=<=<=<=<=<=<@X(X(X(X(X(GTX3X(X(X(v 6/ >    (}r