Extract frame_file_is_setup.

1 files changed, 7 insertions, 3 deletions

diff --git a/_distutils_hack/__init__.py b/_distutils_hack/__init__.py
index ae97a0b2..da51b433 100644
--- a/_distutils_hack/__init__.py
+++ b/_distutils_hack/__init__.py
@@ -106,17 +106,21 @@ class DistutilsMetaFinder:
         clear_distutils()
         self.spec_for_distutils = lambda: None
 
-    @staticmethod
-    def pip_imported_during_build():
+    @classmethod
+    def pip_imported_during_build(cls):
         """
         Detect if pip is being imported in a build script. Ref #2355.
         """
         import traceback
         return any(
-            frame.f_globals['__file__'].endswith('setup.py')
+            cls.frame_file_is_setup(frame)
             for frame, line in traceback.walk_stack(None)
         )
 
+    @staticmethod
+    def frame_file_is_setup(frame):
+        return frame.f_globals['__file__'].endswith('setup.py')
+
 
 DISTUTILS_FINDER = DistutilsMetaFinder()