U
,af @ 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| || td|}|r4| |t d| _nd| _d S )NnextTF)superr
start_treer warning bind_warning
shadowed_next)selfZtreefilenamen __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 qdd |D }d|d _|ttd
|jd| n|rtd|jd}|| nj|rt|r|d }dd d |D d
kr| |t
d S |td nd|kr| |t
d| _d S )Nbaseattrname__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_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 |jD ]&}|jtjkr0 dS t|| r dS qdS )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)r1 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 r4 )r3 )r cr7 r r <genexpr>g s z is_subtree.<locals>.<genexpr>)anyr0 )rootr( r r7 r r3 d s r3 N)__doc__Zpgen2r Zpygramr r5 r r Z
fixer_utilr r r r
ZBaseFixr r% r/ r3 r r r r <module> s @