3
\f @ 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)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__ ./usr/lib64/python3.6/lib2to3/fixes/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 nodeZresultsr 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)r/ 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)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 @