a
XC?hf � @ 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/python39/lib64/python3.9/lib2to3/fixes/fix_next.pyr $ s
zFixNext.start_treec C s |sJ �|� d�}|� d�}|� d�}|rz| jrF|�td|jd�� n2dd� |D �}d|d _|�ttd
|jd�|�� n�|r�td|jd�}|�|� nl|r�t|�r�|d }d�d d� |D ���� d
kr�| � |t
� d S |�td�� n d|v �r| � |t
� d| _d S )N�base�attr�name�__next__)�prefixc S s g | ] }|� � �qS r )Zclone��.0r r r r �
9 � z%FixNext.transform..� � r �headc S s g | ] }t |��qS r )�strr r r r r E r 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 u rdS |jD ]&}|jtjkr0 dS t|| �r dS qdS )NFT)� find_assign�children�typer �EQUAL�
is_subtree)r) Zassign�childr r r r& Q s
r& c C s4 | j tjkr| S | j tjks&| jd u r*d S t| j�S �N)r2 �symsZ expr_stmtZ simple_stmt�parentr0 �r) r r r r0 ] s
r0 c s$ | � kr dS t � fdd�| jD ��S )NTc 3 s | ]}t |� �V qd S r6 )r4 )r �cr9 r r � g r z is_subtree..)�anyr1 )�rootr) r r9 r r4 d s r4 N)�__doc__Zpgen2r Zpygramr r7 r r Z
fixer_utilr r r r
ZBaseFixr r& r0 r4 r r r r � s @