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)NameAttrCallCommaNewlinesymsc s< e Zd ZdZdZdZ fddZ fddZdd 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__)selfargs __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 treefilenamer r r r ! s
zFixExitfunc.start_treec C s( d|v r| j d u r|d | _ d S |d }d|_ttjttdtd}t ||g|j}|
| | j d u rA| |d d S | j j d }|j
tjkr]|t |tdd d S | j j}|j | j }|j} ttjtd tddg}
ttj|
g} ||d t ||d
| d S ) Nr func atexitregisterzKCan't find sys import; Please add an atexit import at the top of your file. import )r cloneprefixr Noder powerr r r replacewarningchildrentypedotted_as_names append_childr parentindex import_name simple_stmt insert_childr ) r noderesultsr r callnamescontaining_stmtpositionstmt_container
new_importnewr r r transform% s6
zFixExitfunc.transform)
__name__
__module__ __qualname__keep_line_order
BM_compatiblePATTERNr
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