File: //usr/local/lib/python3.9/site-packages/libfuturize/fixes/__pycache__/fix_next_call.cpython-39.pyc
a
��hV � @ sn d Z ddlmZ ddlmZ ddlmZ ddlm Z m
Z
m Z dZ G dd� dej
�Zd d
� Zd d � Zd
d� ZdS )a
Based on fix_next.py by Collin Winter.
Replaces it.next() -> next(it), per PEP 3114.
Unlike fix_next.py, this fixer doesn't replace the name of a next method with __next__,
which would break Python 2 compatibility without further help from fixers in
stage 2.
� )�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 )� FixNextCallTz�
power< base=any+ trailer< '.' attr='next' > trailer< '(' ')' > >
|
power< head=any+ trailer< '.' attr='next' > not trailer< '(' ')' > >
|
global=global_stmt< 'global' any* 'next' any* >
�prec s> t t| ��||� td|�}|r4| �|t� d| _nd| _d S )N�nextTF)�superr �
start_treer �warning� bind_warning�
shadowed_next)�self�tree�filename�n�� __class__� �I/usr/local/lib/python3.9/site-packages/libfuturize/fixes/fix_next_call.pyr s
zFixNextCall.start_treec C s� |sJ �|� d�}|� d�}|� d�}|rf| jr2q�dd� |D �}d|d _|�ttd|jd �|�� nb|rln\|r�t|�r�|d
}d�d d� |D ���� d kr�| � |t
� d S nd
|v r�| � |t
� d| _d S )N�base�attr�namec S s g | ] }|� � �qS r )�clone��.0r r r r �
<listcomp>7 � z)FixNextCall.transform.<locals>.<listcomp>� r r
)�prefix�headc S s g | ] }t |��qS r )�strr r r r r E r � __builtin__�globalT) �getr r! �replacer r �is_assign_target�join�stripr
r )r �node�resultsr r r r" r r r � transform* s*
zFixNextCall.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( R s
r( c C s4 | j tjkr| S | j tjks&| jd u r*d S t| j�S �N)r5 �symsZ expr_stmtZ simple_stmt�parentr3 �r+ r r r r3 ^ s
r3 c s$ | � kr dS t � fdd�| jD ��S )NTc 3 s | ]}t |� �V qd S r9 )r7 )r �cr< r r � <genexpr>h r z is_subtree.<locals>.<genexpr>)�anyr4 )�rootr+ r r< r r7 e s r7 N)�__doc__Z
lib2to3.pgen2r Zlib2to3.pygramr r: Zlib2to3r Zlib2to3.fixer_utilr r r r ZBaseFixr r( r3 r7 r r r r �<module> s >