Format: application/x-python-code
1 o
2
3 ��h� � @ s d Z ddlZddlZddlZddlZddlmZmZ ddlmZm Z ddl
4 m Z m Z ddl
5 mZ ddlmZmZmZ dd lmZ dd
6 lmZ G d d � d e�Zd
7 d� Zdd� Zdd� Z z$ej j j Z eee�ge _ ee�e _!ej j"j#Z#e e#_$e ge#_ ddd�ZW n e%y� eZY nw dd� Z&dS )zWindows-specific implementation of process utilities.
8
9 This file is only meant to be imported by process.py, not by end-users.
10 � N)�c_int�POINTER)�LPCWSTR�HLOCAL)�STDOUT�TimeoutExpired)�Thread� )�read_no_interrupt�process_handler� arg_split)� py3compat)�DEFAULT_ENCODINGc @ s e Zd ZdZdd� Zdd� ZdS )� AvoidUNCPatha� A context manager to protect command execution from UNC paths.
11
12 In the Win32 API, commands can't be invoked with the cwd being a UNC path.
13 This context manager temporarily changes directory to the 'C:' drive on
14 entering, and restores the original working directory on exit.
15
16 The context manager returns the starting working directory *if* it made a
17 change and None otherwise, so that users can apply the necessary adjustment
18 to their system calls in the event of a change.
19
20 Examples
21 --------
22 ::
23 cmd = 'dir'
24 with AvoidUNCPath() as path:
25 if path is not None:
26 cmd = '"pushd %s &&"%s' % (path, cmd)
27 os.system(cmd)
28 c C s2 t �� | _| j�d�| _| jrt �d� | jS d S )Nz\\zC:)�os�getcwd�path�
29 startswith� is_unc_path�chdir)�self� r �Rc:\Users\shubh\Videos\agent\nova\Lib\site-packages\IPython/utils/_process_win32.py� __enter__9 s
30
31 zAvoidUNCPath.__enter__c C s | j r t�| j� d S d S )N)r r r r )r �exc_type� exc_value� tracebackr r r �__exit__E s �zAvoidUNCPath.__exit__N)�__name__�
32 __module__� __qualname__�__doc__r r r r r r r % s r c s` t � � �fdd�}� �fdd�}t|d��� t|d��� ��� }|du r-t�d� n|S q ) zCallback for _system.c �2 t �j��� D ]} | �� d�} t| tjd� qd S �N�replace)�file)r
33 �stdout�
34 splitlines�decode�print�sys��line��enc�pr r � stdout_readN � �z!_system_body.<locals>.stdout_readc r" r# )r
35 �stderrr' r( r) r* r+ r- r r � stderr_readS r1 z!_system_body.<locals>.stderr_read)�targetTNg{�G�z�?)r r �start�poll�time�sleep)r/ r0 r3 �resultr r- r � _system_bodyJ s �r: c C sH t � �}|durd|| f } t| t�W d � S 1 s w Y dS )a7 Win32 version of os.system() that works with network shares.
36
37 Note that this implementation returns None, as meant for use in IPython.
38
39 Parameters
40 ----------
41 cmd : str or list
42 A command to be executed in the system shell.
43
44 Returns
45 -------
46 int : child process' exit code.
47 N�"pushd %s &&"%s)r r r: )�cmdr r r r �systemf s
48 $�r= c C s` t � �}|durd|| f } t| dd� t�}W d � n1 s w Y |du r+d}t�|�S )z�Return standard output of executing cmd in a shell.
49
50 Accepts the same arguments as os.system().
51
52 Parameters
53 ----------
54 cmd : str or list
55 A command to be executed in the system shell.
56
57 Returns
58 -------
59 stdout : str
60 Nr; c S s | � � d S )Nr )� communicate)r/ r r r �<lambda>� s zgetoutput.<locals>.<lambda>� )r r r r
61 r( )r< r �outr r r � getoutput~ s �
62 rB FTc C sn | � � dkrg S |st| ||d�S t� }t| �� t�|��}t|j }dd� |� t�
63 |j ��D �}t |�}|S )a4 Split a command line's arguments in a shell-like manner.
64
65 This is a special version for windows that use a ctypes call to CommandLineToArgvW
66 to do the argv splitting. The posix parameter is ignored.
67
68 If strict=False, process_common.arg_split(...strict=False) is used instead.
69 � )�posix�strictc S s g | ]}|�qS r r )�.0�argr r r �
70 <listcomp>� s z arg_split.<locals>.<listcomp>)
71 �strip� py_arg_splitr �CommandLineToArgvW�lstrip�ctypes�byrefr �value� from_address� addressof�contents� LocalFree)Z commandlinerD rE ZargvnZresult_pointerZresult_array_typer9 �retvalr r r r � s
72 r c C s t tjj�dd| ��S )Nr r )�boolrM �windll�kernel32� OpenProcess)�pidr r r � check_pid� s rZ )FT)'r! r r* rM r7 r r Zctypes.wintypesr r �
73 subprocessr r � threadingr Z_process_commonr
74 r r rJ rC r
75 �encodingr �objectr r: r= rB rV �shell32rK Z arg_types�restyperW rS Zres_type�AttributeErrorrZ r r r r �<module> s8 %
76
77
78 �