o
    i                     @   s2  d dl Z d dlZd dlmZ d dlmZmZ d dlmZm	Z	 d dl
Z
d dlmZ d dlmZmZ d dlmZmZmZmZ d dlmZ errd d	lmZ d d
lmZ d dlmZ d dlmZ d dlmZ ededef dZededZG dd deZ								d#ddZ ddde!ddfddZ"d$d!d"Z#dS )%    Nwraps)Threadcurrent_thread)ThreadPoolExecutorFuture)Integrationuse_isolation_scope	use_scope)event_from_exceptioncapture_internal_exceptionsloggerreraise)TYPE_CHECKING)Any)TypeVar)Callable)Optional)ExcInfoF.)boundTc                   @   s8   e Zd ZdZ	ddddeddfdd	Zedd
dZdS )ThreadingIntegration	threadingNTpropagate_hubzOptional[bool]propagate_scopereturnc                 C   s.   |d ur	t d || _|d ur|| _d S d S )NzLDeprecated: propagate_hub is deprecated. This will be removed in the future.)r   warningr   )selfr   r    r    ^/home/ubuntu/veenaModal/venv/lib/python3.10/site-packages/sentry_sdk/integrations/threading.py__init__"   s   
zThreadingIntegration.__init__c               	      s   t jzddlm}  W n ty   d } Y nw z	dd l}|j}W n ttfy-   d }Y nw tj	dk oF|d uoF|dk oF| d uoF| dkoF| dk  t
d fdd}|t _ttj t_d S )Nr   )VERSION)   	   z4.0.0)r$   r   )   r   r   r   ar   kwr   c                    s   t  t}|d u r| g|R i |S |jr8 r+tjddd t  }t  }nt  	 }t  	 }nd }d }t
  t||t| jd| j}|| _W d    n1 sYw   Y  | g|R i |S )NaV  There is a known issue with Django channels 2.x and 3.x when using Python 3.8 or older. (Async support is emulated using threads and some Sentry data may be leaked between those threads.) Please either upgrade to Django channels 4.0+, use Django's async features available in Django 3.1+ instead of Django channels, or upgrade to Python 3.9+.   )
stacklevel__func__)
sentry_sdk
get_clientget_integrationr   r   warningswarnget_isolation_scopeget_current_scopeforkr   	_wrap_rungetattrrun)r   r'   r(   integrationisolation_scopecurrent_scopenew_runis_async_emulated_with_threads	old_startr    r!   sentry_startL   s0   
z5ThreadingIntegration.setup_once.<locals>.sentry_start)r   r   r'   r   r(   r   r   r   )r   startdjangor#   ImportErrorchannels__version__AttributeErrorsysversion_infor    _wrap_threadpool_executor_submitr   submit)django_versionrB   channels_versionr>   r    r;   r!   
setup_once3   s:   

	(
zThreadingIntegration.setup_once)NT)r   N)__name__
__module____qualname__
identifierboolr"   staticmethodrK   r    r    r    r!   r      s    
r   isolation_scope_to_useOptional[sentry_sdk.Scope]current_scope_to_useold_run_funcr   c                    s   t d fdd}|S )Nr'   r   r(   r   c               	      s   d fdd}d urFd urFt ' t | W  d    W  d    S 1 s/w   Y  W d    d S 1 s?w   Y  d S | S )Nr   r   c                     sF   zt  } | g dd  R i W S  ty"   tt   Y d S w )N   )r   	Exceptionr   _capture_exception)r   )r'   r(   rU   r    r!   _run_old_run_func   s    z1_wrap_run.<locals>.run.<locals>._run_old_run_func)r   r   r	   )r'   r(   rY   rT   rR   rU   )r'   r(   r!   r6      s   

"z_wrap_run.<locals>.run)r'   r   r(   r   r   r   r   )rR   rT   rU   r6   r    rZ   r!   r4   {   s   r4   funczCallable[..., Future[T]]r<   c                    s0   t  											d fd
d}|S )zB
    Wrap submit call to propagate scopes on task submission.
    r   r   fnCallable[..., T]argsr   kwargsr   	Future[T]c                    s   t  t}|d u r| g|R i |S |jr%r%t  t   n|jr5t   t    nd d  d fdd}| |g|R i |S )Nr^   r   r_   r   c               	      s   d ur> d ur>t * t  | i |W  d    W  d    S 1 s*w   Y  W d    n1 s9w   Y  | i |S )Nr	   )r^   r_   r9   r\   r8   r    r!   
wrapped_fn   s   

zK_wrap_threadpool_executor_submit.<locals>.sentry_submit.<locals>.wrapped_fn)r^   r   r_   r   r   r   )r,   r-   r.   r   r   r1   r2   r3   )r   r\   r^   r_   r7   rb   r[   r<   ra   r!   sentry_submit   s   

z7_wrap_threadpool_executor_submit.<locals>.sentry_submitN)
r   r   r\   r]   r^   r   r_   r   r   r`   r   )r[   r<   rd   r    rc   r!   rG      s   rG   r   c                  C   sJ   t  } t }|td ur#t| |jdddd\}}tj||d | S )Nr   F)typehandled)client_options	mechanism)hint)	rE   exc_infor,   r-   r.   r   r   optionscapture_event)rj   clienteventri   r    r    r!   rX      s   
rX   )rR   rS   rT   rS   rU   r   r   r   )r   r   )$rE   r/   	functoolsr   r   r   r   concurrent.futuresr   r   r,   sentry_sdk.integrationsr   sentry_sdk.scoper
   r   sentry_sdk.utilsr   r   r   r   typingr   r   r   r   r   sentry_sdk._typesr   r   r   r   r4   rP   rG   rX   r    r    r    r!   <module>   sH    \

)