B
VL{ÄýÉ{^ã @ 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__© ú?/opt/alt/python37/lib64/python3.7/lib2to3/fixes/fix_exitfunc.pyr s zFixExitfunc.__init__c s t t| ƒ ||¡ d | _d S )N)r r
Ú
start_treeÚ
sys_import)r
ZtreeÚfilename)r r r r ! s zFixExitfunc.start_treec C s& d|kr | j d kr |d | _ d S |d ¡ }d|_t tjttdƒtdƒƒ¡}t ||g|jƒ}|
|¡ | j d kr‚| |d¡ d S | j j d }|j
tjkr¸| tƒ ¡ | tddƒ¡ nj| 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 ZcloneÚprefixr ZNoder Zpowerr r r ÚreplaceZwarningZchildrenÚtypeZdotted_as_namesZ append_childr ÚparentÚindexZ import_nameZ simple_stmtZ insert_childr ) r
ZnodeZresultsr r ZcallÚnamesZcontaining_stmtZpositionZstmt_containerZ
new_importÚnewr r r Ú transform% s2
zFixExitfunc.transform)
Ú__name__Ú
__module__Ú __qualname__Zkeep_line_orderZ
BM_compatibleZPATTERNr r r$ Ú
__classcell__r r )r r r
s r
N)
Ú__doc__Zlib2to3r r Zlib2to3.fixer_utilr r r r r r ZBaseFixr
r r r r Ú s