/JfuP ddlZddlZddlZddlZddlmZmZmZddlm Z ej e Z GddZGddZGdd ZGd d eZGd d eZGddZGddZdS)N) ensure_bytesensure_unicodeurlparse) EXCEPTION_MAPc(eZdZgdZdZdZdZdS)Monitor)zbefore-parameter-buildzrequest-createdzresponse-receivedz after-callzafter-call-errorc"||_||_dS)zAbstraction for monitoring clients API calls :param adapter: An adapter that takes event emitter events and produces monitor events :param publisher: A publisher for generated monitor events N)_adapter _publisher)selfadapter publishers V/home/alex/cs2snipeproduction/venv/lib/python3.11/site-packages/botocore/monitoring.py__init__zMonitor.__init__!s  #cP|jD]}|||jdS)z(Register an event emitter to the monitorN)_EVENTS_TO_REGISTER register_lastcapture)r event_emitterevent_to_registers rregisterzMonitor.register,s?!%!9 I I   ' '(94< H H H H I Irc |j||}|r|j|dSdS#t$r)}t d||dYd}~dSd}~wwxYw)zCaptures an incoming event from the event emitter It will feed an event emitter event to the monitor's adaptor to create a monitor event and then publish that event to the monitor's publisher. z:Exception %s raised by client monitor in handling event %sT)exc_infoN)r feedr publish Exceptionloggerdebug)r event_namepayload monitor_eventes rrzMonitor.capture1s  M..z7CCM 7'' 66666 7 7    LLL           s7= A0A++A0N)__name__ __module__ __qualname__rrrrrrrrsV $ $ $III rrcdeZdZejfdZdZdZdZdZdZ dZ dZ d Z d Z d Zd Zd S)MonitorEventAdapterc||_dS)zAdapts event emitter events to produce monitor events :type time: callable :param time: A callable that produces the current time N)_time)r times rrzMonitorEventAdapter.__init__Es  rc8||di|S)aFeed an event emitter event to generate a monitor event :type emitter_event_name: str :param emitter_event_name: The name of the event emitted :type emitter_payload: dict :param emitter_payload: The payload to associated to the event emitted :rtype: BaseMonitorEvent :returns: A monitor event based on the event emitter events fired r') _get_handler)r emitter_event_nameemitter_payloads rrzMonitorEventAdapter.feedMs)5t  !344GGGGGrct|d|ddddzS)N_handle_.r-_)getattrsplitreplace)r r s rr.z MonitorEventAdapter._get_handler]sA *z//44Q7??SIII   rc rt|jj|j||d<dS)Nservice operation timestampcurrent_api_call_event) APICallEvent service_model service_id wire_name_get_current_time)r modelcontextkwargss r_handle_before_parameter_buildz2MonitorEventAdapter._handle_before_parameter_buildbs?,8'2o,,..- - - ()))rc |j}|d|}|j|_|j|_||d<dS)Nr>)r=current_api_call_attempt_event)rEnew_api_call_attemptrCheadersrequest_headersurl)r requestrFrEnew_attempt_events r_handle_request_createdz+MonitorEventAdapter._handle_request_createdis_/# $  )?)?)A)A  B B -4O) ' 4E0111rc |d}|||_|A|dd|_|dd|_|d|_n||_|S)NrIResponseMetadataHTTPStatusCode HTTPHeadersError)pop _get_latencylatencyhttp_status_coderesponse_headersget parsed_errorwire_exception)r parsed_responserE exceptionrF attempt_events r_handle_response_receivedz-MonitorEventAdapter._handle_response_receivedrs $DEE $ 1 1- @ @   &-<".. M *.="..M **9)<))r[retries_exceeded_complete_api_call)r rEparsedrFs r_handle_after_callz&MonitorEventAdapter._handle_after_callsA=C > #"E * * ():&&w///rc l|||d_||SNr>)_is_retryable_exceptionrdre)r rEr_rFs r_handle_after_call_errorz,MonitorEventAdapter._handle_after_call_errors9"99)DD  $ &&w///rcRt|ttdS)NGENERAL_CONNECTION_ERROR) isinstancetupleRETRYABLE_EXCEPTIONS)r r_s rrjz+MonitorEventAdapter._is_retryable_exceptions( u12LMNN   rcd|d}|||_|Sri)rVrWrX)r rE call_events rrez&MonitorEventAdapter._complete_api_calls0[[!9:: !..z:: rc:||jz SN)rCr=r events rrWz MonitorEventAdapter._get_latencys%%''%/99rcJt|dzS)Ni)intr+r s rrCz%MonitorEventAdapter._get_current_times4::<<$&'''rN)r$r%r&r,rrr.rGrPrargrkrjrerWrCr'rrr)r)Ds IHHH       FFF"000 000    :::(((((rr)c eZdZdZdZdZdS)BaseMonitorEventc0||_||_||_dS)aBase monitor event :type service: str :param service: A string identifying the service associated to the event :type operation: str :param operation: A string identifying the operation of service associated to the event :type timestamp: int :param timestamp: Epoch time in milliseconds from when the event began Nr:)r r;r<r=s rrzBaseMonitorEvent.__init__s ""rc0|jjd|jdS)N()) __class__r$__dict__rys r__repr__zBaseMonitorEvent.__repr__s .)>>DM>>>>rcPt||jr|j|jkSdS)NF)rnrr)r others r__eq__zBaseMonitorEvent.__eq__s* eT^ , , 3=EN2 2urN)r$r%r&rrrr'rrr{r{sA###$???rr{c,eZdZ dfd ZdZxZS)r?NFct|||||_||_|g|_||_dS)aMonitor event for a single API call This event corresponds to a single client method call, which includes every HTTP requests attempt made in order to complete the client call :type service: str :param service: A string identifying the service associated to the event :type operation: str :param operation: A string identifying the operation of service associated to the event :type timestamp: int :param timestamp: Epoch time in milliseconds from when the event began :type latency: int :param latency: The time in milliseconds to complete the client call :type attempts: list :param attempts: The list of APICallAttempts associated to the APICall :type retries_exceeded: bool :param retries_exceeded: True if API call exceeded retries. False otherwise r:N)superrrXattemptsrd)r r;r<r=rXrrdrs rrzAPICallEvent.__init__sYH yI        DM 0rcrt|j|j|}|j||S)zInstantiates APICallAttemptEvent associated to the APICallEvent :type timestamp: int :param timestamp: Epoch time in milliseconds to associate to the APICallAttemptEvent r:)APICallAttemptEventr;r<rappend)r r=r`s rrJz!APICallEvent.new_api_call_attemptsA,LDNi    ]+++r)NNF)r$r%r&rrJ __classcell__rs@rr?r?sX +1+1+1+1+1+1Z       rr?c.eZdZ dfd ZxZS)rNc t|||||_||_||_||_||_| |_| |_dS)aMonitor event for a single API call attempt This event corresponds to a single HTTP request attempt in completing the entire client method call. :type service: str :param service: A string identifying the service associated to the event :type operation: str :param operation: A string identifying the operation of service associated to the event :type timestamp: int :param timestamp: Epoch time in milliseconds from when the HTTP request started :type latency: int :param latency: The time in milliseconds to complete the HTTP request whether it succeeded or failed :type url: str :param url: The URL the attempt was sent to :type http_status_code: int :param http_status_code: The HTTP status code of the HTTP response if there was a response :type request_headers: dict :param request_headers: The HTTP headers sent in making the HTTP request :type response_headers: dict :param response_headers: The HTTP headers returned in the HTTP response if there was a response :type parsed_error: dict :param parsed_error: The error parsed if the service returned an error back :type wire_exception: Exception :param wire_exception: The exception raised in sending the HTTP request (i.e. ConnectionError) r:N) rrrXrMrYrLrZr\r]) r r;r<r=rXrMrYrLrZr\r]rs rrzAPICallAttemptEvent.__init__slr yI      0. 0(,r)NNNNNNN)r$r%r&rrrs@rrrs^  B-B-B-B-B-B-B-B-B-B-rrceZdZdZdZdZdZdZddddZe j d e j d d Z gd Z d Z dZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZd Z d!Z!d"Z"d#Z#d$Z$d%Z%d&S)' CSMSerializeriXAmznRequestId XAmzRequestIdXAmzId2)zx-amzn-requestidzx-amz-request-idz x-amz-id-2zSAWS4-HMAC-SHA256 Credential=(?P\w+)/\d+/(?P[a-z0-9-]+)/zAWS (?P\w+):)v4s3) r;r<r=rrXrdrMrLrYrZr\r]c>||||_dS)zSerializes monitor events to CSM (Client Side Monitoring) format :type csm_client_id: str :param csm_client_id: The application identifier to associate to the serialized events N)_validate_client_id csm_client_idr rs rrzCSMSerializer.__init__ds&   ///*rclt||jkrtd|d|jddS)Nz&The value provided for csm_client_id: z exceeds the maximum length of z characters)len_MAX_CLIENT_ID_LENGTH ValueErrorrs rrz!CSMSerializer._validate_client_idns^ }   : : :QQQ)-)CQQQ  ; :rc||}||}||d<|jD]4}t||d}|t|d|z|||5t t j|dS)zSerializes a monitor event to the CSM format :type event: BaseMonitorEvent :param event: The event to serialize to bytes :rtype: bytes :returns: The CSM serialized form of the event TypeN _serialize_) event_type),:) separators)_get_base_event_dict_get_event_type_SERIALIZEABLE_EVENT_PROPERTIESr6rjsondumps)r rv event_dictrattrvalues r serializezCSMSerializer.serializeus..u55 ))%00 ' 68  DE4..E 3md233:*DJzjIIIJJJrcd|jdS)N)VersionClientId)rrus rrz"CSMSerializer._get_base_event_dicts*   rc ||d<dS)NServicer')r r;rrFs r_serialize_servicez CSMSerializer._serialize_services ' 9rc ||d<dS)NApir')r r<rrFs r_serialize_operationz"CSMSerializer._serialize_operations% 5rc ||d<dS)N Timestampr')r r=rrFs r_serialize_timestampz"CSMSerializer._serialize_timestamps"+ ;rc jt||d<|r|||ddSdS)N AttemptCount)r_add_fields_from_last_attempt)r rrrFs r_serialize_attemptsz!CSMSerializer._serialize_attemptssG%(]] >"  I  . .z8B< H H H H H I IrcB|jr>||j}|||d<||j|d<|j |j|d<|j||j|d|j||j|ddSdS)NRegion UserAgentFinalHttpStatusCodeApiCall)rL _get_region_get_user_agentrYr\_serialize_parsed_errorr]_serialize_wire_exception)r r last_attemptregions rrz+CSMSerializer._add_fields_from_last_attempts  ' %%l&BCCF!'- 8$&*&:&:,''J{ #  ( 40<0MJ, -  $ 0  ( ():y     & 2  * *+Z      3 2rc:|dkr||d<dS|dkr||d<dSdS)NrLatencyApiCallAttemptAttemptLatencyr')r rXrrs r_serialize_latencyz CSMSerializer._serialize_latencysB  " "$+Jy ! ! ! + + ++2J' ( ( (, +rc |rdnd|d<dS)NrrMaxRetriesExceededr')r rdrrFs r_serialize_retries_exceededz)CSMSerializer._serialize_retries_exceededs1A+G11a '(((rc 4t|j|d<dS)NFqdn)rnetloc)r rMrrFs r_serialize_urlzCSMSerializer._serialize_urls%c]]1 6rc |||d<||r|||d<||}|||d<d|vr |d|d<dSdS)Nr AccessKeyrzX-Amz-Security-Token SessionToken)r _is_signed_get_access_keyr)r rLrrFrs r_serialize_request_headersz(CSMSerializer._serialize_request_headerss#'"6"6"G"G ; ??? + + L&*&:&:?&K&KJ{ #!!/22  #)Jx !_ 4 4)8&*J~ & & & 5 4rc ||d<dS)NHttpStatusCoder')r rYrrFs r_serialize_http_status_codez)CSMSerializer._serialize_http_status_codes(8 #$$$rc b|jD]\}}||vr ||||<dSrt)"_RESPONSE_HEADERS_TO_EVENT_ENTRIESitems)r rZrrFheaderentrys r_serialize_response_headersz)CSMSerializer._serialize_response_headerssL"DJJLL = =MFE)))$4V$< 5! = =rc |dkrdnd}||d|j||dz<||d|j||dz<dS)NrFinalCode AwsExceptionMessageAwsExceptionMessage) _truncate_MAX_ERROR_CODE_LENGTH_MAX_MESSAGE_LENGTH)r r\rrrF field_prefixs rrz%CSMSerializer._serialize_parsed_errorsu#- "9"9wwr 48NN  $"=5 5 <.01<@>>  #T%=< < <"77888rc |dkrdnd}||jj|j||dz<|t ||j||dz<dS)Nrrr SdkExceptionSdkExceptionMessage)rrr$_MAX_EXCEPTION_CLASS_LENGTHstrr)r r]rrrFrs rrz'CSMSerializer._serialize_wire_exceptionsy#- "9"9wwr 48NN  $ -t/O5 5 <.01<@>>   !9< < <"77888rcbt|trdSt|trdSdS)Nrr)rnr?rrus rrzCSMSerializer._get_event_types< e\ * * $9 2 3 3 $## $ $rc||}||\}}|dS)N access_key)_get_auth_value_get_auth_matchgroup)r rLauth_valr5 auth_matchs rrzCSMSerializer._get_access_keys@''88,,X66 : ---rc||sdS||}||\}}|dkrdS|dS)Nrsigning_region)rrrr)r rLrsignature_versionrs rrzCSMSerializer._get_regionsl// 4''88(,(<(???rc~|jD]"\}}||}|r||fcS#dS)N)NN) _AUTH_REGEXSrmatch)r rrregexrs rrzCSMSerializer._get_auth_matchsZ(,(9(?(?(A(A 0 0 $ uKK))E 0(%//// 0zrcxt||kr&td|||d|S|S)Nz6Truncating following value to maximum length of %s: %s)rrr)r text max_lengths rrzCSMSerializer._truncatesH t99z ! ! LLK      $ $ rN)&r$r%r&rrrrrrrecompilerrrrrrrrrrrrrrrrrrrrrrrrrrrr'rrrrBs"%  ,+**& bj /  bj455 L ' ' '#+++KKK(   (((&&&,,,III *333 HHH 222   888 ===        $$$ ... 222   222@@@rrceZdZdZdZdZdS)SocketPublisheri c4||_||f|_||_dS)a)Publishes monitor events to a socket :type socket: socket.socket :param socket: The socket object to use to publish events :type host: string :param host: The host to send events to :type port: integer :param port: The port on the host to send events to :param serializer: The serializer to use to serialize the event to a form that can be published to the socket. This must have a `serialize()` method that accepts a monitor event and return bytes N)_socket_address _serializer)r sockethostport serializers rrzSocketPublisher.__init__%s$" t  %rc |j|}t||jkr0tdt||jdS|j||jdS)zPublishes a specified monitor event :type event: BaseMonitorEvent :param event: The monitor event to be sent over the publisher's socket to the desired address. z`Serialized event of size %s exceeds the maximum length allowed: %s. Not sending event to socket.N) rrr_MAX_MONITOR_EVENT_LENGTHrrrsendtor)r rvserialized_events rrzSocketPublisher.publish:s +55e<<  4#A A A LL<$%%.     F ,dm<<<<r(s  BBBBBBBBBBGGGGGG  8 $ $))))))))X](](](](](](](](@899999#999xC-C-C-C-C-*C-C-C-L]]]]]]]]@(=(=(=(=(=(=(=(=(=(=r