| Index: third_party/google-endpoints/libpasteurize/fixes/fix_newstyle.py
|
| diff --git a/third_party/google-endpoints/libpasteurize/fixes/fix_newstyle.py b/third_party/google-endpoints/libpasteurize/fixes/fix_newstyle.py
|
| new file mode 100644
|
| index 0000000000000000000000000000000000000000..cc6b3adcb0c1bd41cb7b891777f97420a3633c29
|
| --- /dev/null
|
| +++ b/third_party/google-endpoints/libpasteurize/fixes/fix_newstyle.py
|
| @@ -0,0 +1,33 @@
|
| +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)
|
|
|