관리-도구
편집 파일: fix_unpacking.cpython-37.pyc
B �A�[: � @ 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�}|� |� |� ttjddd�� ttjttjddd�t|�ttjttjd�ttj | dkr�t |�ntdd �ttj d�|dkr�ttjttjd �t |�g�ntdd �g�ttj d�ttj d�g�g�}|� |� |dk�r�|� ttjddd�� ttjt|dd�ttjttjd�ttj ttjttjd �t |�g�ttj d�g�ttj d�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� � �-)�unicoder r Zpowerr �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_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|_ dd� |D �}dd� t||�� g | �D �}|�t� � d |_ tt| j�ttd �|�� g��}t|t t |�t |�| j| j ��} || fS ) Nc 3 s | ]}� � |�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|�� �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|�� �qS r( )r2 r r3 r4 )r- r. r( r( r) r5 6 s c S s g | ] }|� � �qS r( )r4 )r- r. r( r( r) r5 7 s r �list) r r r4 r r r r r" r r* �lenr# ) �self�noder/ r% r1 r& r' �target� setup_line� power_liner( )r/ r) �fix_explicit_context2 s z!FixUnpacking.fix_explicit_contextc s� � 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. c 3 s | ]}� � |�V qd S )N)r, )r- r. )r/ r( r) r0 E s z4FixUnpacking.fix_implicit_context.<locals>.<genexpr>)r% r1 r& �itc S s g | ]}|j tjkr|�� �qS r( )r2 r r3 r4 )r- r. r( r( r) r5 F s z5FixUnpacking.fix_implicit_context.<locals>.<listcomp>r c S s g | ]}|j tjkr|�� �qS r( )r2 r r3 r4 )r- r. r( r( r) r5 H s c S s g | ] }|� � �qS r( )r4 )r- r. r( r( r) r5 I s r r6 ) r r r4 r r r r r" r r# r* r7 ) r8 r9 r/ r% r1 r& r> r: r' r; r< r( )r/ r) �fix_implicit_context@ s " z!FixUnpacking.fix_implicit_contextc C s@ | � d�| _| � d�| _|�d�|�d� }}|dk r�| �||�\}}|j|_t|j�|_|�t � � |j}|� � }|� ||� |� ||� n�|dk �r<| � ||�\}}t |� dd� |jD �d } d |_| jd j|_t| jd �| jd _| � d t � � | � d |� | � d t � � | � 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�implNc S s g | ]}|j tjkr|�qS r( )r2 r �suite)r- �kr( r( r) r5 p s z*FixUnpacking.transform.<locals>.<listcomp>r r r � �lstr )r )�new_namer" r# r, r= r r �parentZ append_childr �removeZ insert_childr? r r$ �value�replacer ) r8 r9 r/ r@ rA r; r< rG �irB r( r( r) � transformS s0 zFixUnpacking.transformN)�__name__� __module__� __qualname__ZPATTERNr= r? rL 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