Edit File: stdlib.cpython-37.opt-1.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�versionZbuildc @ s e Zd ZdZedd� �ZdS )�StdlibIntegrationZstdlibc 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__Z 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� ZhttpsZhttpz:%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* r+ �args�kwargs�hubr, r- r. 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 )Nr3 Z status_code�reason)�getattrr1 �statusZset_http_status�intr> Zfinish)r4 r5 r6 r8 r9 )�real_getresponser r � getresponse_ s z%_install_httplib.<locals>.getresponse)r r= rC )r= rC r )rB r< 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)r5 r6 r ZpositionZsetdefault_callbackr9 r r r �_init_arguments s rE 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 )Nr5 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 �listrE r �tuplerD r �join�map�strr r2 �upper�replacer0 r1 �set_tag�pid) r4 �a�kwr7 r5 rF r) rJ �k�vr8 r9 )�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 r0 r\ r] )r4 r^ r_ r7 r8 )�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 r0 r\ r] )r4 r^ r_ r7 r8 )�old_popen_communicater r � sentry_patched_popen_communicate� s z=_install_subprocess.<locals>.sentry_patched_popen_communicate)rT �Popen�__init__�wait� communicate)rc re rg r )rf rb rd r r! � s - r! c C s t tjdd�S )NrQ )�prefix)r rL rM r r r r �"get_subprocess_traceparent_headers� s rm )N)&rL rT �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.clientZpython_implementation� version_infor r r r rE r! rm r r r r �<module> s8 : R