a dP$@sddlZddlZddlZddlZddlmZmZddlmZddlm Z ddlm Z gdZ ej dkrpd Z d Zneed d Z ejd Zerejejd anejad dZddZddZddZddZd'ddZddZddZddZgZ dd Z!d!d"Z"d#d$Z#d%d&Z$dS)(N)get_start_methodset_start_method)process) reduction)util)_mainfreeze_supportset_executableget_executableget_preparation_dataget_command_lineimport_main_pathwin32Ffrozenzpythonservice.exez python.execCs|adSN _python_exe)exer-/usr/lib64/python3.9/multiprocessing/spawn.pyr )sr cCstSrrrrrrr -sr cCs$t|dkr|ddkrdSdSdS)z= Return whether commandline indicates we are forking r--multiprocessing-forkTFN)len)argvrrr is_forking4srcCshttjrdi}tjddD]0}|d\}}|dkr@d||<qt|||<qtfi|tdS)zE Run code for process object if this in not the main process rN=None)rsysrsplitint spawn_mainexit)kwdsargnamevaluerrrr >s  r cKshttddr(tjdgdd|DSd}|ddd |D;}t}tg|d |dgSd S) zJ Returns prefix of command line used for spawning a child process rFrcSsg|] }d|qS)%s=%rr.0itemrrr Tz$get_command_line..zWr,z#get_command_line..z-cN)getattrr executableitemsjoinr_args_from_interpreter_flagsr)r#progoptsrrrr Ns r c CsttjsJdtjdkrrddl}ddl}|durL||j|jBd|}nd}t j ||d}| |t j }|}n"ddlm} || j_|}t |}t||} t| dS) z7 Run code specified by data received over pipe z Not forkingrrNF)source_processr)resource_tracker)rrrplatformmsvcrt_winapi OpenProcess SYNCHRONIZEPROCESS_DUP_HANDLEr duplicateopen_osfhandleosO_RDONLYr6_resource_tracker_fdduprr") pipe_handle parent_pid tracker_fdr8r9r5 new_handlefdparent_sentinelr6exitcoderrrr!\s*     r!c Cs|tj|dddR}dt_z,tj|}t|tj|}Wt`n t`0Wdn1sh0Y| |S)NrbT)closefd) r?fdopenrcurrent_process _inheritingrpickleloadprepare _bootstrap)rIrJ from_parentpreparation_dataselfrrrrxs  2rcCsttddrtddS)NrPFa 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.rrO RuntimeErrorrrrr_check_not_importing_mainsrYcCstttjtjd}tjdur2tj|d<t j }z| d}Wnt y\Yn 0tj||<|j||t jtjttdt jd}t|jdd}|dur||d<nft jd kststst|d d}|durtj |stjdurtj tj|}tj ||d <|S) zM Return info about parent needed by child to unpickle process object ) log_to_stderrauthkeyN log_levelrA)r%sys_pathsys_argvorig_dirdir start_method__main__r%init_main_from_namer__file__init_main_from_path)rYdictr_log_to_stderrrrOr[_loggergetEffectiveLevelrpathcopyindex ValueError ORIGINAL_DIRupdaterr?getcwdrmodulesr.__spec__r7WINEXE WINSERVICEisabsr1normpath)r%dr]i main_module main_mod_name main_pathrrrr sD        r cCsd|vr|dt_d|vr,|dt_d|vrD|drDtd|vr^t|dd|vrp|dt_ d|vr|dt_ d|vrt |dd|vr|dt_ d |vrt|d d d d |vrt|d nd |vrt|d dS)zE Try to get current process ready to unpickle process object r%r[rZr\r]r^r`r_raT)forcercreN)rrOr%r[rrZ get_loggersetLevelrrjrr?chdirrnr_fixup_main_from_name_fixup_main_from_path)datarrrrSs,   rScCs~tjd}|dks|dr dSt|jdd|kr6dSt|td}t j |ddd}|j ||tjd<tjd<dS)Nrbz .__main__r% __mp_main__T)run_name alter_sys) rrqendswithr.rrold_main_modulesappendtypes ModuleTyperunpy run_module__dict__ro)mod_name current_mainry main_contentrrrrs    rcCstjd}tjtj|d}|dkr.dSt|dd|krBdSt|t d}t j |dd}|j ||tjd<tjd<dS)Nrbripythonrdr)r)rrqr?rjsplitextbasenamer.rrrrrrun_pathrro)r{r main_nameryrrrrr s    rcCs t|dS)z< Set sys.modules['__main__'] to module at main_path N)r)r{rrrr%sr)NN)%r?rrrrArrrcontextrr__all__r7rsrtr.r/lowerrrjr1 exec_prefixrr r rr r r!rrYr rrSrrrrrrr s<        2&