a
   ^a\2                   @   s   d Z dZddlmZ ddlZddlZddlZddlZddlZddl	Z	e
 Zdae Zdd Zee ee	dre	jejejejd	 G d
d deZdd ZG dd dejZG dd dejZdS )zImplements ThreadPoolExecutor.z"Brian Quinlan (brian@sweetapp.com)    )_baseNFc                  C   sf   t  daW d    n1 s0    Y  tt } | D ]\}}|d  q8| D ]\}}|  qPd S NT)_global_shutdown_lock	_shutdownlist_threads_queuesitemsputjoin)r   tq r   //usr/lib/python3.9/concurrent/futures/thread.py_python_exit   s    "r   register_at_fork)beforeafter_in_childafter_in_parentc                   @   s&   e Zd Zdd Zdd ZeejZdS )	_WorkItemc                 C   s   || _ || _|| _|| _d S N)futurefnargskwargs)selfr   r   r   r   r   r   r   __init__/   s    z_WorkItem.__init__c              
   C   sl   | j  sd S z| j| ji | j}W n4 tyZ } z| j | d } W Y d }~nd }~0 0 | j | d S r   )r   Zset_running_or_notify_cancelr   r   r   BaseExceptionset_exceptionZ
set_result)r   resultexcr   r   r   run5   s    
z_WorkItem.runN)	__name__
__module____qualname__r   r    classmethodtypesGenericAlias__class_getitem__r   r   r   r   r   .   s   r   c                 C   s   |d urPz||  W n: t yN   tjjddd |  }|d urH|  Y d S 0 zx|jdd}|d ur|  ~|  }|d ur|j  ~qR|  }t	s|d u s|j	r|d urd|_	|
d  W d S ~qRW n" t y   tjjddd Y n0 d S )NzException in initializer:T)exc_info)blockzException in worker)r   r   ZLOGGERZcritical_initializer_failedgetr    _idle_semaphorereleaser   r	   )Zexecutor_referenceZ
work_queueinitializerinitargsexecutor	work_itemr   r   r   _workerE   s8    

r2   c                   @   s   e Zd ZdZdS )BrokenThreadPoolzR
    Raised when a worker thread in a ThreadPoolExecutor failed initializing.
    N)r!   r"   r#   __doc__r   r   r   r   r3   p   s   r3   c                   @   s`   e Zd Ze jZdddZdd Ze	j
jje_dd	 Zd
d ZdddddZe	j
jje_dS )ThreadPoolExecutorN r   c                 C   s   |du rt dt pdd }|dkr.td|durFt|sFtd|| _t | _	t
d| _t | _d| _d| _t
 | _|pd	|   | _|| _|| _dS )
a  Initializes a new ThreadPoolExecutor instance.

        Args:
            max_workers: The maximum number of threads that can be used to
                execute the given calls.
            thread_name_prefix: An optional name prefix to give our threads.
            initializer: A callable used to initialize worker threads.
            initargs: A tuple of arguments to pass to the initializer.
        N          r   z"max_workers must be greater than 0zinitializer must be a callableFzThreadPoolExecutor-%d)minos	cpu_count
ValueErrorcallable	TypeError_max_workersqueueZSimpleQueue_work_queue	threading	Semaphorer,   set_threads_brokenr   Lock_shutdown_lock_counter_thread_name_prefix_initializer	_initargs)r   max_workersZthread_name_prefixr.   r/   r   r   r   r   {   s$    


zThreadPoolExecutor.__init__c             	   O   s   | j  tv | jrt| j| jr,tdtr8tdt }t||||}| j	
| |   |W  d    W  d    S 1 s0    Y  W d    n1 s0    Y  d S )Nz*cannot schedule new futures after shutdownz6cannot schedule new futures after interpreter shutdown)rI   r   rG   r3   r   RuntimeErrorr   ZFuturer   rB   r	   _adjust_thread_count)r   r   r   r   fwr   r   r   submit   s    
zThreadPoolExecutor.submitc                 C   s   | j jddrd S | jfdd}t| j}|| jk rd| jp>| |f }tj|t	t
| || j| j| jfd}|  | j| | jt|< d S )Nr   )timeoutc                 S   s   | d  d S r   )r	   )_r   r   r   r   
weakref_cb   s    z;ThreadPoolExecutor._adjust_thread_count.<locals>.weakref_cbz%s_%d)nametargetr   )r,   acquirerB   lenrF   r@   rK   rC   Threadr2   weakrefrefrL   rM   startaddr   )r   rV   Znum_threadsZthread_namer   r   r   r   rP      s$    



z'ThreadPoolExecutor._adjust_thread_countc              	   C   st   | j Z d| _z| j }W n tjy4   Y qRY n0 |d ur|jt| j qW d    n1 sf0    Y  d S )NzBA thread initializer failed, the thread pool is not usable anymore)	rI   rG   rB   
get_nowaitrA   Emptyr   r   r3   )r   r1   r   r   r   r*      s    
z&ThreadPoolExecutor._initializer_failedTF)cancel_futuresc             	   C   s   | j b d| _|rNz| j }W n tjy8   Y qNY n0 |d ur|j  q| jd  W d    n1 sn0    Y  |r| j	D ]}|
  qd S r   )rI   r   rB   r`   rA   ra   r   cancelr	   rF   r
   )r   waitrb   r1   r   r   r   r   shutdown   s    
*
zThreadPoolExecutor.shutdown)Nr6   Nr   )T)r!   r"   r#   	itertoolscount__next__rJ   r   rS   r   Executorr4   rP   r*   re   r   r   r   r   r5   v   s   
  
&r5   )r4   
__author__concurrent.futuresr   rf   rA   rC   r%   r\   r;   WeakKeyDictionaryr   r   rH   r   r   _register_atexithasattrr   rY   _at_fork_reinitr-   objectr   r2   ZBrokenExecutorr3   ri   r5   r   r   r   r   <module>   s,   

+