]> git.ipfire.org Git - thirdparty/Python/cpython.git/commitdiff
[3.13] gh-128030: Avoid error from PyModule_GetFilenameObject for non-module (GH...
authorMiss Islington (bot) <31488909+miss-islington@users.noreply.github.com>
Fri, 20 Dec 2024 08:46:54 +0000 (09:46 +0100)
committerGitHub <noreply@github.com>
Fri, 20 Dec 2024 08:46:54 +0000 (08:46 +0000)
gh-128030: Avoid error from PyModule_GetFilenameObject for non-module (GH-128047)

I missed the extra `PyModule_Check` in GH-127660 because I was looking at
3.12 as the base implementation for import from. This meant that I
missed the `PyModuleCheck` introduced in GH-112661.
(cherry picked from commit 45e6dd63b88a782f2ec96ab1da54eb5a074d8f4c)

Co-authored-by: Shantanu <12621235+hauntsaninja@users.noreply.github.com>
Lib/test/test_import/__init__.py
Misc/NEWS.d/next/Core_and_Builtins/2024-12-17-22-28-15.gh-issue-128030.H1ptOD.rst [new file with mode: 0644]
Python/ceval.c

index 58c6aaf0986bcf769f96c5b71417655e130d0d53..364ff08f03533b0d18ee04490cba8999ce0d5bb3 100644 (file)
@@ -870,6 +870,29 @@ from os import this_will_never_exist
                 stdout, stderr = popen.communicate()
                 self.assertIn(expected_error, stdout)
 
+    def test_non_module_from_import_error(self):
+        prefix = """
+import sys
+class NotAModule: ...
+nm = NotAModule()
+nm.symbol = 123
+sys.modules["not_a_module"] = nm
+from not_a_module import symbol
+"""
+        scripts = [
+            prefix + "from not_a_module import missing_symbol",
+            prefix + "nm.__spec__ = []\nfrom not_a_module import missing_symbol",
+        ]
+        for script in scripts:
+            with self.subTest(script=script):
+                expected_error = (
+                    b"ImportError: cannot import name 'missing_symbol' from "
+                    b"'<unknown module name>' (unknown location)"
+                )
+            popen = script_helper.spawn_python("-c", script)
+            stdout, stderr = popen.communicate()
+            self.assertIn(expected_error, stdout)
+
     def test_script_shadowing_stdlib(self):
         script_errors = [
             (
diff --git a/Misc/NEWS.d/next/Core_and_Builtins/2024-12-17-22-28-15.gh-issue-128030.H1ptOD.rst b/Misc/NEWS.d/next/Core_and_Builtins/2024-12-17-22-28-15.gh-issue-128030.H1ptOD.rst
new file mode 100644 (file)
index 0000000..93d7863
--- /dev/null
@@ -0,0 +1 @@
+Avoid error from calling ``PyModule_GetFilenameObject`` on a non-module object when importing a non-existent symbol from a non-module object.
index 4796ed301826fd1e1b99b48dd6ad41126130b232..64424ed939fc5bb07619b92a6cb8dd0ef0735432 100644 (file)
@@ -2781,7 +2781,7 @@ import_from(PyThreadState *tstate, PyObject *v, PyObject *name)
         }
     }
 
-    if (origin == NULL) {
+    if (origin == NULL && PyModule_Check(v)) {
         // Fall back to __file__ for diagnostics if we don't have
         // an origin that is a location
         origin = PyModule_GetFilenameObject(v);