Yf @ sg d Z d d l m Z m Z d d l m Z m Z m Z m Z m Z m
Z
Gd d d e j Z d S)z7
Convert use of sys.exitfunc to use the atexit module.
)pytree
fixer_base)NameAttrCallCommaNewlinesymsc sR e Z d Z d Z d Z d Z f d d Z f d d Z d d 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__ ?/opt/alt/python35/lib64/python3.5/lib2to3/fixes/fix_exitfunc.pyr s zFixExitfunc.__init__c s& t t | j | | d | _ d S)N)r r
start_tree
sys_import)r
Ztreefilename)r r r r ! s zFixExitfunc.start_treec C s d | k r, | j d k r( | d | _ d S| d j } d | _ t j t j t t d t d } t | | g | j } | j
| | j d k r | j | d d S| j j d } | j
t j k r| j t | j t d d n | j j } | j j | j } | j } t j t j t d t d d g }
t j t j |
g } | j | d t | j | d
| d S) Nr func atexitregisterzKCan't find sys import; Please add an atexit import at the top of your file. import )r Zcloneprefixr ZNoder Zpowerr r r replaceZwarningZchildrentypeZdotted_as_namesZ append_childr parentindexZ import_nameZ simple_stmtZ insert_childr ) r
ZnodeZresultsr r ZcallnamesZcontaining_stmtZpositionZstmt_containerZ
new_importnewr r r transform% s2
zFixExitfunc.transform) __name__
__module__ __qualname__Zkeep_line_orderZ
BM_compatibleZPATTERNr r r$ r r )r r r
s r
N)
__doc__Zlib2to3r r Zlib2to3.fixer_utilr r r r r r ZBaseFixr
r r r r <module> s .