Yf& @ sO d Z d d l m Z d d l m Z m Z m Z Gd d d e j Z d S)zA Fixer for imports of itertools.(imap|ifilter|izip|ifilterfalse) )
fixer_base) BlankLinesymstokenc @ s/ e Z d Z d Z d e Z d d Z d S)FixItertoolsImportsTzT
import_from< 'from' 'itertools' 'import' imports=any >
c C s | d } | j t j k s& | j r2 | g } n | j } x | d d d D] } | j t j k ry | j } | } n; | j t j k r d S| j t j k s t | j d } | j } | d k r d | _ | j qO | d k rO | j
| d d
k r d n d | _ qO W| j d d p1| g } d
} x= | D]5 } | rl| j t j k rl| j qA| d
N} qAWx0 | r| d j t j k r| j j q}W| j pt
| d d s| j d k r| j }
t } |
| _ | Sd S)Nimports r imapizipifilter ifilterfalse izip_longest f filterfalse zip_longestTvalue)r r
r )r r
)typer Zimport_as_namechildrenr NAMEr STARAssertionErrorremoveZchangedCOMMApopgetattrparentprefixr ) selfZnodeZresultsr r ZchildZmemberZ name_nodeZ member_nameZ remove_commap r! H/opt/alt/python35/lib64/python3.5/lib2to3/fixes/fix_itertools_imports.py transform sD
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" s