B
� ff � @ 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| ��||� td|�}|r4| �|t� d| _nd| _d S )N�nextTF)�superr �
start_treer �warning� bind_warning�
shadowed_next)�selfZtree�filename�n)� __class__� �;/opt/alt/python37/lib64/python3.7/lib2to3/fixes/fix_next.pyr $ s
zFixNext.start_treec C s� |� d�}|� d�}|� d�}|rr| jr>|�td|jd�� q�dd� |D �}d|d _|�ttd
|jd�|�� n�|r�td|jd�}|�|� nj|r�t|�r�|d }d�d d� |D ���� d
kr�| � |t
� d S |�td�� nd|kr�| � |t
� d| _d S )N�base�attr�name�__next__)�prefixc S s g | ] }|� � �qS r )Zclone)�.0r r r r �
9 s z%FixNext.transform..� � r �headc S s g | ] }t |��qS r )�str)r r r r r r E s Z __builtin__�globalT) �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)r/ �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)r1 )r �c)r& r r � g s z is_subtree..)�anyr. )�rootr& r )r& r r1 d s r1 N)�__doc__Zpgen2r Zpygramr r2 r r Z
fixer_utilr r r r
ZBaseFixr r# r- r1 r r r r � s @