check by equality for __future__ not identity (closes #14378)
authorBenjamin Peterson <benjamin@python.org>
Thu, 22 Mar 2012 12:19:04 +0000 (08:19 -0400)
committerBenjamin Peterson <benjamin@python.org>
Thu, 22 Mar 2012 12:19:04 +0000 (08:19 -0400)
Lib/test/test_ast.py
Misc/NEWS
Python/future.c

index f4ce615ad8fa34e527c7d08f0bc8270157dc5d76..e22d4de88ba763dc396ef3b693868625a83c14c0 100644 (file)
@@ -218,6 +218,12 @@ class AST_Tests(unittest.TestCase):
         im = ast.parse("from . import y").body[0]
         self.assertIsNone(im.module)
 
+    def test_non_interned_future_from_ast(self):
+        mod = ast.parse("from __future__ import division")
+        self.assertIsInstance(mod.body[0], ast.ImportFrom)
+        mod.body[0].module = " __future__ ".strip()
+        compile(mod, "<test>", "exec")
+
     def test_base_classes(self):
         self.assertTrue(issubclass(ast.For, ast.stmt))
         self.assertTrue(issubclass(ast.Name, ast.expr))
index 606898f6d2c4acb61ad4bf9814f29ed7ca456273..1c1bf1646677a631da5b8d3bb77640dded41d653 100644 (file)
--- a/Misc/NEWS
+++ b/Misc/NEWS
@@ -10,6 +10,9 @@ What's New in Python 3.2.4
 Core and Builtins
 -----------------
 
+- Issue #14378: Fix compiling ast.ImportFrom nodes with a "__future__" string as
+  the module name that was not interned.
+
 - Issue #14331: Use significantly less stack space when importing modules by
   allocating path buffers on the heap instead of the stack.
 
index d6b653f31593b8fb5e754d3fb8447a0e9da8dc6a..978dc253065b5131dbbfb1de906979395e42c026 100644 (file)
@@ -60,13 +60,6 @@ future_parse(PyFutureFeatures *ff, mod_ty mod, const char *filename)
 {
     int i, found_docstring = 0, done = 0, prev_line = 0;
 
-    static PyObject *future;
-    if (!future) {
-        future = PyUnicode_InternFromString("__future__");
-        if (!future)
-            return 0;
-    }
-
     if (!(mod->kind == Module_kind || mod->kind == Interactive_kind))
         return 1;
 
@@ -93,7 +86,8 @@ future_parse(PyFutureFeatures *ff, mod_ty mod, const char *filename)
         */
 
         if (s->kind == ImportFrom_kind) {
-            if (s->v.ImportFrom.module == future) {
+            PyObject *modname = s->v.ImportFrom.module;
+            if (!PyUnicode_CompareWithASCIIString(modname, "__future__")) {
                 if (done) {
                     PyErr_SetString(PyExc_SyntaxError,
                                     ERR_LATE_FUTURE);