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 d } | j } | d k r d | _ | j qO | d k rO | j | d d
k r d n d | _ qO W| j d d p| g } d
} x= | D]5 } | rT| j t j
k rT| j q)| d
N} q)Wx0 | r| d j t j
k r| j j qeW| 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 STARremoveZchangedCOMMApopgetattrparentprefixr ) selfZnodeZresultsr r ZchildZmemberZ name_nodeZ member_nameZ remove_commap r H/opt/alt/python35/lib64/python3.5/lib2to3/fixes/fix_itertools_imports.py transform sB
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