관리-도구
편집 파일: httpx.cpython-38.pyc
U �Mg� � @ s� d dl Z d dlmZmZ d dlmZmZ d dlmZ d dl m Z d dl m Z m Z mZmZmZ d dlmZ erxd dlmZ z d d lmZmZmZmZ W n ek r� ed ��Y nX d gZG d d � d e�Zd d� Z dd� Z dS )� N)�OP�SPANDATA)� Integration� DidNotEnable)�BAGGAGE_HEADER_NAME)�should_propagate_trace)�SENSITIVE_DATA_SUBSTITUTE�capture_internal_exceptions�ensure_integration_enabled�logger� parse_url)� TYPE_CHECKING)�Any)� AsyncClient�Client�Request�Responsezhttpx is not installed�HttpxIntegrationc @ s& e Zd ZdZde� �Zedd� �ZdS )r �httpxz auto.http.c C s t � t� dS )z� httpx has its own transport layer and can be customized when needed, so patch Client.send and AsyncClient.send to support both synchronous and async interfaces. N)�_install_httpx_client�_install_httpx_async_client� r r �K/opt/hc_python/lib/python3.8/site-packages/sentry_sdk/integrations/httpx.py� setup_once s zHttpxIntegration.setup_onceN)�__name__� __module__� __qualname__� identifier�origin� staticmethodr r r r r r s c s& t j� tt� �� fdd��} | t _d S )Nc sZ d }t � � tt|j�dd�}W 5 Q R X tjtjd|j|r@|jnt f t j d�� ��}|� t j|j� |d k r�|� d|j� |� t j|j� |� t j|j� tt�� t|j���rt�� �� D ]Z\}}t�dj|||jd�� |tk�r |j� t��r |j| d| 7 < q�||j|< q�� | |f|�}|� |j � |� d |j � |W 5 Q R � S Q R X d S � NF)�sanitizez%s %s)�op�namer �urlzE[Tracing] Adding `{key}` header {value} to outgoing request to {url}.)�key�valuer$ �,�reason) r r �strr$ � sentry_sdk� start_spanr � HTTP_CLIENT�methodr r r �set_datar � HTTP_METHOD� HTTP_QUERY�query� HTTP_FRAGMENT�fragmentr � get_client�get_current_scope� iter_trace_propagation_headersr �debug�formatr �headers�get�set_http_status� status_code� reason_phrase��self�request�kwargs� parsed_url�spanr% r& �rv�Z real_sendr r �send/ sN ��� � ��� z#_install_httpx_client.<locals>.send)r rF r r �rF r rE r r + s .r c s t j� � fdd�} | t _d S )Nc � s� t �� �t�d kr&� | |f|�I d H S d }t� � tt|j�dd�}W 5 Q R X t jt j d|j |rf|jnt f tj d���}|�tj|j � |d k r�|�d|j� |�tj|j� |�tj|j� tt �� t|j���r>t �� �� D ]Z\}}t�dj|||jd�� |tk�r2|j � t��r2|j | d| 7 < q�||j |< q� | |f|�I d H }|� |j � |�d |j � |W 5 Q R � S Q R X d S r )!r* r4 Zget_integrationr r r r) r$ r+ r r, r- r r r. r r/ r0 r1 r2 r3 r r5 r6 r r7 r8 r r9 r: r; r< r= r> rE r r rF e sR ��� � ��� z)_install_httpx_async_client.<locals>.send)r rF rG r rE r r a s 1r ) r* Zsentry_sdk.constsr r Zsentry_sdk.integrationsr r Zsentry_sdk.tracingr Zsentry_sdk.tracing_utilsr Zsentry_sdk.utilsr r r r r �typingr r r r r r r � ImportError�__all__r r r r r r r �<module> s 6