B
    \                 @   s   d Z dZddl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	e
 Zdadd Zee 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)    N)_baseFc              C   sJ   da tt } x| D ]\}}|d  qW x| D ]\}}|  q2W d S )NT)	_shutdownlist_threads_queuesitemsputjoin)r   tq r   *lib/python3.7/concurrent/futures/thread.py_python_exit!   s    r   c               @   s   e Zd Zdd Zdd 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   sf   | j  sd S y| j| j| j}W n2 tk
rT } z| j | d } W d d }~X Y nX | j | d S )N)r   Zset_running_or_notify_cancelr   r   r   BaseExceptionset_exceptionZ
set_result)r   resultexcr   r   r   run4   s    
z_WorkItem.runN)__name__
__module____qualname__r   r   r   r   r   r   r   -   s   r   c             C   s   |d k	rPy||  W n: t k
rN   tjjddd |  }|d k	rJ|  d S X y`xZ|jdd}|d k	rt|  ~qT|  }ts|d ks|jr|d k	rd|_|d  d S ~qTW W n$ t k
r   tjjddd Y nX d S )NzException in initializer:T)exc_info)blockzException in worker)	r   r   ZLOGGERZcritical_initializer_failedgetr   r   r   )Zexecutor_referenceZ
work_queueinitializerinitargsZexecutor	work_itemr   r   r   _workerB   s2    

r$   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   r%   f   s   r%   c               @   sZ   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Ze	j
jje_dS )ThreadPoolExecutorN r   c             C   s   |dkrt  pdd }|dkr(td|dk	r@t|s@td|| _t | _t	 | _
d| _d| _t | _|p|d|   | _|| _|| _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: An 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)os	cpu_count
ValueErrorcallable	TypeError_max_workersqueueZSimpleQueue_work_queueset_threads_brokenr   	threadingZLock_shutdown_lock_counter_thread_name_prefix_initializer	_initargs)r   Zmax_workersZthread_name_prefixr!   r"   r   r   r   r   q   s     

zThreadPoolExecutor.__init__c          	   O   sj   | j Z | jrt| j| jr&tdtr2tdt }t||||}| j	| | 
  |S Q R X d S )Nz*cannot schedule new futures after shutdownz6cannot schedule new futures after interpreter shutdown)r7   r5   r%   r   RuntimeErrorr   ZFuturer   r2   r   _adjust_thread_count)r   r   r   r   fwr   r   r   submit   s    
zThreadPoolExecutor.submitc             C   s   | j fdd}t| j}|| jk r~d| jp,| |f }tj|tt	| || j | j
| jfd}d|_|  | j| | j t|< d S )Nc             S   s   | d  d S )N)r   )_r
   r   r   r   
weakref_cb   s    z;ThreadPoolExecutor._adjust_thread_count.<locals>.weakref_cbz%s_%d)nametargetr   T)r2   lenr4   r0   r9   r6   ZThreadr$   weakrefrefr:   r;   Zdaemonstartaddr   )r   rB   Znum_threadsZthread_namer	   r   r   r   r=      s    



z'ThreadPoolExecutor._adjust_thread_countc          	   C   sd   | j T d| _xFy| j }W n tjk
r6   P Y nX |d k	r|jt| j qW W d Q R X d S )NzBA thread initializer failed, the thread pool is not usable anymore)	r7   r5   r2   Z
get_nowaitr1   ZEmptyr   r   r%   )r   r#   r   r   r   r      s    z&ThreadPoolExecutor._initializer_failedTc          	   C   sD   | j  d| _| jd  W d Q R X |r@x| jD ]}|  q0W d S )NT)r7   r   r2   r   r4   r   )r   waitr	   r   r   r   shutdown   s    zThreadPoolExecutor.shutdown)Nr(   Nr   )T)r   r   r   	itertoolscount__next__r8   r   r@   r   Executorr&   r=   r   rK   r   r   r   r   r'   l   s   
 

r'   )r&   
__author__atexitZconcurrent.futuresr   rL   r1   r6   rF   r+   WeakKeyDictionaryr   r   r   registerobjectr   r$   ZBrokenExecutorr%   rO   r'   r   r   r   r   <module>   s    	
$