From: Benjamin Peterson <benjamin@python.org>
Date: Sat, 19 Jul 2008 14:19:28 +0000 (+0000)
Subject: Merged revisions 65141 via svnmerge from
X-Git-Tag: v3.0b3~260
X-Git-Url: http://git.ipfire.org/gitweb.cgi?a=commitdiff_plain;h=0f6de936f81eeb292542aa91a7dfd3fbaf887066;p=thirdparty%2FPython%2Fcpython.git

Merged revisions 65141 via svnmerge from
svn+ssh://pythondev@svn.python.org/python/trunk

................
  r65141 | benjamin.peterson | 2008-07-19 09:14:06 -0500 (Sat, 19 Jul 2008) | 9 lines

  Merged revisions 65137 via svnmerge from
  svn+ssh://pythondev@svn.python.org/sandbox/trunk/2to3/lib2to3

  ........
    r65137 | georg.brandl | 2008-07-19 08:32:57 -0500 (Sat, 19 Jul 2008) | 2 lines

    #3334: correctly set prefix of imports.
  ........
................
---

diff --git a/Lib/lib2to3/fixes/fix_import.py b/Lib/lib2to3/fixes/fix_import.py
index 64397e4337e2..5612ba6f2c9e 100644
--- a/Lib/lib2to3/fixes/fix_import.py
+++ b/Lib/lib2to3/fixes/fix_import.py
@@ -45,7 +45,7 @@ class FixImport(fixer_base.BaseFix):
             node.changed()
         else:
             new = FromImport('.', getattr(imp, 'content', None) or [imp])
-            new.prefix = node.get_prefix()
+            new.set_prefix(node.get_prefix())
             node = new
         return node
 
diff --git a/Lib/lib2to3/tests/test_fixers.py b/Lib/lib2to3/tests/test_fixers.py
index 6d8b34e9edcc..dce6f608a263 100755
--- a/Lib/lib2to3/tests/test_fixers.py
+++ b/Lib/lib2to3/tests/test_fixers.py
@@ -3319,6 +3319,17 @@ class Test_import(FixerTestCase):
         a = "from . import foo.bar as bang"
         self.check_both(b, a)
 
+    def test_prefix(self):
+        b = """
+        # prefix
+        import foo.bar
+        """
+        a = """
+        # prefix
+        from . import foo.bar
+        """
+        self.check_both(b, a)
+
 
 if __name__ == "__main__":
     import __main__