Reference utility functions from _shutil.

Author: jaraco

setuptools/_shutil.py

@@ -51,3 +51,9 @@ def rmtree(path, ignore_errors=False, onexc=_auto_chmod):
 def rmdir(path, **opts):
     if os.path.isdir(path):
         rmtree(path, **opts)
+
+
+def current_umask():
+    tmp = os.umask(0o022)
+    os.umask(tmp)
+    return tmp

setuptools/command/easy_install.py

@@ -60,7 +60,7 @@
 
 from .._path import ensure_directory
 from .._scripts import CommandSpec, ScriptWriter
-from .._shutil import attempt_chmod_verbose as chmod, rmtree as _rmtree
+from .._shutil import attempt_chmod_verbose as chmod, current_umask, rmtree as _rmtree
 from ..compat import py39, py312
 
 from distutils import dir_util, log
@@ -1954,12 +1954,6 @@ def is_python_script(script_text, filename):
 sys_executable = CommandSpec._sys_executable()
 
 
-def current_umask():
-    tmp = os.umask(0o022)
-    os.umask(tmp)
-    return tmp
-
-
 def only_strs(values):
     """
     Exclude non-str values. Ref #3063.

setuptools/command/install_scripts.py

@@ -58,7 +58,7 @@ def _install_ep_scripts(self):
 
     def write_script(self, script_name, contents, mode: str = "t", *ignored) -> None:
         """Write an executable file to the scripts directory"""
-        from setuptools.command.easy_install import chmod, current_umask
+        from .._shutil import attempt_chmod_verbose as chmod, current_umask
 
         log.info("Installing %s script to %s", script_name, self.install_dir)
         target = os.path.join(self.install_dir, script_name)