CasperSecurity
o
��bc� � @ sJ d Z ddlmZmZ ddlmZmZmZmZm Z m
Z
G dd� dej �Z dS )z7
Convert use of sys.exitfunc to use the atexit module.
� )�pytree�
fixer_base)�Name�Attr�Call�Comma�Newline�symsc s< e Zd ZdZdZdZ� 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 >
)
c s t t| �j|� d S �N)�superr
�__init__)�self�args�� __class__� �1/usr/lib/python3.10/lib2to3/fixes/fix_exitfunc.pyr
s zFixExitfunc.__init__c s t t| ��||� d | _d S r )r r
�
start_tree�
sys_import)r �tree�filenamer r r r ! s
zFixExitfunc.start_treec C s( d|v r| j d u r|d | _ d S |d �� }d|_t�tjttd�td���}t ||g|j�}|�
|� | j d u rA| � |d� d S | j j d }|j
tjkr]|�t� � |�tdd�� d S | j j}|j �| j �}|j} t�tjtd �tdd�g�}
t�tj|
g�} |�|d t� � |�|d
| � d S ) Nr �func� �atexit�registerzKCan't find sys import; Please add an atexit import at the top of your file.� � �import� )r �clone�prefixr �Noder �powerr r r �replace�warning�children�type�dotted_as_names� append_childr �parent�index� import_name� simple_stmt� insert_childr ) r �node�resultsr r �call�names�containing_stmt�position�stmt_container�
new_import�newr r r � transform% s6
�
�zFixExitfunc.transform)
�__name__�
__module__� __qualname__�keep_line_order�
BM_compatible�PATTERNr
r r8 �
__classcell__r r r r r
s r
N)
�__doc__�lib2to3r r �lib2to3.fixer_utilr r r r r r �BaseFixr
r r r r �<module> s