]& @ sI d d l m Z d d l m Z m Z m Z Gd d d e j Z d S) )
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 ) selfZnoderesultsr r ZchildmemberZ name_node member_nameZ remove_commap r# :/opt/alt/python35/lib64/python3.5/fix_itertools_imports.py transform sB
z FixItertoolsImports.transformN)__name__
__module__ __qualname__Z
BM_compatiblelocalsZPATTERNr% r# r# r# r$ r s
r N)Zlib2to3r Zlib2to3.fixer_utilr r r ZBaseFixr r# r# r# r$ s