PeuPddlZddlZddlZddlZddlmZmZmZddlm Z eje Z GddZGddZGdd ZGd d eZGd d eZGddZGddZy)N) ensure_bytesensure_unicodeurlparse) EXCEPTION_MAPc&eZdZgdZdZdZdZy)Monitor)zbefore-parameter-buildzrequest-createdzresponse-receivedz after-callzafter-call-errorc ||_||_y)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 5/usr/lib/python3/dist-packages/botocore/monitoring.py__init__zMonitor.__init__!s  #c^|jD]}|j||j y)z(Register an event emitter to the monitorN)_EVENTS_TO_REGISTER register_lastcapture)r event_emitterevent_to_registers rregisterzMonitor.register,s/!%!9!9 I   ' '(94<< H Irc |jj||}|r|jj|yy#t$r#}t j d||dYd}~yd}~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.capture1si  MM..z7CM'' 6  LLL    s9= A)A$$A)N)__name__ __module__ __qualname__rrrrrrrrs $I rrcleZdZejfdZdZdZdZdZdZ dZ dZ d Z d Z d Zd Zy )MonitorEventAdapterc||_y)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  rc0|j|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  !34GGGrcdt|d|jddjddzS)N_handle_.r-_)getattrsplitreplace)r r s rr.z MonitorEventAdapter._get_handler]s6 *z//4Q7??SII  rc ~t|jj|j|j |d<y)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_buildbs4,8''22oo,,.- ()rc |j}|dj|j}|j|_|j |_||d<y)Nr>)r=current_api_call_attempt_event)rEnew_api_call_attemptrCheadersrequest_headersurl)r requestrFrEnew_attempt_events r_handle_request_createdz+MonitorEventAdapter._handle_request_createdisY//# $  )?)?)A  B -4OO) ' 4E01rc |jd}|j||_|2|dd|_|dd|_|j d|_|S||_|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 $DE $ 1 1- @   &-<".. M *.="..M **9)<))r[retries_exceeded_complete_api_call)r rEparsedrFs r_handle_after_callz&MonitorEventAdapter._handle_after_calls;=C > #"E * ():&&w//rc V|j||d_|j|SNr>)_is_retryable_exceptionrdre)r rEr_rFs r_handle_after_call_errorz,MonitorEventAdapter._handle_after_call_errors3"99)D  $ &&w//rc:t|ttdS)NGENERAL_CONNECTION_ERROR) isinstancetupleRETRYABLE_EXCEPTIONS)r r_s rrjz+MonitorEventAdapter._is_retryable_exceptions  u12LMN  rcT|jd}|j||_|Sri)rVrWrX)r rE call_events rrez&MonitorEventAdapter._complete_api_calls*[[!9: !..z: rc<|j|jz SN)rCr=r events rrWz MonitorEventAdapter._get_latencys%%'%//99rc:t|jdzS)Ni)intr+r s rrCz%MonitorEventAdapter._get_current_times4::<$&''rN)r$r%r&r,rrr.rGrPrargrkrjrerWrCr'rrr)r)DsG IIH   F"0 0  :(rr)ceZdZdZdZdZy)BaseMonitorEventc.||_||_||_y)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 ""rcN|jjd|jdS)N()) __class__r$__dict__rys r__repr__zBaseMonitorEvent.__repr__s%..))*!DMM+>rcbt||jr|j|jk(Sy)NF)rnrr)r others r__eq__zBaseMonitorEvent.__eq__s& eT^^ ,==ENN2 2rN)r$r%r&rrrr'rrr{r{s#$?rr{c,eZdZ dfd ZdZxZS)r?cft||||||_||_|g|_||_y)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__sDH yI      DM 0rct|j|j|}|jj ||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_attempts7,LLDNNi  ]+r)NNF)r$r%r&rrJ __classcell__rs@rr?r?s +1Z rr?c.eZdZ dfd ZxZS)rc t ||||||_||_||_||_||_| |_| |_y)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__sWr yI    0. 0(,r)NNNNNNN)r$r%r&rrrs@rrrs&  B-B-rrceZdZdZdZdZdZdZddddZe jd e jd d Z gd Z d Z dZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZd Z d!Z!d"Z"d#Z#d$Z$d%Z%y&)' 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]c4|j|||_y)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||jkDrtd|d|jdy)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_idnsH }  : : :8H))-)C)C(DKQ  ;rc|j|}|j|}||d<|jD]*}t||d}|t|d|z|||,t 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..u5 ))%0 ' 688 DE4.E 3md23:*  DJJzjIJJrc d|jdS)N)VersionClientId)rrus rrz"CSMSerializer._get_base_event_dicts**  rc ||d<y)NServicer')r r;rrFs r_serialize_servicez CSMSerializer._serialize_services ' 9rc ||d<y)NApir')r r<rrFs r_serialize_operationz"CSMSerializer._serialize_operations % 5rc ||d<y)N Timestampr')r r=rrFs r_serialize_timestampz"CSMSerializer._serialize_timestamps "+ ;rc Pt||d<|r|j||dyy)N AttemptCount)r_add_fields_from_last_attempt)r rrrFs r_serialize_attemptsz!CSMSerializer._serialize_attemptss,%(] >"   . .z8B< H rcx|jr@|j|j}|||d<|j|j|d<|j|j|d<|j|j |j|d|j |j|j |dyy)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&B&BCF!'- 8$&*&:&:,,'J{ #  ( ( 40<0M0MJ, -  $ $ 0  ( ()):y   & & 2  * *++Z  3rc0|dk(r||d<y|dk(r||d<yy)NrLatencyApiCallAttemptAttemptLatencyr')r rXrrs r_serialize_latencyz CSMSerializer._serialize_latencys-  "$+Jy ! + ++2J' (,rc |rd|d<yd|d<y)NrrMaxRetriesExceededr')r rdrrFs r_serialize_retries_exceededz)CSMSerializer._serialize_retries_exceededs1A1 '(a '(rc 4t|j|d<y)NFqdn)rnetloc)r rMrrFs r_serialize_urlzCSMSerializer._serialize_urls%c]11 6rc |j||d<|j|r|j||d<|j|}|||d<d|vr |d|d<yy)Nr AccessKeyrzX-Amz-Security-Token SessionToken)r _is_signed_get_access_keyr)r rLrrFrs r_serialize_request_headersz(CSMSerializer._serialize_request_headerssy#'"6"6"G ; ??? +&*&:&:?&KJ{ #!!/2  #)Jx !_ 4)8&*J~ & 5rc ||d<y)NHttpStatusCoder')r rYrrFs r_serialize_http_status_codez)CSMSerializer._serialize_http_status_codes(8 #$rc b|jjD]\}}||vs ||||<yrt)"_RESPONSE_HEADERS_TO_EVENT_ENTRIESitems)r rZrrFheaderentrys r_serialize_response_headersz)CSMSerializer._serialize_response_headerss>"DDJJL =MFE))$4V$< 5! =rc |dk(rdnd}|j|d|j||dz<|j|d|j||dz<y)NrFinalCode AwsExceptionMessageAwsExceptionMessage) _truncate_MAX_ERROR_CODE_LENGTH_MAX_MESSAGE_LENGTH)r r\rrrF field_prefixs rrz%CSMSerializer._serialize_parsed_errorsf#- "9wr 48NN  $"="=5 <.01<@>>  #T%=%=< <"778rc |dk(rdnd}|j|jj|j||dz<|jt ||j ||dz<y)Nrrr SdkExceptionSdkExceptionMessage)rrr$_MAX_EXCEPTION_CLASS_LENGTHstrr)r r]rrrFrs rrz'CSMSerializer._serialize_wire_exceptionsm#- "9wr 48NN  $ $ - -t/O/O5 <.01<@>>  !9!9< <"778rcHt|tryt|tryy)Nrr)rnr?rrus rrzCSMSerializer._get_event_types" e\ * 2 3#4rcn|j|}|j|\}}|jdS)N access_key)_get_auth_value_get_auth_matchgroup)r rLauth_valr5 auth_matchs rrzCSMSerializer._get_access_keys7''8,,X6 : --rc|j|sy|j|}|j|\}}|dk7ry|jdS)Nrsigning_region)rrrr)r rLrsignature_versionrs rrzCSMSerializer._get_regionsT/''8(,(<(??rcz|jjD]\}}|j|}|s||fcSy)N)NN) _AUTH_REGEXSrmatch)r rrregexrs rrzCSMSerializer._get_auth_matchsG(,(9(9(?(?(A 0 $ uKK)E(%// 0rcZt||kDrtjd|||d|S|S)Nz6Truncating following value to maximum length of %s: %s)rrr)r text max_lengths rrzCSMSerializer._truncates7 t9z ! LLK   $ $ rN)&r$r%r&rrrrrrrecompilerrrrrrrrrrrrrrrrrrrrrrrrrrrr'rrrrBs"%  ,+*& bjj / bjj45 L '#+K( (&,I *3 H 2 8 =    $ . 2 2@rrceZdZdZdZdZy)SocketPublisheri c2||_||f|_||_y)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 |jj|}t||jkDr+tj dt||jy|j j||jy)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:sp ++55e<  4#A#A A LL<$%..    ,dmmr(s  BBG   8 $))X](](@89#9xC-*C-L]]@(=(=r