o
0i
@ sd d dl mZ d dlmZ d dl mZ d dlmZmZmZm Z m
Z
m Z dd Z G dd dej
Zd S )
)pytree)token)
fixer_base)AssignAttrNameis_tupleis_listsymsc c sF t | D ]\}}|jtjkr |jd jdkr || |d fV qd S )N exceptr ) enumeratetyper
except_clausechildrenvalue)Znodesin r ?/opt/alt/python310/lib64/python3.10/lib2to3/fixes/fix_except.py find_excepts s r c @ s e Zd ZdZdZdd ZdS ) FixExceptTa1
try_stmt< 'try' ':' (simple_stmt | suite)
cleanup=(except_clause ':' (simple_stmt | suite))+
tail=(['except' ':' (simple_stmt | suite)]
['else' ':' (simple_stmt | suite)]
['finally' ':' (simple_stmt | suite)]) >
c C sr | j }dd |d D }dd |d D }t|D ]\}}t|jdkr|jdd \}} }
| tdd d
|
jtjkrt| d d
} |
} d | _ |
| |
} |j}
t |
D ] \}}t
|tjrh nq\t|
sqt|
r|t| t| td }nt| | }t|
d | D ]}|d
| q||| q|
j d krd |
_ qdd |jd d D | | }t|j|S )Nc S g | ]}| qS r clone).0r r r r
2 z'FixExcept.transform..tailc S r r r )r Zchr r r r 4 r Zcleanup as )prefix argsr c S r r r )r cr r r r \ r )r
r lenr replacer r r NAMEnew_namer r# r
isinstancer ZNoder r r r reversedZ insert_child)selfZnodeZresultsr
r Z try_cleanupr Ze_suiteEZcommaNZnew_NtargetZ suite_stmtsr ZstmtZassignchildr r r r transform/ s:
zFixExcept.transformN)__name__
__module__ __qualname__Z
BM_compatibleZPATTERNr3 r r r r r $ s r N)r$ r Zpgen2r r Z
fixer_utilr r r r r r
r ZBaseFixr r r r r s