| """Fixer for sys.exc_{type, value, traceback} |
| |
| sys.exc_type -> sys.exc_info()[0] |
| sys.exc_value -> sys.exc_info()[1] |
| sys.exc_traceback -> sys.exc_info()[2] |
| """ |
| |
| # By Jeff Balogh and Benjamin Peterson |
| |
| # Local imports |
| from .. import fixer_base |
| from ..fixer_util import Attr, Call, Name, Number, Subscript, Node, syms |
| |
| class FixSysExc(fixer_base.BaseFix): |
| # This order matches the ordering of sys.exc_info(). |
| exc_info = ["exc_type", "exc_value", "exc_traceback"] |
| PATTERN = """ |
| power< 'sys' trailer< dot='.' attribute=(%s) > > |
| """ % '|'.join("'%s'" % e for e in exc_info) |
| |
| def transform(self, node, results): |
| sys_attr = results["attribute"][0] |
| index = Number(self.exc_info.index(sys_attr.value)) |
| |
| call = Call(Name("exc_info"), prefix=sys_attr.get_prefix()) |
| attr = Attr(Name("sys"), call) |
| attr[1].children[0].set_prefix(results["dot"].get_prefix()) |
| attr.append(Subscript(index)) |
| return Node(syms.power, attr, prefix=node.get_prefix()) |