U
e5d& � @ s6 d dl mZ d dlmZmZmZ G dd� dej�ZdS )� )�
fixer_base)� BlankLine�syms�tokenc @ s" e Zd ZdZde� Zdd� ZdS )�FixItertoolsImportsTzT
import_from< 'from' 'itertools' 'import' imports=any >
c C sZ |d }|j tjks|js"|g}n|j}|d d d� D ]|}|j tjkrR|j}|}n |j tjkrd d S |jd }|j}|dkr�d |_|�� q6|dkr6|� � |d dkr�dnd |_q6|jd d � p�|g}d
} |D ]&}| r�|j tj
kr�|�� q�| d
N } q�|�r |d j tj
k�r |� � �� q�|j�s4t |d d ��r@|j
d k�rV|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 Zchild�memberZ name_node� member_nameZ remove_comma�p� r �;/usr/lib64/python3.8/lib2to3/fixes/fix_itertools_imports.py� transform sF
�
�z FixItertoolsImports.transformN)�__name__�
__module__� __qualname__Z
BM_compatible�localsZPATTERNr r r r r r s
�r N)Zlib2to3r Zlib2to3.fixer_utilr r r ZBaseFixr r r r r �<module> s