43 lines
1.4 KiB
Python
43 lines
1.4 KiB
Python
# Copyright (c) Microsoft Corporation. All rights reserved.
|
|
# 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: # noqa: PTH123
|
|
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 fp_error: # noqa: PTH123
|
|
fp_error.write(traceback.format_exc())
|
|
except Exception:
|
|
pass
|