a
XC?h& � @ s: d Z ddlmZ ddlmZmZmZ G dd� dej�ZdS )zA Fixer for imports of itertools.(imap|ifilter|izip|ifilterfalse) � )�
fixer_base)� BlankLine�syms�tokenc @ s" e Zd ZdZde� Zdd� ZdS )�FixItertoolsImportsTzT
import_from< 'from' 'itertools' 'import' imports=any >
c C sp |d }|j tjks|js"|g}n|j}|d d d� D ]�}|j tjkrR|j}|}n,|j tjkrd d S |j tjkstJ �|jd }|j}|dv r�d |_|�� q6|dv r6|� � |d dkr�dnd |_q6|jd d � p�|g}d
} |D ]*}| �r|j tj
k�r|�� q�| d
N } q�|�r4|d j tj
k�r4|� � �� �q |j�sJt |d d ��rV|j
d u �rl|j}
t� }|
|_|S d S )
N�imports� r )ZimapZizipZifilter)Z ifilterfalseZ izip_longest� �f� filterfalse� zip_longestT������value)�typer Zimport_as_name�childrenr �NAMEr �STAR�removeZchanged�COMMA�pop�getattr�parent�prefixr ) �selfZnodeZresultsr r �child�memberZ name_node� member_nameZ remove_comma�p� r �H/opt/alt/python39/lib64/python3.9/lib2to3/fixes/fix_itertools_imports.py� transform sH
�
�z FixItertoolsImports.transformN)�__name__�
__module__� __qualname__Z
BM_compatible�localsZPATTERNr r r r r r s
�r N) �__doc__Zlib2to3r Zlib2to3.fixer_utilr r r ZBaseFixr r r r r � s