vscode-python/python_files/shell_exec.py at main · andybbruno/vscode-python · GitHub

42 lines (34 loc) · 1.37 KB

# Licensed under the MIT License.

import subprocess

import sys

# This is a simple solution to waiting for completion of commands sent to terminal.

# 1. Intercept commands send to a terminal

# 2. Send commands to our script file with an additional argument

# 3. In here create a file that'll log the progress.

# 4. Calling code monitors the contents of the file to determine state of execution.

# Last argument is a file that's used for synchronizing the actions in the terminal with the calling code in extension.

lock_file = sys.argv[-1]

shell_args = sys.argv[1:-1]

print("Executing command in shell >> " + " ".join(shell_args))

with open(lock_file, "w") as fp:

try:

# Signal start of execution.

fp.write("START\n")

fp.flush()

subprocess.check_call(shell_args, stdout=sys.stdout, stderr=sys.stderr)

# Signal start of execution.

fp.write("END\n")

fp.flush()

except Exception:

import traceback

print(traceback.format_exc())

# Signal end of execution with failure state.

fp.write("FAIL\n")

fp.flush()

try:

# ALso log the error for use from the other side.

with open(lock_file + ".error", "w") as fpError:

fpError.write(traceback.format_exc())

except Exception:

pass

Expand file tree