3
\& @ s: d Z ddlmZ ddlmZmZmZ G dd dejZdS )zA Fixer for imports of itertools.(imap|ifilter|izip|ifilterfalse) )
fixer_base) BlankLinesymstokenc @ s" e Zd ZdZde Zdd ZdS )FixItertoolsImportsTzT
import_from< 'from' 'itertools' 'import' imports=any >
c C s~ |d }|j tjks |j r$|g}n|j}x|d d d D ]}|j tjkrV|j}|}n*|j tjkrfd S |j tjksvt|jd }|j}|dkrd |_|j q:|dkr:|j
|d d
krd nd |_q:W |jd d p|g}d
} x2|D ]*}| r|j tj kr|j q| d
N } qW x*|r>|d j tj kr>|j j qW |jpRt
|dd sd|jd krz|j}
t }|
|_|S d S )Nimports r imapizipifilter ifilterfalse izip_longest f filterfalse zip_longestTvalue)r r
r )r r
)typer Zimport_as_namechildrenr NAMEr STARAssertionErrorremoveZchangedCOMMApopgetattrparentprefixr ) selfZnodeZresultsr r ZchildmemberZ name_node member_nameZ remove_commap r# ;/usr/lib64/python3.6/lib2to3/fixes/fix_itertools_imports.py transform sD
z FixItertoolsImports.transformN)__name__
__module__ __qualname__Z
BM_compatiblelocalsZPATTERNr% r# r# r# r$ r s
r N) __doc__Zlib2to3r Zlib2to3.fixer_utilr r r ZBaseFixr r# r# r# r$ s