U ^{j0@s<dZddlmZddlmZddlmZmZmZm Z m Z m Z edee efZ edee efZede e e efZGdd d eZGd d d eZGd d d eZGdddeZGdddeZdddddZdddddZddddddZdddd d!Zdd"dd#d$Zdd"dd%d&Zdd"dd'd(Z d)S)*z~ Provides various implementations of Session and Target. It includes helpers and constants for the Chrome Devtools Protocol. ) annotations)Enum)AnyMutableMappingNewTypeOptionalTuplecastBrowserResponseBrowserCommand MessageKeyc@seZdZdZdZdS)Ecodez9Ecodes are a list of possible error codes chrome returns.iN)__name__ __module__ __qualname____doc__ZTARGET_NOT_FOUNDrr^/var/www/html/bwave/python/venv/lib/python3.8/site-packages/choreographer/protocol/__init__.pyr sr cs(eZdZdZdddfdd ZZS)DevtoolsProtocolErrorz8Raise a general error reported by the devtools protocol.r Noneresponsereturncs,t||dd|_|dd|_dS)z Construct a new DevtoolsProtocolError. Args: response: the json response that contains the error errorcodemessageN)super__init__rr)selfr __class__rrrs zDevtoolsProtocolError.__init__rrrrr __classcell__rrrrrsrcs,eZdZdZdddddfdd ZZS) MessageTypeError8An error for poorly formatted devtools protocol message.strrtyper)keyvalue expected_typerc s:t|tst|n|}td|d|d|ddS)a Construct a message about a poorly formed protocol message. Args: key: the key that has the badly typed value value: the type of the value that is incorrect expected_type: the type that was expected zMessage with key z must have type z, not .N) isinstancer&rr)rr'r(r)rrrr/s zMessageTypeError.__init__r!rrrrr#,sr#cs*eZdZdZddddfdd ZZS)MissingKeyErrorr$r%r r)r'objrcstd|d|dS)z Construct a MissingKeyError specifying which key was missing. Args: key: the missing key obj: the message without the key zMessage missing required key/s z. Message received: N)rr)rr'r-rrrrBs zMissingKeyError.__init__r!rrrrr,?sr,c@seZdZdZdS)ExperimentalFeatureWarningz8An warning to report that a feature may or may not work.N)rrrrrrrrr.Psr.r)r-rcCsd}ttd}|D]<\}}||kr0t||t|||st|t|||q|d7}d|krl|d7}d|kr||d7}t||krt ddS) a Verify the message obj hast he proper keys and values. Args: obj: the object to check. Raises: MissingKeyError: if a key is missing. MessageTypeError: if a value type is incorrect. RuntimeError: if there are strange keys. r)idmethodparams sessionIdzUMessage objects must have id and method keys, and may have params and sessionId keys.N) intr%itemsr,r+r#r&lenkeys RuntimeError)r-Zn_keys required_keysr'Ztype_keyrrr verify_paramsTs   r;z BrowserResponse | BrowserCommandzMessageKey | None)msgrcCs.|dd}|d}|dkr"dSt||fS)z Given a message to/from the browser, calculate the key corresponding to the command. Every message is uniquely identified by its sessionId and id (counter). Args: msg: the message for which to calculate the key. r4r/N)getr )r< session_id message_idrrrcalculate_message_keyws  rAbool)rr'rcCsX|\}}d|kr|dks2d|kr.|d|kr.ndSd|krPt|dt|krPndSdS)z Report True if a response matches with a certain key (sessionId, id). Args: response: the object response from the browser key: the (sessionId, id) key tubple we're looking for r?r=Fr/T)r%)rr'r?r@rrrmatch_message_keys  rCrcCsddh}||kod|kS)z=Return true if the browser response is an event notification.r0r2r/)r8)rr:rrris_eventsrDz str | NonecCs.d|kr&d|dkr&td|ddSdSdS)z Extract target id from a browser response. Args: response: the browser response to extract the targetId from. resultZtargetIdr%Nr rrrrget_target_id_from_resultsrHcCs.d|kr&d|dkr&td|ddSdSdS)z Extract session id from a browser response. Args: response: the browser response to extract the sessionId from. rEr4r%NrFrGrrrget_session_id_from_resultsrIcCsd|krtd|dSdSdS)z~ Extract error from a browser response. Args: response: the browser response to extract the error from. rr%NrFrGrrrget_error_from_resultsrJN)!r __future__renumrtypingrrrrrr r%r r r5r r Exceptionr TypeErrorr# ValueErrorr, UserWarningr.r;rArCrDrHrIrJrrrrs$   #