3
\f � @ sn d Z ddlmZ ddlmZ ddlmZ ddlm Z m
Z
m Z dZ G dd� dej
�Zd d
� Zd d � Zd
d� ZdS )z.Fixer for it.next() -> next(it), per PEP 3114.� )�token)�python_symbols)�
fixer_base)�Name�Call� find_bindingz;Calls to builtin next() possibly shadowed by global bindingc s0 e Zd ZdZdZdZ� fdd�Zdd� Z� ZS )�FixNextTa�
power< base=any+ trailer< '.' attr='next' > trailer< '(' ')' > >
|
power< head=any+ trailer< '.' attr='next' > not trailer< '(' ')' > >
|
classdef< 'class' any+ ':'
suite< any*
funcdef< 'def'
name='next'
parameters< '(' NAME ')' > any+ >
any* > >
|
global=global_stmt< 'global' any* 'next' any* >
Zprec s> t t| �j||� td|�}|r4| j|t� d| _nd| _d S )N�nextTF)�superr �
start_treer �warning� bind_warning�
shadowed_next)�selfZtree�filename�n)� __class__� �./usr/lib64/python3.6/lib2to3/fixes/fix_next.pyr $ s
zFixNext.start_treec C s |st �|jd�}|jd�}|jd�}|rz| jrF|jtd|jd�� n2dd� |D �}d|d _|jttd
|jd�|�� n�|r�td|jd�}|j|� nl|r�t|�r�|d }djd d� |D ��j � d
kr�| j
|t � d S |jtd�� n d|k�r| j
|t � d| _d S )N�base�attr�name�__next__)�prefixc S s g | ] }|j � �qS r )Zclone)�.0r r r r �
<listcomp>9 s z%FixNext.transform.<locals>.<listcomp>� � r �headc S s g | ] }t |��qS r )�str)r r r r r r E s Z __builtin__�globalT) �AssertionError�getr �replacer r r �is_assign_target�join�stripr r
)r �nodeZresultsr r r r r r r r � transform. s.
zFixNext.transform) �__name__�
__module__� __qualname__Z
BM_compatibleZPATTERN�orderr r( �
__classcell__r r )r r r s
r c C sF t | �}|d krdS x,|jD ]"}|jtjkr0dS t|| �r dS q W dS )NFT)� find_assign�children�typer �EQUAL�
is_subtree)r' ZassignZchildr r r r$ Q s
r$ c C s4 | j tjkr| S | j tjks&| jd kr*d S t| j�S )N)r0 �symsZ expr_stmtZ simple_stmt�parentr. )r' r r r r. ] s
r. c s$ | � kr dS t � fdd�| jD ��S )NTc 3 s | ]}t |� �V qd S )N)r2 )r �c)r' r r � <genexpr>g s z is_subtree.<locals>.<genexpr>)�anyr/ )�rootr' r )r' r r2 d s r2 N)�__doc__Zpgen2r Zpygramr r3 r r Z
fixer_utilr r r r
ZBaseFixr r$ r. r2 r r r r �<module> s @