o
    JAfP$                     @   s  d dl Z d dlZd dlZd dlZddlmZmZ ddlmZ ddlm	Z	 ddlm
Z
 g dZejdkr8d	Zd	Zneed
d	Zej dZerQe jejdanejadd Zdd Zdd Zdd Zdd Zd'ddZdd Zdd Zdd Zg Z dd  Z!d!d" Z"d#d$ Z#d%d& Z$dS )(    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frozenzpythonservice.exez
python.exec                 C   s   | a d S N_python_exe)Zexe r   ,/usr/lib/python3.10/multiprocessing/spawn.pyr
   )   s   r
   c                   C   s   t S r   r   r   r   r   r   r   -   s   r   c                 C   s"   	 t | dkr| d dkrdS dS )N   r   --multiprocessing-forkTF)len)argvr   r   r   
is_forking4   s   r   c                  C   sn   	 t tjr5i } tjdd  D ]}|d\}}|dkr!d | |< qt|| |< qtdi |  t  d S d S )Nr   =Noner   )r   sysr   splitint
spawn_mainexit)kwdsargnamevaluer   r   r   r	   >   s   

r	   c                  K   sf   	 t tddrtjdgdd |  D  S d}|ddd	 |  D ; }t }tg| d
|dg S )Nr   Fr   c                 S   s   g | ]}d | qS )%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  qdS )r&   Nr   r'   r   r   r   	<genexpr>W   s    z#get_command_line.<locals>.<genexpr>z-c)getattrr   
executableitemsjoinr   _args_from_interpreter_flagsr   )r"   progZoptsr   r   r   r   N   s   r   c                 C   s   	 t jdkr1dd l}dd l}|d ur||j|jB d|}nd }tj| |d}|	|t
j}|}nddlm}	 ||	j_| }t
| }t||}
t |
 d S )Nr   r   F)source_processr   )resource_tracker)r   platformmsvcrt_winapiZOpenProcessZSYNCHRONIZEZPROCESS_DUP_HANDLEr   Z	duplicateopen_osfhandleosO_RDONLY r3   _resource_trackerZ_fddupr   r!   )Zpipe_handleZ
parent_pidZ
tracker_fdr5   r6   r2   Z
new_handlefdparent_sentinelr3   exitcoder   r   r   r    \   s*   



r    c              	   C   s|   t j| ddd)}dt _ztj|}t| tj|}W t `nt `w W d    n1 s4w   Y  |	|S )NrbT)closefd)
r8   fdopenr   current_process_inheritingr   pickleloadprepare
_bootstrap)r=   r>   Zfrom_parentZpreparation_dataselfr   r   r   r   x   s   

r   c                   C   s   t t ddrtdd S )NrD   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   rC   RuntimeErrorr   r   r   r   _check_not_importing_main   s   rK   c                 C   s  	 t   ttjt jd}tjd urtj |d< t	j
 }z|d}W n	 ty/   Y nw tj||< |j| |t	jtjt t d t	jd }t|jdd }|d ur[||d< |S t	jdksdtstst|d	d }|d urtj
|stjd urtj
tj|}tj
||d
< |S )N)log_to_stderrauthkey	log_levelr:   )r$   sys_pathsys_argvorig_dirdirstart_method__main__r$   init_main_from_namer   __file__init_main_from_path)rK   dictr   _log_to_stderrr   rC   rM   _loggergetEffectiveLevelr   pathcopyindex
ValueErrorORIGINAL_DIRupdater   r8   getcwdr   modulesr,   __spec__r4   WINEXE
WINSERVICEisabsr/   normpath)r$   drO   imain_moduleZmain_mod_name	main_pathr   r   r   r      sH   



	
r   c                 C   s   	 d| v r| d t  _d| v r| d t  _d| v r#| d r#t  d| v r0t | d  d| v r9| d t_	d| v rB| d t_
d| v rMt| d  d| v rV| d t _d	| v rbt| d	 d
d d| v rnt| d  d S d| v rzt| d  d S d S )Nr$   rM   rL   rN   rO   rP   rR   rQ   rS   T)forcerU   rW   )r   rC   r$   rM   r   rL   
get_loggersetLevelr   r\   r   r8   chdirr`   r   _fixup_main_from_name_fixup_main_from_path)datar   r   r   rG      s0   


rG   c                 C   s~   t jd }| dks| drd S t|jdd | krd S t| td}t	j
| ddd}|j| | t jd< t jd< d S )NrT   z	.__main__r$   __mp_main__T)run_nameZ	alter_sys)r   rc   endswithr,   rd   old_main_modulesappendtypes
ModuleTyperunpyZ
run_module__dict__ra   )Zmod_namecurrent_mainrk   main_contentr   r   r   rq      s   


rq   c                 C   s   t jd }tjtj| d }|dkrd S t|dd | kr!d S t| t	
d}tj| dd}|j| | t jd< t jd< d S )NrT   r   ZipythonrV   rt   )ru   )r   rc   r8   r\   splitextbasenamer,   rw   rx   ry   rz   r{   Zrun_pathr|   ra   )rl   r}   Z	main_namerk   r~   r   r   r   rr   	  s   


rr   c                 C   s   	 t |  d S r   )rr   )rl   r   r   r   r   %  s   r   )NN)%r8   r   r{   ry   r:   r   r   r   contextr   r   __all__r4   re   rf   r,   r-   lowerrv   r\   r/   exec_prefixr   r
   r   r   r	   r   r    r   rK   r   rw   rG   rq   rr   r   r   r   r   r   <module>   s>   



2&