B
    18\                 @   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   ]/oak/stanford/groups/akundaje/marinovg/programs/Python-3.7.3/Lib/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   set_running_or_notify_cancelr   r   r   BaseExceptionset_exception
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   LOGGERcritical_initializer_failedgetr   r   r   )executor_referenceZ
work_queueinitializerinitargs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queueSimpleQueue_work_queueset_threads_brokenr   	threadingLock_shutdown_lock_counter_thread_name_prefix_initializer	_initargs)r   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)r?   r<   r+   r   RuntimeErrorr   Futurer   r9   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)r9   lenr;   r6   rA   r=   Threadr*   weakrefrefrB   rC   daemonstartaddr   )r   rL   Znum_threadsZthread_namer	   r   r   r   rG      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)	r?   r<   r9   
get_nowaitr7   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)r?   r   r9   r   r;   r   )r   waitr	   r   r   r   shutdown   s    zThreadPoolExecutor.shutdown)Nr.   Nr   )T)r   r   r   	itertoolscount__next__r@   r   rJ   r   Executorr,   rG   r#   rY   r   r   r   r   r-   l   s   
 

r-   )r,   
__author__atexitconcurrent.futuresr   rZ   r7   r=   rQ   r1   WeakKeyDictionaryr   r   r   registerobjectr   r*   BrokenExecutorr+   r]   r-   r   r   r   r   <module>   s    	
$