î
j f¿ ã @ 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)ÚNameÚAttrÚCallÚCommaÚNewlineÚsymsc 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__)ÚselfÚargs)Ú __class__© ú?/opt/alt/python34/lib64/python3.4/lib2to3/fixes/fix_exitfunc.pyr s zFixExitfunc.__init__c s& t t | ƒ j | | ƒ d | _ d S)N)r r
Ú
start_treeÚ
sys_import)r
ZtreeÚfilename)r r r r ! s zFixExitfunc.start_treec C s d | k r/ | j d k r+ | d | _ n 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Ú Ú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 ÚcallÚnamesZcontaining_stmtZpositionZstmt_containerZ
new_importÚnewr 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 Ú s .