o
0i
@ sD d dl mZ d dlmZmZmZ d dl mZ G dd dejZdS ) )
fixer_base)NameCallconsuming_calls)patcompc sh e Zd ZdZdZ fddZdd Zdd Zd d
Zd d Z d
Z
e e
Z
dZe eZdd Z ZS ) FixXrangeTz
power<
(name='range'|name='xrange') trailer< '(' args=any ')' >
rest=any* >
c s t t| || t | _d S N)superr
start_treesettransformed_xrangesselfZtreefilename __class__ ?/opt/alt/python310/lib64/python3.10/lib2to3/fixes/fix_xrange.pyr
s zFixXrange.start_treec C s
d | _ d S r )r r
r r r finish_tree s
zFixXrange.finish_treec C s@ |d }|j dkr| ||S |j dkr| ||S tt|)NnameZxrangerange)valuetransform_xrangetransform_range
ValueErrorreprr noderesultsr r r r transform s
zFixXrange.transformc C s0 |d }| td|jd | jt| d S )Nr r prefix)replacer r! r addidr r r r r $ s zFixXrange.transform_xrangec C sj t || jvr1| |s3ttd|d g}ttd|g|jd}|d D ]}|| q'|S d S d S )Nr argslistr rest)r$ r in_special_contextr r Zcloner! Z append_child)r r r Z
range_callZ list_callnr r r r * s zFixXrange.transform_rangez3power< func=NAME trailer< '(' node=any ')' > any* >zfor_stmt< 'for' any 'in' node=any ':' any* >
| comp_for< 'for' any 'in' node=any any* >
| comparison< any 'in' node=any any*>
c C sf |j d u rdS i }|j j d ur%| j|j j |r%|d |u r%|d jtv S | j|j |o2|d |u S )NFr func)parentp1matchr r p2)r r r r r r r( ? s
z FixXrange.in_special_context)__name__
__module__ __qualname__Z
BM_compatibleZPATTERNr
r r r r ZP1r Zcompile_patternr, ZP2r. r(
__classcell__r r r r r s
r N) r Z
fixer_utilr r r r ZBaseFixr r r r r s