o

��bc&�@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 >
              cCsr
|d
}|jt
jks
|js|g
}n|j}|ddd�D]F}|jtjkr)|j}|}n|jtjkr2
dS|jt
jks:J�
|jd}|j}|dvrNd|_|��
q|dvra|
�	�
|ddkr^dn
d	|_q|jdd�pk|g
}d
}	|D]}|	r|jtj
kr|��
qp|	d
N}	qp|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)�typer�import_as_name�childrenr�NAMEr�STAR�remove�changed�COMMA�pop�getattr�parent�prefixr)�self�node�resultsrr�child�member�	name_node�member_name�remove_comma�
p�r)�:/usr/lib/python3.10/lib2to3/fixes/fix_itertools_imports.py�	transformsL


















��






�





�zFixItertoolsImports.transformN)�__name__�
__module__�__qualname__�
BM_compatible�locals�PATTERNr+r)r)r)r*rs

�rN)	�__doc__�lib2to3r�lib2to3.fixer_utilrrr�BaseFixrr)r)r)r*�<module>
s