34 lines
888 B
Python
34 lines
888 B
Python
|
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)
|