Revonzy Mini Shell

a

�DOg�	�@sJdZd
dl
mZmZ
d
dlmZmZmZmZm	Z	m
Z

Gdd�dej�ZdS)z7
Convert use of sys.exitfunc to use the atexit module.
�)�pytree�
fixer_base)�Name�Attr�Call�Comma�Newline�symscs<eZ
dZd
Zd
ZdZ�f
dd�Z�f
dd�Zdd�Z�Z	S)	�FixExitfuncTa�

              (
                  sys_import=import_name<'import'
                      ('sys'
                      |
                      dotted_as_names< (any ',')* 'sys' (',' any)* >
                      )
                  >
              |
                  expr_stmt<
                      power< 'sys' trailer< '.' 'exitfunc' > >
                  '=' func=any >
              )
              c
stt
|�j|
�
dS�
N)�superr
�__init__)�self�args�
�	__class__��2/usr/lib64/python3.9/lib2to3/fixes/fix_exitfunc.pyr
s
zFixExitfunc.__init__cstt
|��|
|�
d|_dSr)rr
�
start_tree�
sys_import)rZtree�filenamerrrr!s

zFixExitfunc.start_treecCs&
d
|vr |jdur|d
|_dS|d�
�}d|_t�tjttd�
td�
��}t	||g
|
j�}|
�
|�

|jdur�|�|
d�
dS|jjd}|j
tjkr�|�t��

|�tdd��

nj|jj}|j�|j�
}|j}	t�tjtd	�
tdd�g�}
t�tj|
g
�}|�|dt��
|�|d
|�
dS)Nr�func��atexit�registerzKCan't find sys import; Please add an atexit import at the top of your file.�
�
 �import�)rZclone�prefixrZNoder	Zpowerrrr�replace�warningZchildren�typeZdotted_as_namesZappend_childr�parent�indexZimport_nameZsimple_stmtZinsert_childr)rZnode�resultsrr�call�namesZcontaining_stmtZpositionZstmt_containerZ
new_import�newrrr�	transform%s6







�









�

zFixExitfunc.transform)
�__name__�
__module__�__qualname__Zkeep_line_orderZ
BM_compatibleZPATTERNr
rr)�
__classcell__rrrrr
s

r
N)
�__doc__Zlib2to3rrZlib2to3.fixer_utilrrrrrr	ZBaseFixr
rrrr�<module>
s