B � f� �@sFdZddlmZddlmZmZmZddlmZGdd�dej�Z dS)z/Fixer that changes xrange(...) into range(...).�)� fixer_base)�Name�Call�consuming_calls)�patcompcsheZdZdZdZ�fdd�Zdd�Zdd�Zd d �Zd d �Z d Z e � e �Z dZe � e�Zdd�Z�ZS)� FixXrangeTz� power< (name='range'|name='xrange') trailer< '(' args=any ')' > rest=any* > cstt|��||�t�|_dS)N)�superr� start_tree�set�transformed_xranges)�self�tree�filename)� __class__��=/opt/alt/python37/lib64/python3.7/lib2to3/fixes/fix_xrange.pyr szFixXrange.start_treecCs d|_dS)N)r )r r rrrr� finish_treeszFixXrange.finish_treecCsD|d}|jdkr|�||�S|jdkr4|�||�Stt|���dS)N�nameZxrange�range)�value�transform_xrange�transform_range� ValueError�repr)r �node�resultsrrrr� transforms     zFixXrange.transformcCs0|d}|�td|jd��|j�t|��dS)Nrr)�prefix)�replacerrr �add�id)r rrrrrrr$szFixXrange.transform_xrangecCsjt|�|jkrf|�|�sfttd�|d��g�}ttd�|g|jd�}x|dD]}|�|�qPW|SdS)Nr�args�list)r�rest)r r �in_special_contextrrZclonerZ append_child)r rrZ range_callZ list_call�nrrrr*s   zFixXrange.transform_rangez3power< func=NAME trailer< '(' node=any ')' > any* >z�for_stmt< 'for' any 'in' node=any ':' any* > | comp_for< 'for' any 'in' node=any any* > | comparison< any 'in' node=any any*> cCsf|jdkrdSi}|jjdk rJ|j�|jj|�rJ|d|krJ|djtkS|j�|j|�od|d|kS)NFr�func)�parent�p1�matchrr�p2)r rrrrrr$?s   zFixXrange.in_special_context)�__name__� __module__� __qualname__Z BM_compatibleZPATTERNr rrrrZP1rZcompile_patternr(ZP2r*r$� __classcell__rr)rrr s     rN) �__doc__�rZ fixer_utilrrrrZBaseFixrrrrr�s