ruạṛ
3 AL�i� � @ 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 }yt | �}t |�}W n$ tk r< t| �}t|�}Y nX | dkr�ttjt|�ttjtt j d�ttj tt j d�t |�g�tt jd�g�g�}|j|� |jtt jddd�� ttjtt j ddd�t|�ttjtt j d�ttj | dkr�t |�ntdd �tt j d�|dk�r(ttjtt jd �t |�g�ntdd �g�tt jd�tt jd�g�g�}|j|� |dk�r�|jtt jddd�� ttjt|dd�ttjtt j d�ttj ttjtt jd �t |�g�tt j d�g�tt jd�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� � �-)�unicode� NameError�strr r �powerr �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+ �H/root/tmp/pip-build-3x0y2zol/future/libpasteurize/fixes/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>7 s z4FixUnpacking.fix_explicit_context.<locals>.<genexpr>r( �namer) r* c S s g | ]}|j tjkr|j� �qS r+ )�typer �NAME�clone)r0 r1 r+ r+ r, � <listcomp>8 s z5FixUnpacking.fix_explicit_context.<locals>.<listcomp>r c S s g | ]}|j tjkr|j� �qS r+ )r5 r r6 r7 )r0 r1 r+ r+ r, r8 : s c S s g | ] }|j � �qS r+ )r7 )r0 r1 r+ r+ r, r8 ; s r �list)r( r4 r) r* ) r r r7 r" r r r r% r r- �lenr& ) �self�noder2 r( r4 r) r* �target� setup_line� power_liner+ )r2 r, �fix_explicit_context6 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/ )r0 r1 )r2 r+ r, r3 I s z4FixUnpacking.fix_implicit_context.<locals>.<genexpr>r( r4 r) �itc S s g | ]}|j tjkr|j� �qS r+ )r5 r r6 r7 )r0 r1 r+ r+ r, r8 J s z5FixUnpacking.fix_implicit_context.<locals>.<listcomp>r c S s g | ]}|j tjkr|j� �qS r+ )r5 r r6 r7 )r0 r1 r+ r+ r, r8 L s c S s g | ] }|j � �qS r+ )r7 )r0 r1 r+ r+ r, r8 M s r r9 )r( r4 r) rA ) r r r7 r" r r r r% r r& r- r: ) r; r< r2 r( r4 r) rA r= r* r> r? r+ )r2 r, �fix_implicit_contextD 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+ )r5 r �suite)r0 �kr+ r+ r, r8 t s z*FixUnpacking.transform.<locals>.<listcomp>r r r � �lstr )r )�new_namer% r& r/ r@ r r �parentZ append_childr �removeZ insert_childrB r r' �value�replacer ) r; r< r2 rC rD r> r? rJ �irE r+ r+ r, � transformW s0 zFixUnpacking.transformN)�__name__� __module__� __qualname__ZPATTERNr@ rB rO 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
cải xoăn