XylotrechusZ
�
i f& � @ sO d Z d d l m Z d d l m Z m Z m Z Gd d � d e j � Z d S)zA Fixer for imports of itertools.(imap|ifilter|izip|ifilterfalse) � )�
fixer_base)� BlankLine�syms�tokenc @ s/ e Z d Z d Z d e � Z d d � Z d S)�FixItertoolsImportsTzT
import_from< 'from' 'itertools' 'import' imports=any >
c C s� | d } | j t j k s& | j r2 | g } n | j } x� | d d d � D]� } | j t j k ry | j } | } n; | j t j k r� d S| j t j k s� t � | j d } | j } | d k r� d | _ | j � qO | d k rO | j
� | d d
k r d n d | _ qO qO W| j d d � p4| g } d
} x= | D]5 } | ro| j t j k ro| j � qD| d
N} qDWx0 | r�| d j t j k r�| j � j � q�W| j p�t
| d d � s�| j d k r�| j }
t � } |
| _ | Sd S)N�imports� r �imap�izip�ifilter� ifilterfalse� izip_longest� �f� filterfalse� zip_longestT�value)r r
r )r r
���)�typer Zimport_as_name�childrenr �NAMEr �STAR�AssertionError�removeZchanged�COMMA�pop�getattr�parent�prefixr ) �selfZnodeZresultsr r ZchildZmemberZ name_nodeZ member_nameZ remove_comma�p� r! �H/opt/alt/python34/lib64/python3.4/lib2to3/fixes/fix_itertools_imports.py� transform sD
z FixItertoolsImports.transformN)�__name__�
__module__� __qualname__Z
BM_compatible�localsZPATTERNr# r! r! r! r" r s
r N) �__doc__Zlib2to3r Zlib2to3.fixer_utilr r r ZBaseFixr r! r! r! r" �<module> s