OLD | NEW |
(Empty) | |
| 1 u"""Fixer for 'raise E(V).with_traceback(T)' -> 'raise E, V, T'""" |
| 2 |
| 3 from lib2to3 import fixer_base |
| 4 from lib2to3.fixer_util import Comma, Node, Leaf, token, syms |
| 5 |
| 6 class FixRaise(fixer_base.BaseFix): |
| 7 |
| 8 PATTERN = u""" |
| 9 raise_stmt< 'raise' (power< name=any [trailer< '(' val=any* ')' >] |
| 10 [trailer< '.' 'with_traceback' > trailer< '(' trc=any ')' >] > | any) ['
from' chain=any] >""" |
| 11 |
| 12 def transform(self, node, results): |
| 13 name, val, trc = (results.get(u"name"), results.get(u"val"), results.get
(u"trc")) |
| 14 chain = results.get(u"chain") |
| 15 if chain is not None: |
| 16 self.warning(node, u"explicit exception chaining is not supported in
Python 2") |
| 17 chain.prev_sibling.remove() |
| 18 chain.remove() |
| 19 if trc is not None: |
| 20 val = val[0] if val else Leaf(token.NAME, u"None") |
| 21 val.prefix = trc.prefix = u" " |
| 22 kids = [Leaf(token.NAME, u"raise"), name.clone(), Comma(), |
| 23 val.clone(), Comma(), trc.clone()] |
| 24 raise_stmt = Node(syms.raise_stmt, kids) |
| 25 node.replace(raise_stmt) |
OLD | NEW |