mirror of
https://github.com/saymrwulf/onnxruntime.git
synced 2026-05-15 20:50:42 +00:00
dda9f53bed
Log build.py command line arguments. Update subprocess logging to format arguments in way that is easier to copy.
50 lines
1.6 KiB
Python
50 lines
1.6 KiB
Python
# Copyright (c) Microsoft Corporation. All rights reserved.
|
|
# Licensed under the MIT License.
|
|
|
|
import logging
|
|
import os
|
|
import shlex
|
|
import subprocess
|
|
|
|
|
|
_log = logging.getLogger("util.run")
|
|
|
|
|
|
def run(*args, cwd=None,
|
|
input=None, capture_stdout=False, capture_stderr=False,
|
|
shell=False, env=None, check=True, quiet=False):
|
|
"""Runs a subprocess.
|
|
|
|
Args:
|
|
*args: The subprocess arguments.
|
|
cwd: The working directory. If None, specifies the current directory.
|
|
input: The optional input byte sequence.
|
|
capture_stdout: Whether to capture stdout.
|
|
capture_stderr: Whether to capture stderr.
|
|
shell: Whether to run using the shell.
|
|
env: The environment variables as a dict. If None, inherits the current
|
|
environment.
|
|
check: Whether to raise an error if the return code is not zero.
|
|
quiet: If true, do not print output from the subprocess.
|
|
|
|
Returns:
|
|
A subprocess.CompletedProcess instance.
|
|
"""
|
|
cmd = [*args]
|
|
|
|
_log.info("Running subprocess in '{0}'\n {1}".format(
|
|
cwd or os.getcwd(), " ".join([shlex.quote(arg) for arg in cmd])))
|
|
|
|
def output(is_stream_captured):
|
|
return subprocess.PIPE if is_stream_captured else \
|
|
(subprocess.DEVNULL if quiet else None)
|
|
|
|
completed_process = subprocess.run(
|
|
cmd, cwd=cwd, check=check, input=input,
|
|
stdout=output(capture_stdout), stderr=output(capture_stderr),
|
|
env=env, shell=shell)
|
|
|
|
_log.debug("Subprocess completed. Return code: {}".format(
|
|
completed_process.returncode))
|
|
|
|
return completed_process
|