File: //lib/python3.6/site-packages/libpasteurize/fixes/__pycache__/fix_unpacking.cpython-36.opt-1.pyc
3
�c: � @ s~ d Z ddlmZ ddlmZ ddlmZmZmZm Z m
Z
m Z m Z m
Z
mZmZ ddlmZmZmZ dd� ZG dd � 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� commatizec C s� g }t | �}t |�}| dkr�ttjt|�ttjttjd�ttj ttj
d�t |�g�ttj d�g�g�}|j
|� |j
ttjddd�� ttjttjddd�t|�ttjttjd�ttj | dkr�t |�ntdd �ttj
d�|dkr�ttjttjd
�t |�g�ntdd �g�ttj d�ttj d�g�g�}|j
|� |dk�r�|j
ttjddd�� ttjt|dd�ttjttjd�ttj ttjttjd
�t |�g�ttj
d�g�ttj d�g�g�} |j
| � 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� � �-)Zunicoder r Zpowerr Ztrailerr
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& �#/usr/lib/python3.6/fix_unpacking.py�assignment_source s J
�
b
r( c @ s( e Zd ZdZdd� 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 |_ d
d� |D �}d d� t||j� g | �D �}|jt� � d |_ tt| j�ttd
�|j� g��}t|t t
|�t
|�| j| j ��} || fS )Nc 3 s | ]}� j |�V qd S )N)�get)�.0�n)�resultsr& r' � <genexpr>3 s z4FixUnpacking.fix_explicit_context.<locals>.<genexpr>r# �namer$ r% c S s g | ]}|j tjkr|j� �qS r& )�typer
�NAME�clone)r+ r, r& r& r' �
<listcomp>4 s z5FixUnpacking.fix_explicit_context.<locals>.<listcomp>r c S s g | ]}|j tjkr|j� �qS r& )r0 r
r1 r2 )r+ r, r& r& r' r3 6 s c S s g | ] }|j � �qS r& )r2 )r+ r, r& r& r' r3 7 s r �list)r# r/ r$ r% ) r r r2 r r r r r r r( �lenr! )
�self�noder- r# r/ r$ r% �target�
setup_line�
power_liner& )r- r' �fix_explicit_context2 s z!FixUnpacking.fix_explicit_contextc s� � fdd�dD �\}}}}dd� |D �}d |_ d
d� |D �}d d� t||j� g | �D �}|jt� � |j� }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.
c 3 s | ]}� j |�V qd S )N)r* )r+ r, )r- r& r' r. E s z4FixUnpacking.fix_implicit_context.<locals>.<genexpr>r# r/ r$ �itc S s g | ]}|j tjkr|j� �qS r& )r0 r
r1 r2 )r+ r, r& r& r' r3 F s z5FixUnpacking.fix_implicit_context.<locals>.<listcomp>r c S s g | ]}|j tjkr|j� �qS r& )r0 r
r1 r2 )r+ r, r& r& r' r3 H s c S s g | ] }|j � �qS r& )r2 )r+ r, r& r& r' r3 I s r r4 )r# r/ r$ r< ) r r r2 r r r r r r r! r( r5 ) r6 r7 r- r# r/ r$ r< r8 r% r9 r: r& )r- r' �fix_implicit_context@ s " z!FixUnpacking.fix_implicit_contextc
C s@ | j d�| _| j d�| _|jd�|jd� }}|dk r�| j||�\}}|j|_t|j�|_|jt � � |j}|j
� }|j ||� |j ||� n�|dk �r<| j ||�\}}t
|� dd� |jD �d } d |_| jd
j|_t| jd �| jd _| j d t � � | j d |� | j d t � � | j d |� |jd �jt| 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�implNc S s g | ]}|j tjkr|�qS r& )r0 r �suite)r+ �kr& r& r' r3 p s z*FixUnpacking.transform.<locals>.<listcomp>r r r � Zlstr )r )�new_namer r! r* r; r r �parentZ append_childr �removeZ insert_childr= r r" �value�replacer )
r6 r7 r- r> r? r9 r: rD �ir@ r& r& r' � transformS s0
zFixUnpacking.transformN)�__name__�
__module__� __qualname__ZPATTERNr; r= rI r& r& r& r' r) & s
r) N)�__doc__Zlib2to3r � itertoolsr Zlib2to3.fixer_utilr r r r r r r
r r r
Zlibfuturize.fixer_utilr r r r( ZBaseFixr) r& r& r& r' �<module> s 0