U
��,af � @ sj d dl mZ d dlmZ d dlmZ d dlmZm Z m
Z
dZ G dd� dej �Z
dd � Zd
d � Zd d
� ZdS )� )�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/python38/lib64/python3.8/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 �
<listcomp>9 s z%FixNext.transform.<locals>.<listcomp>� � r �headc S s g | ] }t |��qS r )�strr 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 |jD ]&}|jtjkr0 dS t|| �r dS qdS )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)r1 �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 r4 )r3 )r �cr7 r r � <genexpr>g s z is_subtree.<locals>.<genexpr>)�anyr0 )�rootr( r r7 r r3 d s r3 N)Zpgen2r Zpygramr r5 r r Z
fixer_utilr r r r
ZBaseFixr r% r/ r3 r r r r �<module> s @