U ^{j&@sdZddlmZddlZddlmZddlZddlmZerhddl Z ddlm Z m Z m Z m Z ddlmZeeZGdd d ZGd d d ZdS) z?Provide a lower-level async interface to the Devtools Protocol.) annotationsN) TYPE_CHECKING)protocol)AnyCallable CoroutineMutableMapping)Brokerc@seZdZUdZded<ded<ded<ddd d d d Zd ddddddZdddddd dddZdd dddZdddddZ d S)!Sessionz8A session is a single conversation with a single target.str session_idint message_idz`MutableMapping[str, tuple[Callable[[protocol.BrowserResponse], Coroutine[Any, Any, Any]], bool]] subscriptionsr None)r brokerreturncCs>t|tstd||_||_td|d|_i|_dS)a, Construct a session from the browser as an object. A session is like an open conversation with a target. All commands are sent on sessions. Args: broker: a reference to the browser's broker session_id: the id given by the browser zsession_id must be a stringz New session: rN) isinstancer TypeError_brokerr _loggerdebugrr)selfr rrd/var/www/html/bwave/python/venv/lib/python3.8/site-packages/choreographer/protocol/devtools_async.py__init__$s zSession.__init__NMutableMapping[str, Any] | Noneprotocol.BrowserResponsecommandparamsrc s|j}|jd7_t||d}|jr4|j|d<|r@||d<td|d|rX|ndd|jd td t| d d |j |Id HS)a/ Send a devtools command on the session. https://chromedevtools.github.io/devtools-protocol/ Args: command: devtools command to send params: the parameters to send Returns: A message key (session, message id) tuple or None )idmethod sessionIdr zCmd 'z', param keys 'z', sessionId ''z Full params: %z%%N) rrZBrowserCommandr rrkeysZdebug2r replacer write_json)rrr Z current_idZ json_commandrrr send_command;s  $zSession.send_commandT repeating>Callable[[protocol.BrowserResponse], Coroutine[Any, Any, Any]]boolstringcallbackr-rcCs8t|std||jkr&tdn||f|j|<dS)a6 Subscribe to an event on this session. Args: string: the name of the event. Can use * wildcard at the end. callback: the callback (which takes a message dict and returns nothing) repeating: default True, should the callback execute more than once z/Call back must be be `async def` type function.zSYou are already subscribed to this string, duplicate subscriptions are not allowed.N)inspectiscoroutinefunctionrr ValueError)rr1r2r-rrr subscribeas  zSession.subscriber1rcCs||jkrdS|j|=dSzh Remove a subscription. Args: string: the subscription to remove. N)rrr1rrr unsubscribes zSession.unsubscribeasyncio.Future[Any]cCs|j|j|S)a Return a future for a browser event. Generally python asyncio doesn't recommend futures. But in this case, one must call subscribe_once and await it later, generally because they must subscribe and then provoke the event. Args: string: the event to subscribe to Returns: A future to be awaited later, the complete event. )rZnew_subscription_futurer r9rrrsubscribe_onceszSession.subscribe_once)N) __name__ __module__ __qualname____doc____annotations__rr+r6r:r<rrrrr s + r c@seZdZUdZded<ded<ddddd Zd d d d dZdd dddZd dddZd,ddddddZ d dddZ dddddZ dd dd!d"d d#d$d%Z dd d&d'd(Z dd)d&d*d+ZdS)-TargetzLA target like a browser, tab, or others. It sends commands. It has sessions.r target_idzMutableMapping[str, Session]sessionsr )rCrcCs:t|tstd||_i|_||_td|ddS)z Create a target after one ahs been created by the browser. Args: broker: a reference to the browser's broker target_id: the id given by the browser ztarget_id must be stringzCreated new target .N)rr rrrDrCrr)rrCrrrrrs zTarget.__init__r r)sessionrcCs"t|tstd||j|j<dS)Nz$session must be a session type class)rr rrDr rrFrrr _add_sessions zTarget._add_session)r rcCs"t|tr|j}|j|d}dS)N)rr r rDpop)rr _rrr_remove_sessions zTarget._remove_session)rcCs(|jstdtt|j}|S)z7Retrieve the first session of the target, if it exists.z9Cannot use this method without at least one valid session)rDvalues RuntimeErrornextiterrGrrr get_sessions  zTarget.get_sessionNrrrcs,|jstd|}|||IdHS)z Send a command to the first session in a target. https://chromedevtools.github.io/devtools-protocol/ Args: command: devtools command to send params: the parameters to send z6Cannot send_command without at least one valid sessionN)rDrLrMrPr+)rrr rFrrrr+s zTarget.send_commandcs^|jjjd|jdddIdH}d|kr8tdt||dd }t||j}|||S) z$Create a new session on this target.zTarget.attachToTargetT)ZtargetIdflatten)r NerrorzCould not create sessionresultr$) r_browserr+rCrMrDevtoolsProtocolErrorr rH)rresponser Z new_sessionrrrcreate_sessions     zTarget.create_sessioncsdt|tr|j}|jjjdd|idIdH}||d|krNtdt |t d|d|S) zm Close a session by session_id. Args: session_id: the session to close zTarget.detachFromTargetr$)rr NrRzCould not close sessionz The session z has been closed.) rr r rrTr+rKrMrrUrr)rr rVrrr close_sessions"   zTarget.close_sessionTr,r.r/r0cCs|}|j|||ddS)aI Subscribe to an event on the main session of this target. Args: string: the name of the event. Can use * wildcard at the end. callback: the callback (which takes a message dict and returns nothing) repeating: default True, should the callback execute more than once r,N)rPr6)rr1r2r-rFrrrr6szTarget.subscriber7cCs|}||dSr8)rPr:rr1rFrrrr:"szTarget.unsubscriber;cCs|}||S)a Return a future for a browser event for the first session of this target. Generally python asyncio doesn't recommend futures. But in this case, one must call subscribe_once and await it later, generally because they must subscribe and then provoke the event. Args: string: the event to subscribe to Returns: A future to be awaited later, the complete event. )rPr<rYrrrr<-szTarget.subscribe_once)N)r=r>r?r@rArrHrKrPr+rWrXr6r:r<rrrrrBs  " rB)r@ __future__rr3typingrZlogistroZ choreographerrZasynciorrrrZchoreographer._brokersr getLoggerr=rr rBrrrrs