You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
50 lines
1.3 KiB
50 lines
1.3 KiB
1 year ago
|
"""
|
||
|
|
Startup site customization for Stability Matrix.
|
||
|
|
|
||
|
|
Currently this installs an audit hook to notify the parent process when input() is called,
|
||
|
|
so we can prompt the user to enter something.
|
||
|
|
"""
|
||
|
|
|
||
|
|
import sys
|
||
|
|
|
||
|
|
# Application Program Command escape sequence
|
||
|
|
# This wraps messages sent to the parent process.
|
||
|
|
esc_apc = "\x9F"
|
||
|
|
esc_prefix = "[SM;"
|
||
|
|
esc_st = "\x9C"
|
||
|
|
|
||
|
|
|
||
|
|
def send_apc(msg: str):
|
||
|
|
"""Send an Application Program Command to the parent process."""
|
||
|
|
sys.stdout.flush()
|
||
|
|
sys.stdout.write(esc_apc + esc_prefix + msg + esc_st)
|
||
|
|
sys.stdout.flush()
|
||
|
|
|
||
|
|
|
||
|
|
def send_apc_input(prompt: str):
|
||
|
|
"""Apc message for input() prompt."""
|
||
|
|
send_apc('{"type":"input","data":"' + str(prompt) + '"}')
|
||
|
|
|
||
|
|
|
||
|
|
def audit(event: str, *args):
|
||
|
|
"""Main audit hook function."""
|
||
|
|
# https://docs.python.org/3/library/functions.html#input
|
||
|
|
# input() raises audit event `builtins.input` with args (prompt: str) *before* reading from stdin.
|
||
|
|
# `builtins.input/result` raised after reading from stdin.
|
||
|
|
|
||
|
|
if event == "builtins.input":
|
||
|
|
try:
|
||
|
|
prompts = args[0] if args else ()
|
||
|
|
prompt = "".join(prompts)
|
||
|
|
send_apc_input(prompt)
|
||
|
|
except Exception:
|
||
|
|
pass
|
||
|
|
|
||
|
|
|
||
|
|
# Reconfigure stdout to UTF-8
|
||
|
|
# noinspection PyUnresolvedReferences
|
||
|
|
sys.stdout.reconfigure(encoding="utf-8")
|
||
|
|
|
||
|
|
# Install the audit hook
|
||
|
|
sys.addaudithook(audit)
|