3
\�" � @ s. d dl Z d dlZd dlZd dlZddlmZmZ ddlmZ ddlm Z ddlm
Z
ddd d
d d d
gZ ej dkrzdZ
dZn&ej dko�eedd�Z
ejj� jd�Zer�e jjejd�anejadd � Zdd
� Zdd� Zdd� Zdd � Zd&dd�Z dd� Z d d � Z d d � Z g Z d d � Z!d!d"� Z"d#d$� Z#d%d
� Z$dS )'� N� )�get_start_method�set_start_method)�process)� reduction)�util�_main�freeze_support�set_executable�get_executable�get_preparation_data�get_command_line�import_main_path�win32F�frozenzpythonservice.exez
python.exec C s | a d S )N)� _python_exe)Zexe� r �-/usr/lib64/python3.6/multiprocessing/spawn.pyr
) s c C s t S )N)r r r r r r - s c C s$ t | �dkr | d dkr dS dS dS )z=
Return whether commandline indicates we are forking
� r z--multiprocessing-forkTFN)�len)�argvr r r �
is_forking4 s r c C sh t tj�rdi } xBtjdd� D ]0}|jd�\}}|dkrBd| |< q t|�| |< q W tf | � tj� dS )zE
Run code for process object if this in not the main process
r N�=�None)r �sysr �split�int�
spawn_main�exit)�kwds�arg�name�valuer r r r > s
c K sh t tdd�r(tjdgdd� | j� D � S d}|djdd � | j� D ��; }tj� }tg| d
|dg S d S ) zJ
Returns prefix of command line used for spawning a child process
r Fz--multiprocessing-forkc S s g | ] }d | �qS )z%s=%rr )�.0�itemr r r �
<listcomp>T s z$get_command_line.<locals>.<listcomp>z<from multiprocessing.spawn import spawn_main; spawn_main(%s)z, c s s | ]}d | V qdS )z%s=%rNr )r# r$ r r r � <genexpr>W s z#get_command_line.<locals>.<genexpr>z-cN)�getattrr �
executable�items�joinr Z _args_from_interpreter_flagsr )r �progZoptsr r r r
N s c C s\ t jdkr.ddl}tj|| �}|j|tj�}nddlm } ||j
_ | }t |�}t j
|� dS )z7
Run code specified by data received over pipe
r r Nr )�semaphore_tracker)r �platform�msvcrtr Z steal_handleZopen_osfhandle�os�O_RDONLY� r, Z_semaphore_trackerZ_fdr r )Z pipe_handleZ
parent_pidZ
tracker_fdr. Z
new_handle�fdr, Zexitcoder r r r \ s
r c C s^ t j| ddd��@}dtj� _z$tjj|�}t|� tjj|�}W d tj� `X W d Q R X |j � S )N�rbT)�closefd)
r/ �fdopenr �current_process� _inheritingr �pickle�load�prepare�
_bootstrap)r2 Z from_parentZpreparation_data�selfr r r r m s
c C s t tj� dd�rtd��d S )Nr7 Fa
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.)r' r r6 � RuntimeErrorr r r r �_check_not_importing_mainy s
r> c C s t � ttjtj� jd�}tjdk r2tjj� |d< t j
j � }y|j d�}W n t
k
r^ Y n X tj||< |j| |t jtjtj� t� d� t jd }t|jdd�}|dk r�||d< nht jd ks�t o�t �r t|d
d�}|dk �r tj
j|� o�tjdk �r tj
jtj|�}tj
j |�|d < |S ) zM
Return info about parent needed by child to unpickle process object
)�
log_to_stderr�authkeyN� log_levelr1 )r! �sys_path�sys_argv�orig_dir�dir� start_method�__main__r! �init_main_from_namer �__file__�init_main_from_path) r> �dictr Z_log_to_stderrr r6 r@ Z_loggerZgetEffectiveLevelr �path�copy�index�
ValueError� ORIGINAL_DIR�updater r/ �getcwdr �modulesr' �__spec__r- �WINEXE�
WINSERVICE�isabsr* �normpath)r! �drB �i� main_moduleZ
main_mod_name� main_pathr r r r � s>
c C s� d| kr| d t j� _d| kr,| d t j� _d| krD| d rDtj� d| kr^tj� j| d � d| krp| d t_ d| kr�| d t_
d| kr�t j | d � d| kr�| d t _
d | kr�t| d d
d � d | kr�t| d � nd
| kr�t| d
� dS )zE
Try to get current process ready to unpickle process object
r! r@ r? rA rB rC rE rD rF T)ZforcerH rJ N)r r6 r! r@ r r? Z
get_loggerZsetLevelr rL r r/ �chdirrP r �_fixup_main_from_name�_fixup_main_from_path)�datar r r r: � s,
r: c C s~ t jd }| dks | jd�r d S t|jdd �| kr6d S tj|� tjd�}t j
| ddd�}|j j |� | t jd< t jd< d S )NrG z .__main__r! � __mp_main__T)�run_nameZ alter_sys)
r rS �endswithr' rT �old_main_modules�append�types�
ModuleType�runpyZ
run_module�__dict__rQ )Zmod_name� current_mainr[ � main_contentr r r r^ � s
r^ c C s� t jd }tjjtjj| ��d }|dkr.d S t|dd �| krBd S tj|� t j
d�}t j | dd�}|j
j|� | t jd< t jd< d S )NrG r ZipythonrI ra )rb )r rS r/ rL �splitext�basenamer' rd re rf rg rh Zrun_pathri rQ )r\ rj Z main_namer[ rk r r r r_ � s
r_ c C s t | � dS )z<
Set sys.modules['__main__'] to module at main_path
N)r_ )r\ r r r r s )NN)%r/ r rh rf r1 r r r �contextr r �__all__r- rU rV r' r( �lowerrc rL r* � exec_prefixr r
r r r r
r r r> r rd r: r^ r_ r r r r r �<module> s>
2&