관리-도구
편집 파일: stdlib.cpython-38.pyc
U �MgQ" � @ sB d dl Z d dlZd dlZd dlZd dlmZ d dlZd dlmZm Z d dl m Z d dl m Z d dlmZmZ d dlmZmZmZmZmZmZmZ d dlmZ er�d d lmZ d d lm Z d d lm Z d d lm Z d d lm Z d dl m!Z!m"Z" e�#� dej$dd� ej%d�Z&G dd� de �Z'dd� Z(d dd�Z)dd� Z*dd� Z+dS ) � N)�HTTPConnection)�OP�SPANDATA)� Integration)�add_global_event_processor)�EnvironHeaders�should_propagate_trace)�SENSITIVE_DATA_SUBSTITUTE�capture_internal_exceptions�ensure_integration_enabled� is_sentry_url�logger� safe_repr� parse_url)� TYPE_CHECKING)�Any)�Callable)�Dict)�Optional)�List)�Event�Hintz%s.%s.%s� )�name�version�buildc @ s e Zd ZdZedd� �ZdS )�StdlibIntegration�stdlibc C s t � t� tdd� �} d S )Nc S s< t �� �t�d k r8| �di �}t|t�r8d|kr8t|d< | S )N�contexts�runtime)� sentry_sdk� get_client�get_integrationr � setdefault� isinstance�dict�_RUNTIME_CONTEXT)�event�hintr � r) �L/opt/hc_python/lib/python3.8/site-packages/sentry_sdk/integrations/stdlib.py�add_python_runtime_context2 s z@StdlibIntegration.setup_once.<locals>.add_python_runtime_context)�_install_httplib�_install_subprocessr )r+ r) r) r* � setup_once, s z StdlibIntegration.setup_onceN)�__name__� __module__� __qualname__� identifier� staticmethodr. r) r) r) r* r ) s r c s4 t j�t j� �fdd�} � fdd�}| t _|t _d S )Nc s| | j }| j}| j}t�� }|�t�d ks2t||�rF� | ||f|�|�S |} | d ks\| �d�s�d|dkrjdpld|||kr~d| p�d|f } d } t � � t | dd �} W 5 Q R X tj t j d || r�| jntf d d �} | �tj|� | d k �r| �d | j� | �tj| j� | �tj| j� � | ||f|�|�} t|| ��rrt�� j| d�D ],\} }t�dj | || d�� | � | |� �qD| | _ | S )N)zhttp://zhttps://z %s://%s%s%si� �https�httpz:%s� F)�sanitizez%s %szauto.http.stdlib.httplib��opr �origin�url��spanzJ[Tracing] Adding `{key}` header {value} to outgoing request to {real_url}.)�key�value�real_url) �host�port� default_portr r! r" r r � startswithr r � start_spanr Z HTTP_CLIENTr; r �set_datar Z HTTP_METHODZ HTTP_QUERY�queryZ HTTP_FRAGMENT�fragmentr �get_current_scope� iter_trace_propagation_headersr �debug�format� putheader�_sentrysdk_span)�self�methodr; �args�kwargsrA rB rC �clientr@ � parsed_urlr= �rvr> r? )�real_putrequestr) r* � putrequestB sd ���� �� ��z$_install_httplib.<locals>.putrequestc s^ t | dd �}|d kr$� | f|�|�S � | f|�|�}|�t|j�� |�d|j� |�� |S )NrN �reason)�getattrZset_http_status�int�statusrF rX �finish)rO rQ rR r= rU )�real_getresponser) r* � getresponse{ s z%_install_httplib.<locals>.getresponse)r rW r^ )rW r^ r) )r] rV r* r, = s 9 r, c C s� ||kr2|| }|dk r ||�}|dk r�|||< nR|t | �k rh| | }|dk rV||�}|dk r�|| |< n |or|d�}|dk r�|||< |S )a? given (*args, **kwargs) of a function call, retrieve (and optionally set a default for) an argument by either name or position. This is useful for wrapping functions with complex type signatures and extracting a few arguments without needing to redefine that function's entire type signature. N)�len)rQ rR r �positionZsetdefault_callbackrU r) r) r* �_init_argument� s ra c sv t jj�tt���fdd��} | t j_t jj�tt���fdd��}|t j_t jj� tt� �� fdd��}|t j_d S )Nc s& t |�}t||dd�pg }t||dd�}d }t|t tf�rht|�dk rht� � d�tt|��}W 5 Q R X |d krxt |�}d }t j t j |dd���}t �� j|d �D ]<\}} |d kr�t||d d d d � �}| |d|�� �dd� < q�|r�|�d|� � | f|�|�} |�d| j� | W 5 Q R � S Q R X d S )NrQ r �cwd� �d � �!auto.subprocess.stdlib.subprocessr8 r< �env� c S s t | d k r| ntj�S )N)r% �os�environ)�xr) r) r* �<lambda>� � zH_install_subprocess.<locals>.sentry_patched_popen_init.<locals>.<lambda>� SUBPROCESS_�-�_zsubprocess.cwd�subprocess.pid)�listra r$ �tupler_ r �join�map�strr r rE r Z SUBPROCESSrI rJ �upper�replacerF �set_tag�pid) rO �a�kwrQ rb � descriptionrg r= �k�vrU )�old_popen_initr) r* �sentry_patched_popen_init� sB ��� z6_install_subprocess.<locals>.sentry_patched_popen_initc sF t jtjdd��,}|�d| j� � | f|�|�W 5 Q R � S Q R X d S �Nrf )r9 r: rq )r rE r ZSUBPROCESS_WAITry rz �rO r{ r| r= )�old_popen_waitr) r* �sentry_patched_popen_wait� s �z6_install_subprocess.<locals>.sentry_patched_popen_waitc sF t jtjdd��,}|�d| j� � | f|�|�W 5 Q R � S Q R X d S r� )r rE r ZSUBPROCESS_COMMUNICATEry rz r� )�old_popen_communicater) r* � sentry_patched_popen_communicate� s �z=_install_subprocess.<locals>.sentry_patched_popen_communicate)� subprocess�Popen�__init__r r �wait� communicate)r� r� r� r) )r� r� r� r* r- � s 4 r- c C s t tjdd�S )Nrn )�prefix)r ri rj r) r) r) r* �"get_subprocess_traceparent_headers s r� )N),ri r� �sys�platform� http.clientr r Zsentry_sdk.constsr r Zsentry_sdk.integrationsr Zsentry_sdk.scoper Zsentry_sdk.tracing_utilsr r Zsentry_sdk.utilsr r r r r r r �typingr r r r r r Zsentry_sdk._typesr r �python_implementation� version_infor r& r r, ra r- r� r) r) r) r* �<module> s6 $ �Q X