Avoid KeyError in distutils hack when pip is imported during ensurepip.

Fixes #2940.
pypa · Dec 21, 2021 · 137ab9d · 137ab9d
1 parent 82726bb
commit 137ab9d
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/_distutils_hack/__init__.py b/_distutils_hack/__init__.py
@@ -119,7 +119,11 @@ def pip_imported_during_build(cls):
 
     @staticmethod
     def frame_file_is_setup(frame):
-        return frame.f_globals['__file__'].endswith('setup.py')
+        """
+        Return True if the indicated frame suggests a setup.py file.
+        """
+        # some frames may not have __file__ (#2940)
+        return frame.f_globals.get('__file__', '').endswith('setup.py')
 
 
 DISTUTILS_FINDER = DistutilsMetaFinder()

diff --git a/changelog.d/2940.misc.rst b/changelog.d/2940.misc.rst
@@ -0,0 +1 @@
+Avoid KeyError in distutils hack when pip is imported during ensurepip.