��]� �@sYddlmZddlmZmZmZddlmZGdd�dej�ZdS)�)� fixer_base)�Name�Call�consuming_calls)�patcompcs�eZdZdZdZ�fdd�Zdd�Zdd�Zd d �Zdd�Z d Z eje �Z dZeje�Zdd�Z�S)� FixXrangeTz� power< (name='range'|name='xrange') trailer< '(' args=any ')' > rest=any* > cs)tt|�j||�t�|_dS)N)�superr� start_tree�set�transformed_xranges)�self�tree�filename)� __class__��//opt/alt/python35/lib64/python3.5/fix_xrange.pyr szFixXrange.start_treecCs d|_dS)N)r)rr rrrr�finish_treeszFixXrange.finish_treecCs^|d}|jdkr)|j||�S|jdkrH|j||�Stt|��dS)N�name�xrange�range)�value�transform_xrange�transform_range� ValueError�repr)r�node�resultsrrrr� transforms zFixXrange.transformcCs@|d}|jtdd|j��|jjt|��dS)Nrr�prefix)�replacerrr�add�id)rrrrrrrr$s zFixXrange.transform_xrangecCs�t|�|jkr�|j|�r�ttd�|dj�g�}ttd�|gd|j�}x|dD]}|j|�qsW|SdS)Nr�args�listr�rest)r!r�in_special_contextrrZclonerZappend_child)rrrZ 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*> cCs�|jdkrdSi}|jjdk rg|jj|jj|�rg|d|krg|djtkS|jj|j|�o�|d|kS)NFr�func)�parent�p1�matchrr�p2)rrrrrrr%?szFixXrange.in_special_context)�__name__� __module__�__qualname__Z BM_compatibleZPATTERNr rrrrZP1rZcompile_patternr)ZP2r+r%rr)rrrs rN) �rZ fixer_utilrrrrZBaseFixrrrrr�s