Better fixer for unicode invokes that disables accidental attribute r…

…ewrites. This fixes #1
mitsuhiko · Mar 14, 2012 · 7b7ef03 · 7b7ef03
1 parent c1ff4ca
commit 7b7ef03
Showing 1 changed file with 10 additions and 5 deletions.
diff --git a/libmodernize/fixes/fix_unicode.py b/libmodernize/fixes/fix_unicode.py
@@ -8,14 +8,19 @@
 
 class FixUnicode(fixer_base.BaseFix):
     BM_compatible = True
-    PATTERN = "STRING | 'unicode'"
+    PATTERN = """
+        STRING |
+        power< name='unicode'
+            trailer< '(' [any] ')' >
+            any *
+        >
+    """
 
     def transform(self, node, results):
-        if node.type == token.NAME:
+        if 'name' in results:
             touch_import(None, u'six', node)
-            new = node.clone()
+            name = results['name']
-            new.value = u'six.text_type'
+            name.replace(Name(u'six.text_type', prefix=name.prefix))
-            return new
         elif node.type == token.STRING and _literal_re.match(node.value):
             touch_import(None, u'six', node)
             new = node.clone()