26 lines
1.1 KiB
Python
26 lines
1.1 KiB
Python
u"""Fixer for 'raise E(V).with_traceback(T)' -> 'raise E, V, T'"""
|
|
|
|
from lib2to3 import fixer_base
|
|
from lib2to3.fixer_util import Comma, Node, Leaf, token, syms
|
|
|
|
class FixRaise(fixer_base.BaseFix):
|
|
|
|
PATTERN = u"""
|
|
raise_stmt< 'raise' (power< name=any [trailer< '(' val=any* ')' >]
|
|
[trailer< '.' 'with_traceback' > trailer< '(' trc=any ')' >] > | any) ['from' chain=any] >"""
|
|
|
|
def transform(self, node, results):
|
|
name, val, trc = (results.get(u"name"), results.get(u"val"), results.get(u"trc"))
|
|
chain = results.get(u"chain")
|
|
if chain is not None:
|
|
self.warning(node, u"explicit exception chaining is not supported in Python 2")
|
|
chain.prev_sibling.remove()
|
|
chain.remove()
|
|
if trc is not None:
|
|
val = val[0] if val else Leaf(token.NAME, u"None")
|
|
val.prefix = trc.prefix = u" "
|
|
kids = [Leaf(token.NAME, u"raise"), name.clone(), Comma(),
|
|
val.clone(), Comma(), trc.clone()]
|
|
raise_stmt = Node(syms.raise_stmt, kids)
|
|
node.replace(raise_stmt)
|