| 123456789101112131415161718192021222324252627282930313233 |
- u"""
- Fixer for "class Foo: ..." -> "class Foo(object): ..."
- """
- from lib2to3 import fixer_base
- from lib2to3.fixer_util import LParen, RParen, Name
- from libfuturize.fixer_util import touch_import_top
- def insert_object(node, idx):
- node.insert_child(idx, RParen())
- node.insert_child(idx, Name(u"object"))
- node.insert_child(idx, LParen())
- class FixNewstyle(fixer_base.BaseFix):
- # Match:
- # class Blah:
- # and:
- # class Blah():
- PATTERN = u"classdef< 'class' NAME ['(' ')'] colon=':' any >"
- def transform(self, node, results):
- colon = results[u"colon"]
- idx = node.children.index(colon)
- if (node.children[idx-2].value == '(' and
- node.children[idx-1].value == ')'):
- del node.children[idx-2:idx]
- idx -= 2
- insert_object(node, idx)
- touch_import_top(u'builtins', 'object', node)
|
