a
i
@ sb 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 sD t | D ]6\}}|jtjkr|jd jdkr|| |d fV qd S )N exceptr ) enumeratetyper
except_clausechildrenvalue)Znodesin r 0/usr/lib64/python3.9/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 sx | j }dd |d D }dd |d D }t|D ]\}}t|jdkr2|jdd \}} }
| tdd d
|
jtjkr8t| d d
} |
} d | _ |
| |
} |j}
t |
D ]\}}t
|tjr qqt|
st|
rt| t| td }n
t| | }t|
d | D ]}|d
| q||| q2|
j d kr2d |
_ q2dd |jd d D | | }t|j|S )Nc S s g | ] }| qS r clone).0r r r r
2 z'FixExcept.transform..tailc S s g | ] }| qS r r )r Zchr r r r 4 r cleanup as )prefix argsr c S s g | ] }| qS 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)selfZnoderesultsr
r Z try_cleanupr Ze_suiteEZcommaNZnew_NtargetZ suite_stmtsr ZstmtZassignchildr r r r transform/ s6
zFixExcept.transformN)__name__
__module__ __qualname__Z
BM_compatibleZPATTERNr4 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