a

�DOg&�@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 >
              cCsZ
|d
}|jt
jks|js"|g
}n|j}|ddd�D]|}|jtjkrR|j}|}n|jtjkrd
dS|jd}|j}|dvr�d|_|��
q6|dvr6|
�	�
|ddkr�dnd	|_q6|jdd�p�|g
}d
}	|D]&}|	r�|jtj
kr�|��
q�|	d
N}	q�|�
r|djtj
k�
r|����
q�|j�
s4t|dd��
r@|j
du�
rV|
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�	transformsF

















�







�


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


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