B �A�[:�@s~dZddlmZddlmZddlmZmZmZm Z m Z mZmZm Z mZmZddlmZmZmZdd�ZGdd �d ej�Zd S)zD Fixer for: (a,)* *b (,c)* [,] = s for (a,)* *b (,c)* [,] in d: ... �)� fixer_base)�count) �Assign�Comma�Call�Newline�Name�Number�token�syms�Node�Leaf)�indentation�suitify� commatizecCs�g}t|�}t|�}|dkr�ttjt|�ttjttjd�ttj ttj d�t|�g�ttjd�g�g�}|� |�|� ttjddd��ttjttjddd�t|�ttjttjd�ttj |dkr�t|�ntdd �ttj d�|dkr�ttjttjd �t|�g�ntdd �g�ttjd�ttjd�g�g�}|� |�|dk�r�|� ttjddd��ttjt|dd�ttjttjd�ttj ttjttjd �t|�g�ttj d�g�ttjd�g�g�} |� | �ttj|�} | S)z� Accepts num_pre and num_post, which are counts of values before and after the starg (not including the starg) Returns a source fit for Assign() from fixer_util r�[�:�]�+� )�prefix��-)�unicoderrZpowerr�trailerr r �LSQBZ subscript�COLONr �RSQB�append�PLUSZfactor�MINUSZ arith_expr)Znum_preZnum_post�LISTNAME�ITERNAME�children�pre�postZpre_partZ main_partZ post_part�source�r(�R/opt/alt/python37/lib/python3.7/site-packages/libpasteurize/fixes/fix_unpacking.py�assignment_sourcesJ � b r*c@s(eZdZdZdd�Zdd�Zdd�ZdS) �FixUnpackinga9 expl=expr_stmt< testlist_star_expr< pre=(any ',')* star_expr< '*' name=NAME > post=(',' any)* [','] > '=' source=any > | impl=for_stmt< 'for' lst=exprlist< pre=(any ',')* star_expr< '*' name=NAME > post=(',' any)* [','] > 'in' it=any ':' suite=any>c s��fdd�dD�\}}}}dd�|D�}d|_dd�|D�}dd�t||��g|�D�}|�t��d |_tt|j�ttd �|��g��}t|t t |�t |�|j|j��} || fS)Nc3s|]}��|�VqdS)N)�get)�.0�n)�resultsr(r)� 3sz4FixUnpacking.fix_explicit_context..)r%�namer&r'cSs g|]}|jtjkr|��qSr()�typer �NAME�clone)r-r.r(r(r)� 4sz5FixUnpacking.fix_explicit_context..rcSs g|]}|jtjkr|��qSr()r2r r3r4)r-r.r(r(r)r56scSsg|]}|��qSr()r4)r-r.r(r(r)r57sr�list)rrr4rrrrr"rr*�lenr#) �self�noder/r%r1r&r'�target� setup_line� power_liner()r/r)�fix_explicit_context2s z!FixUnpacking.fix_explicit_contextcs��fdd�dD�\}}}}dd�|D�}d|_dd�|D�}dd�t||��g|�D�}|�t��|��}d |_tt|j�ttd �t|j �g��} t|t t|�t|�|j|j ��} | | fS)z_ Only example of the implicit context is a for loop, so only fix that. c3s|]}��|�VqdS)N)r,)r-r.)r/r(r)r0Esz4FixUnpacking.fix_implicit_context..)r%r1r&�itcSs g|]}|jtjkr|��qSr()r2r r3r4)r-r.r(r(r)r5Fsz5FixUnpacking.fix_implicit_context..rcSs g|]}|jtjkr|��qSr()r2r r3r4)r-r.r(r(r)r5HscSsg|]}|��qSr()r4)r-r.r(r(r)r5Isrr6)rrr4rrrrr"rr#r*r7)r8r9r/r%r1r&r>r:r'r;r<r()r/r)�fix_implicit_context@s " z!FixUnpacking.fix_implicit_contextc Cs@|�d�|_|�d�|_|�d�|�d�}}|dk r�|�||�\}}|j|_t|j�|_|�t ��|j}|� �}|�||�|�||�n�|dk �r<|�||�\}}t |�dd�|jD�d} d |_| jd j|_t| jd�| jd_| �dt ��| �d|�| �dt ��| �d|�|�d��t|jd d��dS)a� a,b,c,d,e,f,*g,h,i = range(100) changes to _3to2list = list(range(100)) a,b,c,d,e,f,g,h,i, = _3to2list[:6] + [_3to2list[6:-2]] + _3to2list[-2:] and for a,b,*c,d,e in iter_of_iters: do_stuff changes to for _3to2iter in iter_of_iters: _3to2list = list(_3to2iter) a,b,c,d,e, = _3to2list[:2] + [_3to2list[2:-2]] + _3to2list[-2:] do_stuff Z _3to2listZ _3to2iter�expl�implNcSsg|]}|jtjkr|�qSr()r2r�suite)r-�kr(r(r)r5psz*FixUnpacking.transform..rrr��lstr)r)�new_namer"r#r,r=rr�parentZappend_childr�removeZinsert_childr?rr$�value�replacer) r8r9r/r@rAr;r<rG�irBr(r(r)� transformSs0 zFixUnpacking.transformN)�__name__� __module__�__qualname__ZPATTERNr=r?rLr(r(r(r)r+&s r+N)�__doc__Zlib2to3r� itertoolsrZlib2to3.fixer_utilrrrrrr r rrr Zlibfuturize.fixer_utilrrrr*ZBaseFixr+r(r(r(r)�s0