o
    JAf                     @   s   d dl Z d dlZd dlZd dlZd dlZddlmZmZmZ ddl	m
Z
 ddl	mZ dgZdZejdko9eed	d
Zej dZdd Zeejej Zdd ZG dd deZdS )    N   )	reductionget_spawning_popenset_spawning_popen)spawn)utilPopeni   win32frozenFzpythonservice.exec                 C   s    | |kpt j| t j|kS N)ospathnormcase)Zp1Zp2 r   8/usr/lib/python3.10/multiprocessing/popen_spawn_win32.py_path_eq   s    r   c                  G   s   | D ]}t | qd S r   )_winapiCloseHandle)Zhandleshandler   r   r   _close_handles   s   r   c                   @   sJ   e Zd ZdZdZdd Zdd Zddd	Zd
d Zdd Z	e	Z
dd ZdS )r   z@
    Start a subprocess to run the code of a process object
    r   c                 C   sl  t |j}td d\}}t|d}t jt	 |d}d
dd |D }t  }trAt|tjrAtj}tj }tj|d< nd }t|ddd	b}	zt||d d d
d|d d 	\}
}}}t| W n	   t|  || _d | _|
| _t|
| _t| t| jt|f| _t|  zt !||	 t !||	 W td  ntd  w W d    d S 1 sw   Y  d S )Nr   )Z
parent_pidZpipe_handle c                 s   s    | ]}d | V  qdS )z"%s"Nr   ).0xr   r   r   	<genexpr>9   s    z!Popen.__init__.<locals>.<genexpr>__PYVENV_LAUNCHER__wbT)closefdF)"r   Zget_preparation_data_namer   Z
CreatePipemsvcrtopen_osfhandleZget_command_liner   getpidjoinZget_executableWINENVr   sys
executable_base_executableenvironcopyopenZCreateProcessr   pid
returncode_handleintsentinelr   Finalizer   	finalizerr   r   dump)selfprocess_objZ	prep_dataZrhandleZwhandleZwfdcmdZ
python_exeenvZto_childZhpZhtr)   tidr   r   r   __init__,   sH   



"zPopen.__init__c                 C   s   | t  u sJ t|| jS r   )r   r   Z	duplicater-   )r1   r   r   r   r   duplicate_for_childa   s   zPopen.duplicate_for_childNc                 C   st   | j d u r7|d u rtj}ntdt|d d }tt| j|}|tjkr7t| j}|t	kr4t
j }|| _ | j S )Nr   i  g      ?)r*   r   ZINFINITEmaxr,   ZWaitForSingleObjectr+   ZWAIT_OBJECT_0ZGetExitCodeProcess	TERMINATEsignalSIGTERM)r1   timeoutmsecsrescoder   r   r   waite   s   

z
Popen.waitc                 C   s   | j ddS )Nr   r<   )r@   r1   r   r   r   pollu      z
Popen.pollc                 C   sN   | j d u r%ztt| jt W d S  ty$   | jddd u r! Y d S w d S )Ng      ?rA   )r*   r   ZTerminateProcessr,   r+   r9   OSErrorr@   rB   r   r   r   	terminatex   s   
zPopen.terminatec                 C   s   |    d S r   )r/   rB   r   r   r   close   rD   zPopen.closer   )__name__
__module____qualname____doc__methodr6   r7   r@   rC   rF   killrG   r   r   r   r   r   &   s    5
)r   r   r:   r#   r   contextr   r   r    r   r   __all__r9   platformgetattrZWINEXEr$   lowerendswithZ
WINSERVICEr   r%   r"   r   objectr   r   r   r   r   <module>   s     
