File: //opt/alt/python35/lib64/python3.5/lib2to3/fixes/__pycache__/fix_exitfunc.cpython-35.opt-1.pyc
��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)�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/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
Ztree�filename)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� �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$ 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 .