U
    .d"                     @   s   d dl Z d dlZd dlZd dlZd dlmZmZ ejdkrDdZdZ	n:d dl
Z
d dlmZ ejdkoleeddZej dZ	e	re jejdZnejZd	d
 Zdd ZdddZg Zdd Zdd Zdd ZdS )    N)processutilwin32F)	duplicatefrozenzpythonservice.exez
python.exec                   C   s   t S )N)_python_exe r   r   G/tmp/pip-unpacked-wheel-ho8k9mhl/joblib/externals/loky/backend/spawn.pyget_executable   s    r
   c                   C   s   t t ddrtdd S )NZ_inheritingFa  
        An attempt has been made to start a new process before the
        current process has finished its bootstrapping phase.

        This probably means that you are not using fork to start your
        child processes and you have forgotten to use the proper idiom
        in the main module:

            if __name__ == '__main__':
                freeze_support()
                ...

        The "freeze_support()" line can be omitted if the program
        is not going to be frozen to produce an executable.)getattrr   current_processRuntimeErrorr   r   r   r	   _check_not_importing_main#   s    r   Tc           
      C   s  t   ttjtt j| tj	tj
t d}dd tjD |d< tjdk	rztj |d< tjjrztjjd }|jj|d< d	d
lm} |  d|ji|d< tjdkrtt|jdd}||d d< n|j|d d< tjdkrtjdkrdd
lm} |  |j|jd|d< |rtjd }zt |j!dd}W n t"k
rP   d}Y nX |dk	rf||d< nhtjdks~t#st$st |dd}	|	dk	rtj%|	stj
dk	rtj&tj
|	}	tj'|	|d< |S )zM
    Return info about parent needed by child to unpickle process object
    )log_to_stderrauthkeynamesys_argvorig_dirdirc                 S   s   g | ]}|d kr|nt jqS ) )r   ORIGINAL_DIR).0pr   r   r	   
<listcomp>D   s     z(get_preparation_data.<locals>.<listcomp>sys_pathN	log_levelr   log_fmt   _resource_trackerpidtracker_argsr   T)inheritablefhfd)      posix)r$   r    mp_tracker_args__main__r   init_main_from_name__file__init_main_from_path)(r   dictr   Z_log_to_stderrbytesr   r   r   sysargvr   osgetcwdpathZ_loggergetEffectiveLevelhandlers	formatter_fmtresource_trackerr   Zensure_running_pidplatformr   msvcrtZget_osfhandle_fdversion_infor    multiprocessing.resource_trackermodulesr   __spec__BaseExceptionWINEXE
WINSERVICEisabsjoinnormpath)
r   Zinit_main_moduledhr   Zchild_wmp_resource_trackermain_moduleZmain_mod_name	main_pathr   r   r	   get_preparation_data5   s`    



 	 





rL   c                 C   s  d| kr| d t  _d| kr,| d t  _d| krD| d rDt  d| kr^t | d  d| krddl}t j	d 
|| d  d| kr| d t_d	| kr| d	 t_d
| krt| d
  d| kr| d t _d| kr
ddlm} | d d |_| d d |_d| krdddlm} | d d |_tjdkrV| d d }t|d|_n| d d |_d| kr|t| d  nd| krt| d  dS )zE
    Try to get current process ready to unpickle process object
    r   r   r   r   r   r   Nr   r   r   r   r(   r   r$   r    r!   r   r   r#   r*   r,   )r   r   r   r   r   r   Z
get_loggersetLevelloggingr5   setFormatter	Formatterr/   r3   r0   r1   chdirr   r>   r   r<   r9   r8   r:   r;   open_osfhandle_fixup_main_from_name_fixup_main_from_path)datarN   rI   r   handler   r   r	   prepare   sH    






rW   c                 C   s~   t jd }| dks| dr d S t|jdd | kr6d S t| td}t	j
| ddd}|j| | t jd< t jd< d S )Nr)   z	.__main__r   __mp_main__T)run_nameZ	alter_sys)r/   r?   endswithr   r@   old_main_modulesappendtypes
ModuleTyperunpyZ
run_module__dict__update)mod_namecurrent_mainrJ   main_contentr   r   r	   rS      s    


rS   c                 C   s   t jd }tjtj| d }|dkr.d S t|dd | krBd S t| t	
d}tj| dd}|j| | t jd< t jd< d S )Nr)   r   Zipythonr+   rX   )rY   )r/   r?   r1   r3   splitextbasenamer   r[   r\   r]   r^   r_   Zrun_pathr`   ra   )rK   rc   Z	main_namerJ   rd   r   r   r	   rT      s    


rT   )T)r1   r/   r_   r]   multiprocessingr   r   r:   rB   rC   r;   Zmultiprocessing.reductionr   r   
executablelowerrZ   r3   rE   exec_prefixr   r
   r   rL   r[   rW   rS   rT   r   r   r   r	   <module>	   s*   

O9