B

 ��g&�@s:dZd
dl
mZ
d
dlmZmZmZ
Gdd�dej�ZdS)zA Fixer for imports of itertools.(imap|ifilter|izip|ifilterfalse) �)
�
fixer_base)�	BlankLine�syms�tokenc@s"eZ
dZd
Zde�Zdd�ZdS)�FixItertoolsImportsTzT
              import_from< 'from' 'itertools' 'import' imports=any >
              cCsf
|d
}|jt
jks|js"|g
}n|j}x�|ddd�D]z}|jtjkrT|j}|}n|jtjkrddS|jd}|j}|dkr�d|_|��
q8|dkr8|
�	�
|ddkr�dnd	|_q8W|jdd�p�|g
}d
}	x.|D]&}|	r�|jtj
kr�|��
q�|	d
N}	q�Wx*|�
r(|djtj
k�
r(|����
�
qW|j�
s@t|dd��
rL|j
dk�
rb|
j}
t�}
|
|
_|
SdS)
N�imports�r
)ZimapZizipZifilter)ZifilterfalseZizip_longest�
�
f�filterfalse�zip_longestT����value)�typerZimport_as_name�childrenr�NAMEr�STAR�removeZchanged�COMMA�pop�getattr�parent�prefixr)�selfZnode�resultsrrZchild�memberZ	name_node�member_nameZremove_comma�
p�r�?/usr/local/lib/python3.7/lib2to3/fixes/fix_itertools_imports.py�	transformsB






























zFixItertoolsImports.transformN)�__name__�
__module__�__qualname__Z
BM_compatible�localsZPATTERNr rrrrrs

rN)	�__doc__Zlib2to3rZlib2to3.fixer_utilrrrZBaseFixrrrrr�<module>
s