Format: application/x-python-code
1 o
2
3 ��h�3 � @ s� d Z ddlZddlZddlZddlZddlZddlZddlZddlZddl m
4 Z
5 ddl m Z ddl
6 mZmZmZmZ ddlmZ ddlmZ ddlmZmZmZmZ dd lm Z d
7 d � Z G d d
8 � d
9 e �Z eG dd� de��Z dS )z5Magic functions for running cells in various scripts.� N)�CalledProcessError)�Thread)�Any�Dict�List�default)�magic_arguments)�
10 _AsyncIOProxy)�Magics�
11 cell_magic�
12 line_magic� magics_class)� arg_splitc
13 C sb t jdtdd�t jdtdd�t jdddd �t jd
14 td d�t jd d
15 ddd�g}|D ]}|| �} q(| S )z'single decorator for adding script argsz--outz�The variable in which to store stdout from the script.
16 If the script is backgrounded, this will be the stdout *pipe*,
17 instead of the stderr text itself and will not be auto closed.
18 )�type�helpz--errz�The variable in which to store stderr from the script.
19 If the script is backgrounded, this will be the stderr *pipe*,
20 instead of the stderr text itself and will not be autoclosed.
21 z--bg�
22 store_truez�Whether to run the script in the background.
23 If given, the only way to see the output of the command is
24 with --out/err.
25 )�actionr z--procztThe variable in which to store Popen instance.
26 This is used only when --bg option is given.
27 z--no-raise-error� store_false� raise_errorz�Whether you should raise an error message in addition to
28 a stream on stderr if you get a nonzero exit code.
29 )r �destr )r Zargument�str)�f�args�arg� r �Pc:\Users\shubh\Videos\agent\nova\Lib\site-packages\IPython/core/magics/script.py� script_args s0 ������#
30 r c @ s e Zd ZdS )�RaiseAfterInterruptN)�__name__�
31 __module__� __qualname__r r r r r F s r c s� e Zd ZU dZedd�Zedd�jdd�Zee d< e
32 d�dd � �Z e d
33 d�jdd�Z
34 d"� fd d
35 � Zdd� Zdd� Zdd� Ze�� eed�dd� ���Zejdke_dd� Zed�d#d d ��Zd d � Z d d!� Z � Z S )$� ScriptMagicsz�Magics for talking to scripts
36
37 This defines a base `%%script` cell magic for running a cell
38 with a program in a subprocess, and registers a few top-level
39 magics that call %%script with common interpreters.
40 z�
41 The event loop on which to run subprocesses
42
43 Not the main event loop,
44 because we want to be able to make blocking calls
45 and have certain requirements we don't want to impose on the main loop.
46 )r z�Extra script cell magics to define
47
48 This generates simple wrappers of `%%script foo` as `%%foo`.
49
50 If you want to add script magics that aren't on your path,
51 specify them in script_paths
52 T)�config�
53 script_magicsc C s" g d�}t jdkr|�dg� |S )z$default to a common list of programs)�sh�bash�perl�ruby�python�python2�python3�pypy�nt�cmd)�os�name�extend)�self�defaultsr r r �_script_magics_defaultg s
54
55 �z#ScriptMagics._script_magics_defaultz�Dict mapping short 'ruby' names to full paths, such as '/opt/secret/bin/ruby'
56
57 Only necessary for items in script_magics where the default path will not
58 find the right interpreter.
59 Nc s0 t t| �j|d� | �� g | _t�| j� d S )N)�shell)�superr! �__init__�_generate_script_magics� bg_processes�atexit�register�kill_bg_processes)r1 r4 �� __class__r r r6 � s zScriptMagics.__init__c C s | � � d S �N)r; �r1 r r r �__del__� s zScriptMagics.__del__c C s( | j d }| jD ] }| �|�||< qd S )N�cell)Zmagicsr# �_make_script_magic)r1 Z cell_magicsr/ r r r r7 � s
60
61 �z$ScriptMagics._generate_script_magicsc s@ �j �||�� t�� t� �fdd���}djdi t� ��|_|S )zAmake a named magic, that calls %%script with a particular programc s" | r d� | f } n� } �� | |�S )Nz%s %s)�shebang)�linerA ��scriptr1 r r �named_script_magic� s z;ScriptMagics._make_script_magic.<locals>.named_script_magicz�%%{name} script magic
62
63 Run cells with {script} in a subprocess.
64
65 This is a shortcut for `%%script {script}`
66 Nr )� script_paths�getr r �format�locals�__doc__)r1 r/ rG r rE r rB � s
67 ���z ScriptMagics._make_script_magicrF c
68 s� �j du r$tjdkrt�� �� �nt�� ���_ t�jdd�}|�� n�j ��fdd�}dd� ���fd d
69 �� � �fd d �}t |tj�
70 d
71 � d�}�j j �
72 |�\�}z|tj|tjjtjjtjjd����W n" ty� } z|jtjkr�td|d � W Y d}~dS � d}~ww ��d�s��d7 ���dd���jr��j��� ��� g ��jr�t �j ���j j �j< n���j � �j r�t �j!���j j �j < n���j!� ��"����fdd�� �j#r�t ���} t �j ��| _ t �j!��| _!| �j j �j#< dS z ||���� W n� t$�y� z[��%t&j'� |tj(��)� dd�� �j*du�r5td� �j+�r1t,� �W Y dS ��-� |tj(��)� dd�� �j*du�rYtd� �j+�rUt,� �W Y dS ��.� td� �j+�rht,� �W n. t,�ys Y n% t�y| Y n t/�y� } ztd �j0|f � W Y d}~nd}~ww �j+�r�t1�j*��d�Y dS w �j+�r��j*dk�r��j*�p�d }
73 t1|
74 ���dS dS ) ah Run a cell via a shell command
75
76 The `%%script` line is like the #! line of script,
77 specifying a program (bash, perl, ruby, etc.) with which to run.
78
79 The rest of the cell is run by that program.
80
81 .. versionchanged:: 9.0
82 Interrupting the script executed without `--bg` will end in
83 raising an exception (unless `--no-raise-error` is passed).
84
85 Examples
86 --------
87 ::
88
89 In [1]: %%script bash
90 ...: for i in 1 2 3; do
91 ...: echo $i
92 ...: done
93 1
94 2
95 3
96 N�win32T)�target�daemonc s t �| � ��� S )z&Call a coroutine on the asyncio thread)�asyncio�run_coroutine_threadsafe�result)�coro)�
97 event_loopr r � in_thread� s z'ScriptMagics.shebang.<locals>.in_threadc
98 � s| �z | � d�I d H W S tjjy! } z |jW Y d }~S d }~w tjjy= } z| � |j�I d H W Y d }~S d }~ww )N�d )�readrP �
99 exceptions�IncompleteReadError�partial�LimitOverrunError�consumed)�stream�er r r �
100 _readchunk� s �� ��z(ScriptMagics.shebang.<locals>._readchunkc � sJ � � | �I d H j ddd�}|sd S |r|�jj|< n |�|� |�� q)NT�utf8�replace)�errors)�decoder4 �user_ns�write�flush)r] Z
101 stream_argZ file_object�chunk)r_ r1 r r �_handle_stream� s �
102 �z,ScriptMagics.shebang.<locals>._handle_streamc � sn �| j �|� | j �� t�� | j�jtj��}t�� | j�j tj��}t�
103 ||g�I d H | �
104 � I d H d S r> ) �stdinre �closerP � create_task�stdout�out�sys�stderr�err�wait)�processrA Z stdout_taskZ stderr_task)rh r r r �_stream_communicate� s �
105 ��z1ScriptMagics.shebang.<locals>._stream_communicate�win)�posix)rl ro ri zCouldn't find program: %rr �
106 r` ra c s t ����� ���S r> )rP �Task� _run_scriptr )rA �pr1 �to_closer r �<lambda> s z&ScriptMagics.shebang.<locals>.<lambda>皙�����?)�timeoutzProcess was interrupted.zProcess was terminated.zProcess was killed.z/Error while terminating subprocess (pid=%i): %si����)2rT rn �platformrP � WindowsProactorEventLoopPolicy�new_event_loopr � run_forever�startr �
107 startswithrC �parserZparse_known_args�create_subprocess_exec�
108 subprocess�PIPE�OSError�errno�ENOENT�print�endswith�encode�bgr8 �append�_gc_bg_processesrm r rl r4 rd rp ro �call_soon_threadsafe�proc�KeyboardInterrupt� send_signal�signal�SIGINT�wait_forrq �
109 returncoder r � terminate�kill� Exception�pidr ) r1 rD rA Zasyncio_threadrU rs �argvr- r^ Z
110 proc_proxy�rcr )rh r_ r rA rT ry r1 rz r rC � s�
111
112
113 �� ��
114 �
115 � ���!
116 �zScriptMagics.shebangru c � sj �|j �|� |j �� I dH |j �� |j �� I dH |�� I dH |D ] }|�� I dH q%| �� dS )z1callback for running the script in the backgroundN)ri re �drainrj � wait_closedrq rW r� )r1 ry rA rz �sr r r rx T s �
117 zScriptMagics._run_script�
118 killbgscripts� c C s | � � td� dS )z9Kill all BG processes started by %%script and its family.z%All background processes were killed.N)r; r� )r1 Z_nouse_r r r r� b s zScriptMagics.killbgscriptsc C s� | j sdS | j D ]}|jdu r z|�tj� W q Y qqt�d� | �� | j s,dS | j D ]}|jdu rBz|�� W q/ Y q/q/t�d� | �� | j sQdS | j D ]}|jdu rgz|� � W qT Y qTqT| �� dS )z.Kill all BG processes which are still running.Nr| )
119 r8 r� r� r� r� �time�sleepr� r� r� )r1 ry r r r r; h s@
120
121 �
122
123
124 �
125
126
127 � z ScriptMagics.kill_bg_processesc C s dd� | j D �| _ d S )Nc S s g | ] }|j d u r|�qS r> )r� )�.0ry r r r �
128 <listcomp>� s z1ScriptMagics._gc_bg_processes.<locals>.<listcomp>)r8 r? r r r r� � s z ScriptMagics._gc_bg_processesr> )r� ) r r r rL r rT r �tagr# �__annotations__r r3 r rH r6 r@ r7 rB r r r rC r. r/ Z__skip_doctest__rx r r� r; r� �
129 __classcell__r r r< r r! J s@
130 �
131 ��
132
133 �� $ r! )!rL rP Zasyncio.exceptionsr9 r� r. r� rn r� r� r � threadingr Z traitletsr r r r Z IPython.corer ZIPython.core.async_helpersr ZIPython.core.magicr
134 r r r
135 ZIPython.utils.processr r r� r r! r r r r �<module> s( *