U
��,a& � @ 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 tjkstt�|jd }|j}|dkr�d |_|� � q6|dkr6|�
� |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|jd k�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�AssertionError�removeZchanged�COMMA�pop�getattr�parent�prefixr ) �selfZnodeZresultsr r Zchild�memberZ name_node� member_nameZ remove_comma�p� r �H/opt/alt/python38/lib64/python3.8/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 �<module> s