a
i& @ s: d Z ddlmZ ddlmZmZmZ G dd dejZdS )zA Fixer for imports of itertools.(imap|ifilter|izip|ifilterfalse) )
fixer_base) BlankLinesymstokenc @ 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 rd |_| q6|dv r6| |d dkrdnd |_q6|jd d p|g}d
} |D ]*}| r|j tj
kr| q| d
N } q|r4|d j tj
kr4| q |jsJt |d d rV|j
d u rl|j}
t }|
|_|S d S )
Nimports r )imapizipifilter) ifilterfalse izip_longest f filterfalse zip_longestTvalue)typer Zimport_as_namechildrenr NAMEr STARremovechangedCOMMApopgetattrparentprefixr ) selfZnoderesultsr r childmemberZ name_node member_nameZ remove_commap r% ;/usr/lib64/python3.9/lib2to3/fixes/fix_itertools_imports.py transform sH
z FixItertoolsImports.transformN)__name__
__module__ __qualname__Z
BM_compatiblelocalsZPATTERNr' r% r% r% r& r s
r N) __doc__Zlib2to3r Zlib2to3.fixer_utilr r r ZBaseFixr r% r% r% r&