custom_fixers/fix_alt_unicode.py

   1 from lib2to3 import fixer_base
   2 from lib2to3.fixer_util import Name, BlankLine
   3
   4
   5 class FixAltUnicode(fixer_base.BaseFix):
   6     PATTERN = """
   7     func=funcdef< 'def' name='__unicode__'
   8                   parameters< '(' NAME ')' > any+ >
   9     """
  10
  11     def transform(self, node, results):
  12         name = results['name']
  13         name.replace(Name('__str__', prefix=name.prefix))