B
_ @ 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 sH xBt | D ]6\}}|jtjkr
|jd jdkr
|| |d fV q
W d S )N exceptr ) enumeratetyper
except_clausechildrenvalue)Znodesin r =/opt/alt/python37/lib64/python3.7/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 s | j }dd |d D }dd |d D }x*t|D ] \}}t|jdkr6|jdd \}} }
| tdd d
|
jtjkrDt| d d
} |
} d | _ |
| |
} |j}
x"t |
D ]\}}t
|tjrP qW t|
st|
rt| t| td }n
t| | }x&t|
d | D ]}|d
| q W ||| q6|
j d kr6d |
_ q6W dd |jd d D | | }t|j|S )Nc S s g | ] }| qS r )clone).0r r r r
2 s z'FixExcept.transform..tailc S s g | ] }| qS r )r )r Zchr r r r 4 s Zcleanup as )prefix argsr c S s g | ] }| qS r )r )r cr r r r \ s )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 ZstmtZassignZchildr r r r transform/ s6
zFixExcept.transformN)__name__
__module__ __qualname__Z
BM_compatibleZPATTERNr0 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