o

�
i&�@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 >
              cCsb
|d
}|jt
jks
|js|g
}n|j}|ddd�D]>}|jtjkr)|j}|}n|jtjkr2
dS|jd}|j}|dvrFd|_|��
q|dvrY|
�	�
|ddkrVdn
d	|_q|jdd�pc|g
}d
}	|D]}|	rw|jtj
krw|��
qh|	d
N}	qh|r�|djtj
kr�|����
|r�|djtj
ks�|js�t|dd�r�|j
dur�|
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/local/lib/python3.10/lib2to3/fixes/fix_itertools_imports.py�	transformsJ

















��






�





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

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