a

R�f&�@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 >
              cCsp
|d
}|jt
jks|js"|g
}n|j}|ddd�D]�}|jtjkrR|j}|}n,|jtjkrd
dS|jt
jkstJ�
|jd}|j}|dvr�d|_|��
q6|dvr6|
�	�
|ddkr�dnd	|_q6|jdd�p�|g
}d
}	|D]*}|	�
r|jtj
k�
r|��
q�|	d
N}	q�|�
r4|djtj
k�
r4|����
�
q|j�
sJt|dd��
rV|j
du�
rl|
j}
t�}
|
|
_|
SdS)
N�imports�r
)�imap�izip�ifilter)�ifilterfalse�izip_longest�
�
f�filterfalse�zip_longestT����value)�typerZimport_as_name�childrenr�NAMEr�STAR�remove�changed�COMMA�pop�getattr�parent�prefixr)�selfZnode�resultsrr�child�memberZ	name_node�member_nameZremove_comma�
p�r%�;/usr/lib64/python3.9/lib2to3/fixes/fix_itertools_imports.py�	transformsH


















�







�


zFixItertoolsImports.transformN)�__name__�
__module__�__qualname__Z
BM_compatible�localsZPATTERNr'r%r%r%r&rs


�rN)	�__doc__Zlib2to3rZlib2to3.fixer_utilrrrZBaseFixrr%r%r%r&�<module>
s