|
Server : Apache System : Linux iad1-shared-b8-43 6.6.49-grsec-jammy+ #10 SMP Thu Sep 12 23:23:08 UTC 2024 x86_64 User : dh_edsupp ( 6597262) PHP Version : 8.2.26 Disable Function : NONE Directory : /lib/python3/dist-packages/libpasteurize/fixes/__pycache__/ |
Upload File : |
o
,�]: � @ 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 |�}| dkrCttjt|�ttjttjd�ttj ttj
d�t |�g�ttj d�g�g�}|�
|� |�
ttjddd�� ttjttjddd�t|�ttjttjd�ttj | dkrct |�ntdd �ttj
d�|dkr~ttjttjd
�t |�g�ntdd �g�ttj d�ttj d�g�g�}|�
|� |dkr�|�
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 �powerr �trailerr
r
�LSQB� subscript�COLONr �RSQB�append�PLUS�factor�MINUS�
arith_expr) �num_pre�num_post�LISTNAME�ITERNAME�children�pre�post�pre_part� main_part� post_part�source� r2 �C/usr/lib/python3/dist-packages/libpasteurize/fixes/fix_unpacking.py�assignment_source s J
�
b
r4 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 � � | ]}� � |�V qd S �N��get��.0�n��resultsr2 r3 � <genexpr>3 � � z4FixUnpacking.fix_explicit_context.<locals>.<genexpr>)r, �namer- r1 c S � g | ] }|j tjkr|�� �qS r2 ��typer
�NAME�cloner: r2 r2 r3 �
<listcomp>4 � z5FixUnpacking.fix_explicit_context.<locals>.<listcomp>r c S rB r2 rC r: r2 r2 r3 rG 6 rH c S � g | ]}|� � �qS r2 �rF r: r2 r2 r3 rG 7 � r �list) r r rF r" r r r r) r r4 �lenr* )
�self�noder> r, rA r- r1 �target�
setup_line�
power_liner2 r= r3 �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 r6 r7 r8 r: r= r2 r3 r? E r@ z4FixUnpacking.fix_implicit_context.<locals>.<genexpr>)r, rA r- �itc S rB r2 rC r: r2 r2 r3 rG F rH z5FixUnpacking.fix_implicit_context.<locals>.<listcomp>r c S rB r2 rC r: r2 r2 r3 rG H rH c S rI r2 rJ r: r2 r2 r3 rG I rK r rL ) r r rF r" r r r r) r r* r4 rM ) rN rO r> r, rA r- rT rP r1 rQ rR r2 r= r3 �fix_implicit_context@ s " z!FixUnpacking.fix_implicit_contextc
C sD | � d�| _| � d�| _|�d�|�d�}}|durH| �||�\}}|j|_t|j�|_|�t � � |j}|�
� }|� ||� |� ||� dS |dur�| � ||�\}}t
|� dd� |jD �d } d |_| jd
j|_t| jd �| jd _| � d t � � | � d |� | � d t � � | � d |� |�d ��t| jd
d�� dS 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
� _3to2list� _3to2iter�expl�implNc S s g | ]
}|j tjkr|�qS r2 )rD r �suite)r; �kr2 r2 r3 rG p s z*FixUnpacking.transform.<locals>.<listcomp>r r r � �lstr r )�new_namer) r* r9 rS r r �parent� append_childr �remove� insert_childrU r r+ �value�replacer )
rN rO r> rX rY rQ rR r_ �irZ r2 r2 r3 � transformS s2 �zFixUnpacking.transformN)�__name__�
__module__� __qualname__�PATTERNrS rU rf r2 r2 r2 r3 r5 & s
r5 N)�__doc__�lib2to3r � itertoolsr �lib2to3.fixer_utilr r r r r r r
r r r
�libfuturize.fixer_utilr r r r4 �BaseFixr5 r2 r2 r2 r3 �<module> s 0