3
2af @ 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)NameCall find_bindingz;Calls to builtin next() possibly shadowed by global bindingc s0 e Zd ZdZdZdZ fddZdd 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 )NnextTF)superr
start_treer warning bind_warning
shadowed_next)selfZtreefilenamen) __class__ -/opt/alt/python36/lib64/python3.6/fix_next.pyr $ s
zFixNext.start_treec C s |j d}|j d}|j d}|rr| jr>|jtd|jd qdd |D }d|d _|jttd
|jd| n|rtd|jd}|j| nj|rt|r|d }djd d |D j d
kr| j |t
d S |jtd nd|kr| j |t
d| _d S )Nbaseattrname__next__)prefixc S s g | ] }|j 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_targetjoinstripr r
)r noderesultsr r r r r r r r transform. s,
zFixNext.transform) __name__
__module__ __qualname__Z
BM_compatibleZPATTERNorderr 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_assignchildrentyper EQUAL
is_subtree)r& ZassignZchildr r r r# Q s
r# c C s4 | j tjkr| S | j tjks&| jd kr*d S t| jS )N)r0 symsZ expr_stmtZ simple_stmtparentr. )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 g s z is_subtree..)anyr/ )rootr& r )r& r r2 d s r2 N)Zpgen2r Zpygramr r3 r r Z
fixer_utilr r r r
ZBaseFixr r# r. r2 r r r r s @