Edit File: stdlib.cpython-37.pyc
B Z~�_� � @ sD d dl Z d dlZd dlZd dlZd dlmZ d dlmZ d dlm Z d dl m Z d dl m Z mZ d dlmZ er�d dlmZ d d lmZ d d lmZ d d lmZ d d lmZ d d lmZmZ yd dlmZ W n ek r� d dl mZ Y nX e� � dej dd� ej d�Z G dd� de�Z!dd� Z"d dd�Z#dd� Z$dd� Z%dS ) � N)�Hub)� Integration)�add_global_event_processor)�EnvironHeaders)�capture_internal_exceptions� safe_repr)�MYPY)�Any)�Callable)�Dict)�Optional)�List)�Event�Hint)�HTTPConnectionz%s.%s.%s� )�name�version�buildc @ s e Zd ZdZedd� �ZdS )�StdlibIntegration�stdlibc C s t � t� tdd� �} d S )Nc S s: t j�t�d k r6| �di �}t|t�r6d|kr6t|d< | S )N�contextsZruntime)r �current�get_integrationr � setdefault� isinstance�dict�_RUNTIME_CONTEXT)�eventZhintr � r �O/opt/alt/python37/lib/python3.7/site-packages/sentry_sdk/integrations/stdlib.py�add_python_runtime_context. 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� t j}|�t�d kr(� | ||f|�|�S | j}| j}| j}|} | �d�std|dkrVdpXd|||krjd| pld|f } |jdd|| f d �} | � d |� | � d | � � | ||f|�|�} x |� � D ]\} } | � | | � q�W | | _ | S ) N)zhttp://zhttps://z %s://%s%s%si� �https�httpz:%s� z%s %s)�op� description�method�url) r r r r �host�port� default_port� startswith� start_span�set_data� iter_trace_propagation_headersZ putheader�_sentrysdk_span)�selfr/ r0 �args�kwargs�hubr1 r2 r3 Zreal_url�span�rv�key�value)�real_putrequestr r � putrequest> s* z$_install_httplib.<locals>.putrequestc sl t | dd �}|d kr$� | f|�|�S � | f|�|�}|�d|j� |�t|j�� |�d|j� |�� |S )Nr8 Z status_code�reason)�getattrr6 �statusZset_http_status�intrC Zfinish)r9 r: r; r= r> )�real_getresponser r � getresponse_ s z%_install_httplib.<locals>.getresponse)r rB rH )rB rH r )rG rA r r" 9 s ! 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)r: r; r �positionZsetdefault_callbackr> r r r �_init_arguments s rK c sX t jj��fdd�} | t j_t jj��fdd�}|t j_t jj� � fdd�}|t j_d S )Nc s8 t j}|�t�d kr$� | f|�|�S t|�}t||dd�p<g }t||dd�}d }t|ttf�r�t|�dk r�t � � d� t t |��}W d Q R X |d kr�t |�}d }xH|�� D ]<\}} |d kr�t||ddd d � �}| |d |�� �d d � < q�W |jd|d��6} |�r | �d|� � | f|�|�} | �d| j� | S Q R X d S )Nr: r �cwd� �d � �env� c S s t | p tj�S )N)r �os�environ)�xr r r �<lambda>� � zH_install_subprocess.<locals>.sentry_patched_popen_init.<locals>.<lambda>� SUBPROCESS_�-�_� subprocess)r- r. zsubprocess.cwdzsubprocess.pid)r r r r �listrK r �tuplerI r �join�map�strr r7 �upper�replacer5 r6 �set_tag�pid) r9 �a�kwr< r: rL r. rP �k�vr= r> )�old_popen_initr r �sentry_patched_popen_init� s. z6_install_subprocess.<locals>.sentry_patched_popen_initc sZ t j}|�t�d kr$� | f|�|�S |jdd�� }|�d| j� � | f|�|�S Q R X d S )Nzsubprocess.wait)r- zsubprocess.pid)r r r r r5 rb rc )r9 rd re r< r= )�old_popen_waitr r �sentry_patched_popen_wait� s z6_install_subprocess.<locals>.sentry_patched_popen_waitc sZ t j}|�t�d kr$� | f|�|�S |jdd�� }|�d| j� � | f|�|�S Q R X d S )Nzsubprocess.communicate)r- zsubprocess.pid)r r r r r5 rb rc )r9 rd re r< r= )�old_popen_communicater r � sentry_patched_popen_communicate� s z=_install_subprocess.<locals>.sentry_patched_popen_communicate)rZ �Popen�__init__�wait� communicate)ri rk rm r )rl rh rj r r# � s - r# c C s t tjdd�S )NrW )�prefix)r rR rS r r r r �"get_subprocess_traceparent_headers� s rs )N)&rR rZ �sys�platformZsentry_sdk.hubr Zsentry_sdk.integrationsr Zsentry_sdk.scoper Zsentry_sdk.tracingr Zsentry_sdk.utilsr r Zsentry_sdk._typesr �typingr r r r r r r Zhttplibr � ImportErrorZ http.client�python_implementation� version_infor r r r" rK r# rs r r r r �<module> s8 : R