Rename __unicode__ to __str__

3 files changed, 13 insertions, 0 deletions

diff --git a/custom_fixers/__init__.py b/custom_fixers/__init__.py
new file mode 100644
index 00000000..e69de29b
--- /dev/null
+++ b/custom_fixers/__init__.py
diff --git a/custom_fixers/fix_alt_unicode.py b/custom_fixers/fix_alt_unicode.py
new file mode 100644
index 00000000..55175e90
--- /dev/null
+++ b/custom_fixers/fix_alt_unicode.py
@@ -0,0 +1,12 @@
+from lib2to3.fixer_base import BaseFix
+from lib2to3.fixer_util import Name
+
+class FixAltUnicode(BaseFix):
+    PATTERN = """
+    func=funcdef< 'def' name='__unicode__'
+                  parameters< '(' NAME ')' > any+ >
+    """
+
+    def transform(self, node, results):
+        name = results['name']
+        name.replace(Name('__str__', prefix=name.prefix))
diff --git a/setup.py b/setup.py
index 8d06f569..fe4066b8 100644
--- a/setup.py
+++ b/setup.py
@@ -199,4 +199,5 @@ setup(
     install_requires=requires,
     cmdclass=cmdclass,
     use_2to3=True,
+    use_2to3_fixers=['custom_fixers'],
 )