Refactor call_subprocess into call_subprocess_for_install

Author: swchoi727

news/7711.feature

@@ -0,0 +1 @@
+Refactor `call_subprocess` by renaming it into `call_subprocess_for_install` and making it a wrapper of the former `call_subprocess` function.

src/pip/_internal/build_env.py

@@ -17,7 +17,7 @@
 
 from pip import __file__ as pip_location
 from pip._internal.cli.spinners import open_spinner
-from pip._internal.utils.subprocess import call_subprocess
+from pip._internal.utils.subprocess import call_subprocess_for_install
 from pip._internal.utils.temp_dir import TempDirectory, tempdir_kinds
 from pip._internal.utils.typing import MYPY_CHECK_RUNNING
 
@@ -196,7 +196,7 @@ def install_requirements(
         args.append('--')
         args.extend(requirements)
         with open_spinner(message) as spinner:
-            call_subprocess(args, spinner=spinner)
+            call_subprocess_for_install(args, spinner=spinner)
 
 
 class NoOpBuildEnvironment(BuildEnvironment):

src/pip/_internal/operations/build/metadata_legacy.py

@@ -7,7 +7,7 @@
 from pip._internal.exceptions import InstallationError
 from pip._internal.utils.misc import ensure_dir
 from pip._internal.utils.setuptools_build import make_setuptools_egg_info_args
-from pip._internal.utils.subprocess import call_subprocess
+from pip._internal.utils.subprocess import call_subprocess_for_install
 from pip._internal.utils.typing import MYPY_CHECK_RUNNING
 from pip._internal.vcs import vcs
 
@@ -112,7 +112,7 @@ def generate_metadata(
     )
 
     with build_env:
-        call_subprocess(
+        call_subprocess_for_install(
             args,
             cwd=source_dir,
             command_desc='python setup.py egg_info',

src/pip/_internal/operations/build/wheel_legacy.py

@@ -7,7 +7,7 @@
 )
 from pip._internal.utils.subprocess import (
     LOG_DIVIDER,
-    call_subprocess,
+    call_subprocess_for_install,
     format_command_args,
 )
 from pip._internal.utils.typing import MYPY_CHECK_RUNNING
@@ -94,7 +94,7 @@ def build_wheel_legacy(
         logger.debug('Destination directory: %s', tempd)
 
         try:
-            output = call_subprocess(
+            output = call_subprocess_for_install(
                 wheel_args,
                 cwd=source_dir,
                 spinner=spinner,

src/pip/_internal/operations/install/editable_legacy.py

@@ -4,7 +4,7 @@
 
 from pip._internal.utils.logging import indent_log
 from pip._internal.utils.setuptools_build import make_setuptools_develop_args
-from pip._internal.utils.subprocess import call_subprocess
+from pip._internal.utils.subprocess import call_subprocess_for_install
 from pip._internal.utils.typing import MYPY_CHECK_RUNNING
 
 if MYPY_CHECK_RUNNING:
@@ -46,7 +46,7 @@ def install_editable(
 
     with indent_log():
         with build_env:
-            call_subprocess(
+            call_subprocess_for_install(
                 args,
                 cwd=unpacked_source_directory,
             )

src/pip/_internal/utils/subprocess.py

@@ -27,6 +27,18 @@
 LOG_DIVIDER = '----------------------------------------'
 
 
+class SubProcessResult(object):
+    def __init__(self, cmd, returncode, had_error, output=None):
+        # type (...) -> SubProcessResult
+        """
+        Initialize a SubProcessResult
+        """
+        self.cmd = cmd
+        self.returncode = returncode
+        self.output = output
+        self.had_error = had_error
+
+
 def make_command(*args):
     # type: (Union[str, HiddenText, CommandArgs]) -> CommandArgs
     """
@@ -113,7 +125,7 @@ def make_subprocess_output_error(
     return msg
 
 
-def call_subprocess(
+def call_subprocess_for_install(
     cmd,  # type: Union[List[str], CommandArgs]
     show_stdout=False,  # type: bool
     cwd=None,  # type: Optional[str]
@@ -136,10 +148,6 @@ def call_subprocess(
         prior to calling subprocess.Popen().
       log_failed_cmd: if false, failed commands are not logged, only raised.
     """
-    if extra_ok_returncodes is None:
-        extra_ok_returncodes = []
-    if unset_environ is None:
-        unset_environ = []
     # Most places in pip use show_stdout=False. What this means is--
     #
     # - We connect the child's output (combined stderr and stdout) to a
@@ -155,55 +163,101 @@ def call_subprocess(
     #
     # If show_stdout=True, then the above is still done, but with DEBUG
     # replaced by INFO.
-    if show_stdout:
-        # Then log the subprocess output at INFO level.
-        log_subprocess = subprocess_logger.info
-        used_level = logging.INFO
-    else:
-        # Then log the subprocess output using DEBUG.  This also ensures
-        # it will be logged to the log file (aka user_log), if enabled.
-        log_subprocess = subprocess_logger.debug
-        used_level = logging.DEBUG
-
+    used_level = logging.INFO if show_stdout else logging.DEBUG
     # Whether the subprocess will be visible in the console.
     showing_subprocess = subprocess_logger.getEffectiveLevel() <= used_level
-
-    # Only use the spinner if we're not showing the subprocess output
-    # and we have a spinner.
-    use_spinner = not showing_subprocess and spinner is not None
-
     if command_desc is None:
         command_desc = format_command_args(cmd)
+    proc = _call_subprocess(
+        cmd=cmd, cwd=cwd, spinner=spinner, command_desc=command_desc,
+        extra_environ=extra_environ, unset_environ=unset_environ,
+        log_failed_cmd=log_failed_cmd, log_level=used_level,
+        extra_ok_returncodes=extra_ok_returncodes)
+    if proc.had_error:
+        if on_returncode == "raise":
+            if not showing_subprocess and log_failed_cmd:
+                # Then the subprocess streams haven't been logged to the
+                # console yet.
+                msg = make_subprocess_output_error(
+                    cmd_args=cmd,
+                    cwd=cwd,
+                    lines=proc.output,
+                    exit_status=proc.returncode,
+                )
+                subprocess_logger.error(msg)
+            exc_msg = (
+                "Command errored out with exit status {}: {} "
+                "Check the logs for full command output."
+            ).format(proc.returncode, command_desc)
+            raise InstallationError(exc_msg)
+        elif on_returncode == "warn":
+            subprocess_logger.warning(
+                "Command '{}' had error code {} in {}".format(
+                    command_desc, proc.returncode, cwd))
+        elif on_returncode == "ignore":
+            pass
+        else:
+            raise ValueError(
+                "Invalid value: on_returncode={!r}".format(on_returncode))
+    return "".join(proc.output)
 
-    log_subprocess("Running command %s", command_desc)
+
+def _call_subprocess(
+    cmd,                        # type: Union[List[str], CommandArgs]
+    spinner=None,               # type: Optional[SpinnerInterface]
+    cwd=None,                   # type: Optional[str]
+    log_level=logging.DEBUG,    # type: int
+    command_desc=None,          # type: Optional[str]
+    extra_environ=None,         # type: Optional[Mapping[str, Any]]
+    unset_environ=None,         # type: Optional[Iterable[str]]
+    extra_ok_returncodes=None,  # type: Optional[Iterable[int]]
+    log_failed_cmd=True         # type: Optional[bool]
+):
+    """
+    type: (...) -> SubProcessResult
+    This function calls the subprocess and returns a SubProcessResult
+    object representing the stdout lines, returncode, and command that
+    was executed. If subprocess logging is enabled, then display it
+    on stdout with the INFO logging level
+    """
+    if unset_environ is None:
+        unset_environ = []
+    if extra_ok_returncodes is None:
+        extra_ok_returncodes = []
+    if log_level == logging.DEBUG:
+        log_subprocess = subprocess_logger.debug
+    else:
+        log_subprocess = subprocess_logger.info
+    showing_subprocess = subprocess_logger.getEffectiveLevel() <= log_level
+    use_spinner = not showing_subprocess and spinner is not None
+    log_subprocess("Running command {}".format(command_desc))
     env = os.environ.copy()
     if extra_environ:
         env.update(extra_environ)
     for name in unset_environ:
         env.pop(name, None)
+
+    all_output = []
     try:
+        # Convert HiddenText objects to the underlying str.
         proc = subprocess.Popen(
-            # Convert HiddenText objects to the underlying str.
             reveal_command_args(cmd),
             stderr=subprocess.STDOUT, stdin=subprocess.PIPE,
-            stdout=subprocess.PIPE, cwd=cwd, env=env,
+            stdout=subprocess.PIPE, cwd=cwd, env=env
         )
-        proc.stdin.close()
     except Exception as exc:
         if log_failed_cmd:
             subprocess_logger.critical(
-                "Error %s while executing command %s", exc, command_desc,
+                "Error %s while executing command %s", exc, command_desc
             )
         raise
-    all_output = []
     while True:
         # The "line" value is a unicode string in Python 2.
         line = console_to_str(proc.stdout.readline())
         if not line:
             break
         line = line.rstrip()
-        all_output.append(line + '\n')
-
+        all_output.append("{}\n".format(line))
         # Show the line immediately.
         log_subprocess(line)
         # Update the spinner.
@@ -214,42 +268,16 @@ def call_subprocess(
     finally:
         if proc.stdout:
             proc.stdout.close()
-    proc_had_error = (
-        proc.returncode and proc.returncode not in extra_ok_returncodes
-    )
+    had_error = proc.returncode and proc.returncode not in extra_ok_returncodes
     if use_spinner:
-        if proc_had_error:
+        if had_error:
             spinner.finish("error")
         else:
             spinner.finish("done")
-    if proc_had_error:
-        if on_returncode == 'raise':
-            if not showing_subprocess and log_failed_cmd:
-                # Then the subprocess streams haven't been logged to the
-                # console yet.
-                msg = make_subprocess_output_error(
-                    cmd_args=cmd,
-                    cwd=cwd,
-                    lines=all_output,
-                    exit_status=proc.returncode,
-                )
-                subprocess_logger.error(msg)
-            exc_msg = (
-                'Command errored out with exit status {}: {} '
-                'Check the logs for full command output.'
-            ).format(proc.returncode, command_desc)
-            raise InstallationError(exc_msg)
-        elif on_returncode == 'warn':
-            subprocess_logger.warning(
-                'Command "{}" had error code {} in {}'.format(
-                    command_desc, proc.returncode, cwd)
-            )
-        elif on_returncode == 'ignore':
-            pass
-        else:
-            raise ValueError('Invalid value: on_returncode={!r}'.format(
-                             on_returncode))
-    return ''.join(all_output)
+    return SubProcessResult(
+        cmd=cmd, returncode=proc.returncode,
+        had_error=had_error, output=all_output
+    )
 
 
 def runner_with_spinner_message(message):
@@ -267,7 +295,7 @@ def runner(
     ):
         # type: (...) -> None
         with open_spinner(message) as spinner:
-            call_subprocess(
+            call_subprocess_for_install(
                 cmd,
                 cwd=cwd,
                 extra_environ=extra_environ,

src/pip/_internal/vcs/subversion.py

@@ -270,7 +270,7 @@ def get_remote_call_options(self):
         svn_version = self.get_vcs_version()
         # By default, Subversion >= 1.8 runs in non-interactive mode if
         # stdin is not a TTY. Since that is how pip invokes SVN, in
-        # call_subprocess(), pip must pass --force-interactive to ensure
+        # call_subprocess_for_install(), pip must pass --force-interactive to ensure
         # the user can be prompted for a password, if required.
         #   SVN added the --force-interactive option in SVN 1.8. Since
         # e.g. RHEL/CentOS 7, which is supported until 2024, ships with

src/pip/_internal/vcs/versioncontrol.py

@@ -21,7 +21,9 @@
     hide_value,
     rmtree,
 )
-from pip._internal.utils.subprocess import call_subprocess, make_command
+from pip._internal.utils.subprocess import (
+    call_subprocess_for_install, make_command
+)
 from pip._internal.utils.typing import MYPY_CHECK_RUNNING
 from pip._internal.utils.urls import get_url_scheme
 
@@ -279,7 +281,8 @@ class VersionControl(object):
     repo_name = ''
     # List of supported schemes for this Version Control
     schemes = ()  # type: Tuple[str, ...]
-    # Iterable of environment variable names to pass to call_subprocess().
+    # Iterable of environment variable names to pass to 
+    # call_subprocess_for_install().
     unset_environ = ()  # type: Tuple[str, ...]
     default_arg_rev = None  # type: Optional[str]
 
@@ -665,12 +668,13 @@ def run_command(
         # type: (...) -> Text
         """
         Run a VCS subcommand
-        This is simply a wrapper around call_subprocess that adds the VCS
-        command name, and checks that the VCS is available
+        This is simply a wrapper around call_subprocess_for_install 
+        that adds the VCS command name, and checks that the VCS
+         is available
         """
         cmd = make_command(cls.name, *cmd)
         try:
-            return call_subprocess(cmd, show_stdout, cwd,
+            return call_subprocess_for_install(cmd, show_stdout, cwd,
                                    on_returncode=on_returncode,
                                    extra_ok_returncodes=extra_ok_returncodes,
                                    command_desc=command_desc,

src/pip/_internal/wheel_builder.py

@@ -15,7 +15,7 @@
 from pip._internal.utils.logging import indent_log
 from pip._internal.utils.misc import ensure_dir, hash_file, is_wheel_installed
 from pip._internal.utils.setuptools_build import make_setuptools_clean_args
-from pip._internal.utils.subprocess import call_subprocess
+from pip._internal.utils.subprocess import call_subprocess_for_install
 from pip._internal.utils.temp_dir import TempDirectory
 from pip._internal.utils.typing import MYPY_CHECK_RUNNING
 from pip._internal.utils.urls import path_to_url
@@ -246,7 +246,7 @@ def _clean_one_legacy(req, global_options):
 
     logger.info('Running setup.py clean for %s', req.name)
     try:
-        call_subprocess(clean_args, cwd=req.source_dir)
+        call_subprocess_for_install(clean_args, cwd=req.source_dir)
         return True
     except Exception:
         logger.error('Failed cleaning build dir for %s', req.name)