3
Z_ð]K  ã               @   sL   d dl mZ d dlmZ ejjZejjZdd„ Z	dd„ Z
dd„ Zd	d
„ ZdS )é    )Úgreenthread)Úapic             C   s    t jƒ rt jƒ | _t| |ƒ d S )N)r   Z
is_tracingZget_trace_dataÚ
trace_dataÚ__original_init__)ÚselfÚparent© r   ú?/tmp/pip-build-yz_zf6az/eventlet/eventlet/zipkin/greenthread.pyÚ_patched__init
   s    
r
   c             C   s(   t | dƒrtj| jƒ t| |||ƒ d S )Nr   )Úhasattrr   Zset_trace_datar   Ú__original_main__)r   ÚfunctionÚargsÚkwargsr   r   r	   Ú_patched_main   s    
r   c               C   s   t tj_ttj_d S )N)r
   r   ÚGreenThreadÚ__init__r   Úmainr   r   r   r	   Úpatch   s    r   c               C   s   t tj_ttj_d S )N)r   r   r   r   r   r   r   r   r   r	   Úunpatch   s    r   N)Zeventletr   Zeventlet.zipkinr   r   r   r   r   r   r
   r   r   r   r   r   r   r	   Ú<module>   s   