관리-도구
편집 파일: transport.cpython-38.pyc
U �Mg,� � @ s� d dl mZmZ d dlZd dlZd dlZd dlZd dlZd dlZd dl Z d dl m Z m Z m Z d dl mZ d dlmZ z d dlZW n ek r� dZY nX d dlZd dlZd dlZd dlmZ d dlmZmZmZ d dl m Z d d l m Z m Z m!Z! d d l"m#Z# e#�r�d d l"m$Z$ d d l"m%Z% d d l"m&Z& d dl"m'Z' d dl"m(Z( d dl"m)Z) d dl"m*Z* d dl"m+Z+ d dl"m,Z, d dl"m-Z- d dl"m.Z. d dl"m/Z/ d dl0m1Z1 d dl0m2Z2 d dl3m4Z4m5Z5 g Z6ej7dd� d fej8d d� d fej8d d� d fej8d!d� d"ffD ]@Z9z"e6�:e9d e9d � e9d# f� W n e;k �r8 Y nX �q�G d$d%� d%e�Z<d3d&d'�Z=G d(d)� d)e<�Z>G d*d+� d+e>�Z?z d dl@Z@W n& ek �r� G d,d-� d-e?�ZAY nX G d.d-� d-e>�ZAG d/d0� d0e<�ZBd1d2� ZCdS )4� )�ABC�abstractmethodN)�datetime� timedelta�timezone)� defaultdict)� getproxies)� EndpointType)�Dsn�logger�capture_internal_exceptions)�BackgroundWorker)�Envelope�Item� PayloadRef)� TYPE_CHECKING)�Any)�Callable)�Dict)� DefaultDict)�Iterable)�List)�Mapping)�Optional)�Self)�Tuple)�Type)�Union)� PoolManager)� ProxyManager)�Event�EventDataCategoryc C s t td�S )N� SO_KEEPALIVE��getattr�socket� r& r& �B/opt/hc_python/lib/python3.8/site-packages/sentry_sdk/transport.py�<lambda>2 � r( � c C s t td�S )N� TCP_KEEPIDLEr# r& r& r& r' r( 3 r) �- c C s t td�S )N� TCP_KEEPINTVLr# r& r& r& r' r( 4 r) � c C s t td�S )N� TCP_KEEPCNTr# r& r& r& r' r( 5 r) � � c @ sd e Zd ZdZdZddd�Zdd� Zedd� �Zdd d �Z d d � Z dd d�dd�Z dd� Z dd� Z dS )� TransportzWBaseclass for all transports. A transport is used to send an event to sentry. Nc C s8 || _ |r.|d d k r.|d r.t|d �| _nd | _d S )N�dsn)�optionsr � parsed_dsn��selfr4 r& r& r' �__init__G s zTransport.__init__c C s. t jdtdd� t� }|�|� | �|� dS )z� DEPRECATED: Please use capture_envelope instead. This gets invoked with the event dictionary when an event should be sent to sentry. zAcapture_event is deprecated, please use capture_envelope instead!r1 �� stacklevelN)�warnings�warn�DeprecationWarningr Z add_event�capture_envelope)r7 �event�enveloper& r& r' � capture_eventO s � zTransport.capture_eventc C s dS )a Send an envelope to Sentry. Envelopes are a data container format that can hold any type of data submitted to Sentry. We use it to send all event data (including errors, transactions, crons check-ins, etc.) to Sentry. Nr& )r7 r@ r& r& r' r> b s zTransport.capture_envelopec C s dS )z� Wait `timeout` seconds for the current events to be sent out. The default implementation is a no-op, since this method may only be relevant to some transports. Subclasses should override this method if necessary. Nr& �r7 �timeout�callbackr& r& r' �flushn s zTransport.flushc C s dS )z� Forcefully kills the transport. The default implementation is a no-op, since this method may only be relevant to some transports. Subclasses should override this method if necessary. Nr& �r7 r& r& r' �kill| s zTransport.killr* ��quantityc C s dS )a� This increments a counter for event loss by reason and data category by the given positive-int quantity (default 1). If an item is provided, the data category and quantity are extracted from the item, and the values passed for data_category and quantity are ignored. When recording a lost transaction via data_category="transaction", the calling code should also record the lost spans via this method. When recording lost spans, `quantity` should be set to the number of contained spans, plus one for the transaction itself. When passing an Item containing a transaction via the `item` parameter, this method automatically records the lost spans. Nr& )r7 �reason� data_category�itemrI r& r& r' �record_lost_event� s zTransport.record_lost_eventc C s dS )NTr& rF r& r& r' � is_healthy� s zTransport.is_healthyc C s& z | � � W n tk r Y nX d S �N)rG � ExceptionrF r& r& r' �__del__� s zTransport.__del__)N)N)NN)�__name__� __module__� __qualname__�__doc__r5 r8 rA r r> rE rG rM rN rQ r& r& r& r' r2 ? s � �� r2 c c s� |d krt �tj�}| �d�D ]�}z�|�� �d�}|d d� \}}|tt|�d� }|rb|�d�pddD ]Z}|dkr�z|d �d�}W n tk r� g }Y nX |r�d |kr�||fV qf||fV qfW q t t fk r� Y q Y q X q d S ) N�,�:r1 ��seconds�;rO � metric_bucket� Zcustom) r �nowr �utc�split�stripr �int� IndexError� LookupError� ValueError) �headerr] �limit� parametersZretry_after_val� categories� retry_after�category� namespacesr& r& r' �_parse_rate_limits� s$ rl c @ s� e Zd ZdZdd� Zd6dd�dd�Zd d � Zd d � Zej dfd d�Z dd� Z d7dd�Z d8dd�Z dd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd9d#d$�Zd%d&� Zd'd(� Zd)d*� Zd+d,� Zd:d-d.�Zd/d0� Zed1d2� �Z e d3d4� �Z e j d5d4� �Z dS );�BaseHttpTransportzThe base HTTP transport.c C sz ddl m} t�| |� | jd k s&t�|| _t|d d�| _| j� d| �| _ i | _ t j �� | _tt�| _t�� | _| j| j|d |d |d |d |d |d d �| _tj| _|�d i �}|�d|�d��}|�d|d k s�td kr�dnd�}|dk�rtd k�rt � d� d}d }|dk�r8t � d|� d| _ d | _ n|| _ |d k �rP|| _ n&| j dk�rdd| _ n| j dk�rvd| _ d S )Nr )�VERSIONZtransport_queue_size)Z queue_sizezsentry.python/%s� http_proxy� https_proxy�ca_certs� cert_file�key_file� proxy_headers)ro rp rq rr rs rt � _experimentsZtransport_compression_levelZ transport_zlib_compression_levelZtransport_compression_algo�gzip�brzSYou asked for brotli compression without the Brotli module, falling back to gzip -9)rw rv z2Unknown compression algo %s, disabling compression� r\ ) �sentry_sdk.constsrn r2 r8 r5 �AssertionErrorr4 r �_workerZto_auth�_auth�_disabled_until�urllib3�util�Retry�_retryr ra �_discarded_events�time�_last_client_report_sent� _make_pool�_pool� sentry_sdkZHub�_hub_cls�get�brotlir �warning�_compression_level�_compression_algo)r7 r4 rn Z experimentsZcompression_levelZcompression_algor& r& r' r8 � sn � � ���� � � zBaseHttpTransport.__init__Nr* rH c C s� | j d sd S |d k rv|j}d}|dkr\|�� p2i }t|�d�pBg �d }| j|d|d� q�|dkr�t|�� �prd}n|d kr�td��| j||f |7 < d S ) N�send_client_reportsr* � transaction�spans�spanrH � attachmentzdata category not provided) r4 rK Zget_transaction_event�lenr� rM � get_bytes� TypeErrorr� )r7 rJ rK rL rI r? Z span_countr& r& r' rM s z#BaseHttpTransport.record_lost_eventc C s |j �|�S rO )�headersr� �r7 �responsere r& r& r' �_get_header_value0 s z#BaseHttpTransport._get_header_valuec C s� | � |d�}|r,t�d� | j�t|�� nX|jdkr�t�d� | � |d�}|d k r`| j�|�nd pfd}t � t j �t |d� | jd < d S )Nzx-sentry-rate-limitsz%Rate-limited via x-sentry-rate-limits� zRate-limited via 429z Retry-After�< rX )r� r r� r} �updaterl �statusr� �parse_retry_afterr r] r r^ r )r7 r� re Zretry_after_valueri r& r& r' �_update_rate_limits4 s ��� �z%BaseHttpTransport._update_rate_limitsc s� � �fdd�}|� t�jj�t�j�� �d�� z��d|||�}W n( tk rl ��d� |d� � Y nX zr�� |� |j dkr���d� nN|j d ks�|j d k r�t � d |j t |d t |d d ��� ��d�|j �� |d� W 5 |�� X d S )Nc s6 � d kr�j | dd� n� jD ]}�j | |d� q d S )N�error)rK �rL )rM �items)rJ rL �r@ r7 r& r' � record_lossV s z4BaseHttpTransport._send_request.<locals>.record_loss)z User-Agentz X-Sentry-Auth�POST�networkZ network_errorr� Z status_429i, �� z%Unexpected status code: %s (body: %s)�data�contentz status_{})r� �strr| �clientZ to_header�_requestrP �on_dropped_event�closer� r� r r� r$ �format)r7 �bodyr� � endpoint_typer@ r� r� r& r� r' � _send_requestM s> ��� � z BaseHttpTransport._send_requestc C s d S rO r& )r7 Z_reasonr& r& r' r� � s z"BaseHttpTransport.on_dropped_eventFr� c C st | j d sd S |s(| jt�� | k s(d S | j}tt�| _t�� | _|sJd S ttt�� dd� |�� D �d�d�dd�S )Nr� c S s g | ]\\}}}|||d ��qS ))rJ rj rI r& )�.0rj rJ rI r& r& r' � <listcomp>� s �zBBaseHttpTransport._fetch_pending_client_report.<locals>.<listcomp>)� timestamp�discarded_events)�json� client_report)�type) r4 r� r� r� r ra r r r� )r7 �force�intervalr� r& r& r' � _fetch_pending_client_report� s& ��� �z.BaseHttpTransport._fetch_pending_client_reportc C s, | j |dd�}|d k r(| �t|gd�� d S )Nr� )r� r� )r� )r� r> r )r7 r� r� r& r& r' �_flush_client_reports� s z'BaseHttpTransport._flush_client_reportsc s � fdd�}||�p|d �S )Nc s0 | dkr d} � j �| �}|d k o.|t�tj�kS )N�statsdr[ )r} r� r r] r r^ )Zbucket�tsrF r& r' � _disabled� s z4BaseHttpTransport._check_disabled.<locals>._disabledr& )r7 rj r� r& rF r' �_check_disabled� s z!BaseHttpTransport._check_disabledc C s t dd� | j�� D ��S )Nc s s | ]}|t �tj�kV qd S rO )r r] r r^ )r� r� r& r& r' � <genexpr>� s z5BaseHttpTransport._is_rate_limited.<locals>.<genexpr>)�anyr} �valuesrF r& r& r' �_is_rate_limited� s �z"BaseHttpTransport._is_rate_limitedc C s | j �� S rO )r{ �fullrF r& r& r' �_is_worker_full� s z!BaseHttpTransport._is_worker_fullc C s | � � p| �� S rO )r� r� rF r& r& r' rN � s z BaseHttpTransport.is_healthyc C s� g }|j D ]>}| �|j�r>|jdkr.| �d� | jd|d� q |�|� q t|j|d�}|j sbd S | jdd�}|d k r�|j �|� | � |�\}}| j d k s�t �t � d|j| j j| j j� d d i}|r�||d < | j|�� |tj|d � d S ) N)r� r� �defaultr� Zself_rate_limitsZratelimit_backoffr� )r� r� � )r� z(Sending envelope [%s] project:%s host:%sz Content-Typez application/x-sentry-envelopezContent-Encoding)r� r� r@ )r� r� rK r� rM �appendr r� r� �_serialize_enveloper5 rz r �debug� descriptionZ project_id�hostr� �getvaluer �ENVELOPE)r7 r@ Z new_itemsrL Zclient_report_item�content_encodingr� r� r& r& r' �_send_envelope� sB � ��z BaseHttpTransport._send_envelopec C s� d }t �� }| jdks | jd kr,|�|� n\| j}| jdkr`td k r`|�tj|�� | jd�� n(t j |d| jd��}|�|� W 5 Q R X ||fS )Nr rw )Zquality�w)�fileobj�mode� compresslevel) �io�BytesIOr� r� Zserialize_intor� �write�compress� serializerv �GzipFile)r7 r@ r� r� �fr& r& r' r� s( �� �z%BaseHttpTransport._serialize_envelopec C s t � �d S rO ��NotImplementedError)r7 rq rr rs r& r& r' �_get_pool_options s z#BaseHttpTransport._get_pool_optionsc C sN t � �d�}|sdS |�d�D ]*}|�� }|j�|�sB|j�|�r dS q dS )N�noFrV T)r r� r_ r` r� �endswith�netloc)r7 r5 �no_proxyr� r& r& r' � _in_no_proxy s z BaseHttpTransport._in_no_proxyc C s t � �d S rO r� )r7 r5 ro rp rq rr rs rt r& r& r' r� & s z BaseHttpTransport._make_poolc C s t � �d S rO r� �r7 �methodr� r� r� r& r& r' r� 3 s zBaseHttpTransport._requestc sB � �fdd�}�j �|�s>��d� � jD ]}�jd|d� q*d S )Nc s( t � � ��� � ��� W 5 Q R X d S rO )r r� r� r& r� r& r' �send_envelope_wrapperA s zABaseHttpTransport.capture_envelope.<locals>.send_envelope_wrapperZ full_queueZqueue_overflowr� )r{ �submitr� r� rM )r7 r@ r� rL r& r� r' r> = s z"BaseHttpTransport.capture_envelopec s8 t �d� |dkr4� j�� fdd�� � j�||� d S )NzFlushing HTTP transportr c s � j dd�S )NT)r� )r� r& rF r& r' r( U r) z)BaseHttpTransport.flush.<locals>.<lambda>)r r� r{ r� rE rB r& rF r' rE L s zBaseHttpTransport.flushc C s t �d� | j�� d S )NzKilling HTTP transport)r r� r{ rG rF r& r& r' rG X s zBaseHttpTransport.killc C s t jdtdd� dS )zRConvenience method to warn users about the deprecation of the `hub_cls` attribute.zNThe `hub_cls` attribute is deprecated and will be removed in a future release.� r9 N)r; r< r= r& r& r& r' � _warn_hub_cls] s �z BaseHttpTransport._warn_hub_clsc C s t �� | jS )�QDEPRECATED: This attribute is deprecated and will be removed in a future release.�� HttpTransportr� r� rF r& r& r' �hub_clsg s zBaseHttpTransport.hub_clsc C s t �� || _dS )r� Nr� )r7 �valuer& r& r' r� n s )NN)Fr� )F)NN)N) rR rS rT rU r8 rM r� r� r r� r� r� r� r� r� r� r� rN r� r� r� r� r� r� r> rE rG � staticmethodr� �propertyr� �setterr& r& r& r' rm � sF H �� " � 9 5 � rm c @ s< e Zd ZU er eeef ed< d dd�Zdd� Z dd� Z dS ) r� r� Nc C s� | j �di ��d�}|d kr dnt|�dd�}d }| j d d k rJ| j d }| j d r�|d kr`g }dd � |D �}tD ]"}|d |d f|krr|�|� qr|d k r�||d< |p�tj�d �p�tj�d �p�t�� |d< |p�tj�d�|d< |p�tj�d�|d< |S )Nru Ztransport_num_poolsr1 � CERT_REQUIRED)� num_pools� cert_reqs�socket_options� keep_alivec S s h | ]}|d |d f�qS �r r* r& �r� �or& r& r' � <setcomp>� s z2HttpTransport._get_pool_options.<locals>.<setcomp>r r* � SSL_CERT_FILE�REQUESTS_CA_BUNDLErq �CLIENT_CERT_FILErr �CLIENT_KEY_FILErs ) r4 r� ra �KEEP_ALIVE_SOCKET_OPTIONSr� �os�environ�certifi�where) r7 rq rr rs r� r4 r� � used_options�default_optionr& r& r' r� z s4 � � ��z HttpTransport._get_pool_optionsc C s� d }| � |�} |jdkr6|dkr6|p4| o4t� �d�}|sX|dkrX|pV| oVt� �d�}| �|||�} |r�|rv|| d< |�d�r�d} zddlm} W n$ tk r� d } t � d |� Y nX | r�| |f| �S t j f | �S q�t j |f| �S n t j f | �S d S ) N�https� �httprt �socksTr )�SOCKSProxyManagerFz�You have configured a SOCKS proxy (%s) but support for SOCKS proxies is not installed. Disabling proxy support. Please add `PySocks` (or `urllib3` with the `[socks]` extra) to your dependencies.)r� �schemer r� r� � startswithZurllib3.contrib.socksr � ImportErrorr r� r~ r r ) r7 r5 ro rp rq rr rs rt �proxyr� �optsZuse_socks_proxyr r& r& r' r� � s2 � zHttpTransport._make_poolc C s | j j|| j�|�||d�S )N)r� r� �r� �requestr| Z get_api_urlr� r& r& r' r� � s �zHttpTransport._request)NN) rR rS rT r r r r �__annotations__r� r� r� r& r& r& r' r� v s &1r� c s e Zd Z� fdd�Z� ZS )�Http2Transportc s t � �|� t�d� d S )NzhYou tried to use HTTP2Transport but don't have httpcore[http2] installed. Falling back to HTTPTransport.)�superr8 r r� r6 �� __class__r& r' r8 � s �zHttp2Transport.__init__)rR rS rT r8 � __classcell__r& r& r r' r � s r c @ sP e Zd ZU dZer*eejejej f e d< dd� Z dd� Z d dd �Z d d � ZdS ) r z&The HTTP2 transport based on httpcore.r� c s t � fdd�|jD �d �S )Nc 3 s. | ]&\}}|� d ��� � kr|� d �V qdS )�asciiN)�decode�lower)r� �key�val�re r& r' r� � s �z3Http2Transport._get_header_value.<locals>.<genexpr>)�nextr� r� r& r r' r� � s ��z Http2Transport._get_header_valuec C s | j j|| j�|�||d�}|S )N)r� r� r )r7 r� r� r� r� r� r& r& r' r� s �zHttp2Transport._requestNc C s� ddd�}| j d d k r"| j d ng }dd� |D �}tD ]"}|d |d f|kr8|�|� q8||d< t�� }|�|p�tj�d �p�tj�d �p�t � � � |p�tj�d �}|p�tj�d �}|d k r�|� ||� ||d < |S )NTr� )Zhttp2�retriesr� c S s h | ]}|d |d f�qS r� r& r� r& r& r' r� s z3Http2Transport._get_pool_options.<locals>.<setcomp>r r* r� r� r� r � ssl_context) r4 r r� �ssl�create_default_context�load_verify_locationsr r r� r r �load_cert_chain) r7 rq rr rs r4 r� r r r" r& r& r' r� s6 � � � � �� z Http2Transport._get_pool_optionsc C s� d }| � |�} |jdkr6|dkr6|p4| o4t� �d�}|sX|dkrX|pV| oVt� �d�}| �|||�} |r�|rv|| d< |�d�r�z6d| kr�| �d�} | r�t�d� t j f d|i| ��W S t k r� t�d |� Y q�X nt j f d|i| ��S t j f | �S ) Nr r r rt r r� zqYou have defined socket_options but using a SOCKS proxy which doesn't support these. We'll ignore socket_options.� proxy_urlzoYou have configured a SOCKS proxy (%s) but support for SOCKS proxies is not installed. Disabling proxy support.)r� r r r� r� r �popr r� �httpcore� SOCKSProxy� RuntimeError� HTTPProxy�ConnectionPool) r7 r5 ro rp rq rr rs rt r r� r r� r& r& r' r� 6 s4 �� zHttp2Transport._make_pool)NN)rR rS rT rU r r r) r* r, r- r r� r� r� r� r& r& r& r' r � s � $c @ s0 e Zd ZdZdd� Zdd� Zedd�dd �ZdS ) �_FunctionTransportz� DEPRECATED: Users wishing to provide a custom transport should subclass the Transport class, rather than providing a function. c C s t �| � || _d S rO )r2 r8 �_func)r7 �funcr& r& r' r8 l s z_FunctionTransport.__init__c C s | � |� d S rO )r/ )r7 r? r& r& r' rA s s z _FunctionTransport.capture_eventN)r@ �returnc C s |� � }|d k r| �|� d S rO )Z get_eventrA )r7 r@ r? r& r& r' r> z s z#_FunctionTransport.capture_envelope)rR rS rT rU r8 rA r r> r& r& r& r' r. f s r. c C s� | d }| � di �� dd�}|r$tnt}t|t�r6|S t|t�rPt|t�rP|}n t|�rptj dt dd� t |�S | d r�|| �S d S ) N� transportru Ztransport_http2Fz�Function transports are deprecated and will be removed in a future release.Please provide a Transport instance or subclass, instead.r1 r9 r3 ) r� r r� � isinstancer2 r� � issubclass�callabler; r< r= r. )r4 Z ref_transportZuse_http2_transportZ transport_clsr& r& r' �make_transport� s$ � �r6 )N)D�abcr r r� r rv r% r# r� r; r r r � collectionsr �urllib.requestr r� r r~ r r� ry r Zsentry_sdk.utilsr r r Zsentry_sdk.workerr Zsentry_sdk.enveloper r r �typingr r r r r r r r r r r r r Zurllib3.poolmanagerr r Zsentry_sdk._typesr r! r � SOL_SOCKET�SOL_TCP�optionr� �AttributeErrorr2 rl rm r� r) r r. r6 r& r& r& r' �<module> s| �" l 3k w