File: //proc/self/root/lib/python3.6/site-packages/botocore/__pycache__/monitoring.cpython-36.pyc
3
�T_jP � @ s� d dl Z d dlZd dlZd dlZd dlmZmZmZ d dlm Z
ej e �Z
G dd� de�ZG dd� de�ZG dd � d e�ZG d
d � d e�ZG d d
� d
e�ZG dd� de�ZG dd� de�ZdS )� N)�ensure_unicode� ensure_bytes�urlparse)�
EXCEPTION_MAPc @ s2 e Zd ZdddddgZdd� Zdd � Zd
d � Zd S )
�Monitorzbefore-parameter-buildzrequest-createdzresponse-receivedz
after-callzafter-call-errorc C s || _ || _dS )z�Abstraction 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)�selfZadapterZ publisher� r
� /usr/lib/python3.6/monitoring.py�__init__" s zMonitor.__init__c C s" x | j D ]}|j|| j� qW dS )z(Register an event emitter to the monitorN)�_EVENTS_TO_REGISTERZ
register_last�capture)r Z
event_emitterZevent_to_registerr
r
r �register- s zMonitor.registerc K s\ y"| j j||�}|r | jj|� W n4 tk
rV } ztjd||dd� W Y dd}~X nX dS )z�Captures 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� Exception�logger�debug)r �
event_nameZpayloadZ
monitor_event�er
r
r r 2 s zMonitor.captureN)�__name__�
__module__� __qualname__r
r r r r
r
r
r r s r c @ sr e Zd Zejfdd�Zdd� Zdd� Zdd� Zd d
� Zd d � Z d
d� Z
dd� Z dd� Z dd� Z
dd� Zdd� ZdS )�MonitorEventAdapterc C s
|| _ dS )z�Adapts event emitter events to produce monitor events
:type time: callable
:param time: A callable that produces the current time
N)�_time)r �timer
r
r r C s z MonitorEventAdapter.__init__c C s | j |�f |�S )a� Feed 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
)� _get_handler)r Zemitter_event_nameZemitter_payloadr
r
r r K s zMonitorEventAdapter.feedc C s t | d|jd�d jdd� �S )NZ_handle_�.r �-�_)�getattr�split�replace)r r r
r
r r [ s z MonitorEventAdapter._get_handlerc K s t |jj|j| j� d�|d< d S )N)�service� operation� timestamp�current_api_call_event)� APICallEventZ
service_modelZ
service_idZ wire_name�_get_current_time)r Zmodel�context�kwargsr
r
r � _handle_before_parameter_build` s z2MonitorEventAdapter._handle_before_parameter_buildc K s6 |j }|d j| j� d�}|j|_|j|_||d< d S )Nr( )r' � current_api_call_attempt_event)r+ �new_api_call_attemptr* Zheaders�request_headers�url)r Zrequestr, r+ Znew_attempt_eventr
r
r �_handle_request_createdg s z+MonitorEventAdapter._handle_request_createdc K sR |j d�}| j|�|_|d k rH|d d |_|d d |_|jd�|_n||_|S )Nr. �ResponseMetadataZHTTPStatusCodeZ HTTPHeaders�Error)�pop� _get_latency�latency�http_status_code�response_headers�get� parsed_error�wire_exception)r Zparsed_responser+ � exceptionr, �
attempt_eventr
r
r �_handle_response_receivedp s
z-MonitorEventAdapter._handle_response_receivedc K s |d j dd�|d _| j|�S )Nr3 ZMaxAttemptsReachedFr( )r: �retries_exceeded�_complete_api_call)r r+ Zparsedr, r
r
r �_handle_after_call~ s z&MonitorEventAdapter._handle_after_callc K s | j |�|d _| j|�S )Nr( )�_is_retryable_exceptionr@ rA )r r+ r= r, r
r
r �_handle_after_call_error� s z,MonitorEventAdapter._handle_after_call_errorc C s t |ttd ��S )NZGENERAL_CONNECTION_ERROR)�
isinstance�tuple�RETRYABLE_EXCEPTIONS)r r= r
r
r rC � s z+MonitorEventAdapter._is_retryable_exceptionc C s |j d�}| j|�|_|S )Nr( )r5 r6 r7 )r r+ Z
call_eventr
r
r rA � s
z&MonitorEventAdapter._complete_api_callc C s | j � |j S )N)r* r' )r �eventr
r
r r6 � s z MonitorEventAdapter._get_latencyc C s t | j� d �S )Ni� )�intr )r r
r
r r* � s z%MonitorEventAdapter._get_current_timeN)r r r r r r r r- r2 r? rB rD rC rA r6 r* r
r
r
r r B s r c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�BaseMonitorEventc C s || _ || _|| _dS )a� Base 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
N)r% r& r' )r r% r& r' r
r
r r � s zBaseMonitorEvent.__init__c C s d| j j| jf S )Nz%s(%r))� __class__r �__dict__)r r
r
r �__repr__� s zBaseMonitorEvent.__repr__c C s t || j�r| j|jkS dS )NF)rE rK rL )r �otherr
r
r �__eq__� s zBaseMonitorEvent.__eq__N)r r r r rM rO r
r
r
r rJ � s rJ c s& e Zd Zd� fdd� Zdd� Z� ZS )r) NFc s: t t| �j|||d� || _|| _|dkr0g | _|| _dS )a� Monitor 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% r& r' N)�superr) r r7 �attemptsr@ )r r% r& r' r7 rQ r@ )rK r
r r � s
zAPICallEvent.__init__c C s" t | j| j|d�}| jj|� |S )z�Instantiates APICallAttemptEvent associated to the APICallEvent
:type timestamp: int
:param timestamp: Epoch time in milliseconds to associate to the
APICallAttemptEvent
)r% r& r' )�APICallAttemptEventr% r& rQ �append)r r' r> r
r
r r/ � s z!APICallEvent.new_api_call_attempt)NNF)r r r r r/ �
__classcell__r
r
)rK r r) � s $r) c s e Zd Zd� fdd� Z� ZS )rR Nc sD t t| �j|||d� || _|| _|| _|| _|| _| | _|
| _ dS )a� Monitor 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% r& r' N)
rP rR r r7 r1 r8 r0 r9 r; r< ) r r% r& r' r7 r1 r8 r0 r9 r; r< )rK r
r r � s 0
z APICallAttemptEvent.__init__)NNNNNNN)r r r r rT r
r
)rK r rR � s rR c @ s& e Zd ZdZdZdZdZdZdddd�Ze j
d �e j
d
�d �Z d d
ddddddddddg Z dd� Z
dd� Zd d � Zd d � Zd d!� Zd"d#� Zd$d%� Zd&d'� Zd(d)� Zd*d+� Zd,d-� Zd.d/� Zd0d1� Zd2d3� Zd4d5� Zd6d7� Z d8d9� Z d:d;� Z d<d=� Z d>d?� Z d@dA� Z!dBdC� Z"dDdE� Z#dFdG� Z$dHdI� Z%dJS )K�
CSMSerializer� � � i ZXAmznRequestIdZ
XAmzRequestIdZXAmzId2)zx-amzn-requestidzx-amz-request-idz
x-amz-id-2zSAWS4-HMAC-SHA256 Credential=(?P<access_key>\w+)/\d+/(?P<signing_region>[a-z0-9-]+)/zAWS (?P<access_key>\w+):)�v4Zs3r% r& r' rQ r7 r@ r1 r0 r8 r9 r; r< c C s | j |� || _dS )z�Serializes 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_id)r r[ r
r
r r O s
zCSMSerializer.__init__c C s$ t |�| jkr td|| jf ��d S )NzTThe value provided for csm_client_id: %s exceeds the maximum length of %s characters)�len�_MAX_CLIENT_ID_LENGTH�
ValueError)r r[ r
r
r rZ Y s z!CSMSerializer._validate_client_idc C sj | j |�}| j|�}||d< x:| jD ]0}t||d�}|dk r$t| d| �|||d� q$W ttj|dd��S ) z�Serializes 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
ZTypeNZ _serialize_)�
event_type�,�:)Z
separators)r` ra )�_get_base_event_dict�_get_event_type� _SERIALIZEABLE_EVENT_PROPERTIESr" r �json�dumps)r rH �
event_dictr_ �attr�valuer
r
r � serializea s
zCSMSerializer.serializec C s d| j d�S )N� )ZVersionZClientId)r[ )r rH r
r
r rb u s z"CSMSerializer._get_base_event_dictc K s ||d< d S )NZServicer
)r r% rg r, r
r
r �_serialize_service{ s z CSMSerializer._serialize_servicec K s ||d< d S )NZApir
)r r& rg r, r
r
r �_serialize_operation~ s z"CSMSerializer._serialize_operationc K s ||d< d S )NZ Timestampr
)r r' rg r, r
r
r �_serialize_timestamp� s z"CSMSerializer._serialize_timestampc K s$ t |�|d< |r | j||d � d S )NZ AttemptCountrk ���)r\ � _add_fields_from_last_attempt)r rQ rg r, r
r
r �_serialize_attempts� s z!CSMSerializer._serialize_attemptsc C s~ |j r2| j|j �}|d k r"||d< | j|j �|d< |jd k rF|j|d< |jd k r`| j|j|d� |jd k rz| j|j|d� d S )N�Region� UserAgentZFinalHttpStatusCode�ApiCall)r0 � _get_region�_get_user_agentr8 r; �_serialize_parsed_errorr< �_serialize_wire_exception)r rg Z last_attempt�regionr
r
r rp � s
z+CSMSerializer._add_fields_from_last_attemptc C s& |dkr||d< n|dkr"||d< d S )Nrt ZLatency�ApiCallAttemptZAttemptLatencyr
)r r7 rg r_ r
r
r �_serialize_latency� s
z CSMSerializer._serialize_latencyc K s |rdnd|d< d S )Nrk r ZMaxRetriesExceededr
)r r@ rg r, r
r
r �_serialize_retries_exceeded� s z)CSMSerializer._serialize_retries_exceededc K s t |�j|d< d S )NZFqdn)r Znetloc)r r1 rg r, r
r
r �_serialize_url� s z CSMSerializer._serialize_urlc K sX | j |�|d< | j|�r&| j|�|d< | j|�}|d k r@||d< d|krT|d |d< d S )Nrs Z AccessKeyrr zX-Amz-Security-TokenZ SessionToken)rv �
_is_signed�_get_access_keyru )r r0 rg r, ry r
r
r �_serialize_request_headers� s
z(CSMSerializer._serialize_request_headersc K s ||d< d S )NZHttpStatusCoder
)r r8 rg r, r
r
r �_serialize_http_status_code� s z)CSMSerializer._serialize_http_status_codec K s0 x*| j j� D ] \}}||kr || ||< q W d S )N)�"_RESPONSE_HEADERS_TO_EVENT_ENTRIES�items)r r9 rg r, �header�entryr
r
r �_serialize_response_headers� s z)CSMSerializer._serialize_response_headersc K sH |dkr dnd}| j |d | j�||d < | j |d | j�||d < d S )Nrt �Final� ZCodeZ AwsExceptionZMessageZAwsExceptionMessage)� _truncate�_MAX_ERROR_CODE_LENGTH�_MAX_MESSAGE_LENGTH)r r; rg r_ r, � field_prefixr
r
r rw � s
z%CSMSerializer._serialize_parsed_errorc K sH |dkr dnd}| j |jj| j�||d < | j t|�| j�||d < d S )Nrt r� r� Z SdkExceptionZSdkExceptionMessage)r� rK r �_MAX_EXCEPTION_CLASS_LENGTH�strr� )r r< rg r_ r, r� r
r
r rx � s z'CSMSerializer._serialize_wire_exceptionc C s t |t�rdS t |t�r dS d S )Nrt rz )rE r) rR )r rH r
r
r rc � s
z CSMSerializer._get_event_typec C s" | j |�}| j|�\}}|jd�S )NZ
access_key)�_get_auth_value�_get_auth_match�group)r r0 �auth_valr! �
auth_matchr
r
r r � s
z CSMSerializer._get_access_keyc C s< | j |�sd S | j|�}| j|�\}}|dkr2d S |jd�S )NrY Zsigning_region)r~ r� r� r� )r r0 r� �signature_versionr� r
r
r ru � s
zCSMSerializer._get_regionc C s | j t|jdd��| j�S )Nz
User-Agentr� )r� r r: �_MAX_USER_AGENT_LENGTH)r r0 r
r
r rv � s z CSMSerializer._get_user_agentc C s d|kS )N�
Authorizationr
)r r0 r
r
r r~ � s zCSMSerializer._is_signedc C s t |d �S )Nr� )r )r r0 r
r
r r� � s z CSMSerializer._get_auth_valuec C s2 x,| j j� D ] \}}|j|�}|r ||fS q W dS )N)NN)� _AUTH_REGEXSr� �match)r r� r� Zregexr� r
r
r r� � s
z CSMSerializer._get_auth_matchc C s* t |�|kr&tjd||� |d |� S |S )Nz6Truncating following value to maximum length of %s: %s)r\ r r )r �textZ
max_lengthr
r
r r� � s zCSMSerializer._truncateN)&r r r r] r� r� r� r� r� �re�compiler� rd r rZ rj rb rl rm rn rq rp r{ r| r} r� r� r� rw rx rc r ru rv r~ r� r� r� r
r
r
r rU + sb
rU c @ s e Zd ZdZdd� Zdd� ZdS ) �SocketPublisher� i c C s || _ ||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 Zsocket�hostZportZ
serializerr
r
r r s
zSocketPublisher.__init__c C sF | j j|�}t|�| jkr2tjdt|�| j� dS | jj|| j� dS )z�Publishes 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) r� rj r\ �_MAX_MONITOR_EVENT_LENGTHr r r� Zsendtor� )r rH Zserialized_eventr
r
r r s zSocketPublisher.publishNi )r r r r� r r r
r
r
r r� � s r� )re Zloggingr� r Zbotocore.compatr r r Zbotocore.retryhandlerr rG Z getLoggerr r �objectr r rJ r) rR rU r� r
r
r
r �<module>
s
)Z 6= U