Current File : //snap/core22/current/usr/lib/python3.10/lib2to3/fixes/__pycache__/fix_exitfunc.cpython-310.pyc
o

��bc�	�@sJdZddlmZmZddlmZmZmZmZm	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<eZdZdZdZdZ�fdd�Z�fdd�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 >
              )
              cstt|�j|�dS�N)�superr
�__init__)�self�args��	__class__��1/usr/lib/python3.10/lib2to3/fixes/fix_exitfunc.pyr
szFixExitfunc.__init__cstt|��||�d|_dSr)rr
�
start_tree�
sys_import)r�tree�filenamerrrr!s
zFixExitfunc.start_treecCs(d|vr|jdur|d|_dS|d��}d|_t�tjttd�td���}t	||g|j�}|�
|�|jdurA|�|d�dS|jjd}|j
tjkr]|�t��|�tdd��dS|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�)r�clone�prefixr�Noder	�powerrrr�replace�warning�children�type�dotted_as_names�append_childr�parent�index�import_name�simple_stmt�insert_childr)r�node�resultsrr�call�names�containing_stmt�position�stmt_container�
new_import�newrrr�	transform%s6

�

�zFixExitfunc.transform)
�__name__�
__module__�__qualname__�keep_line_order�
BM_compatible�PATTERNr
rr8�
__classcell__rrrrr
sr
N)
�__doc__�lib2to3rr�lib2to3.fixer_utilrrrrrr	�BaseFixr
rrrr�<module>s