Function mssh

misc/python/materialize/scratch.py:229–264

Runs a command over SSH via EC2 Instance Connect.

(
    instance: Instance,
    command: str,
    *,
    extra_ssh_args: list[str] | None = None,
    input: bytes | None = None,
    quiet: bool = False,
)

def mssh(
    instance: Instance,
    command: str,
    *,
    extra_ssh_args: list[str] | None = None,
    input: bytes | None = None,
    quiet: bool = False,
) -> None:
    """Runs a command over SSH via EC2 Instance Connect."""
    extra_ssh_args = extra_ssh_args or []
    host = instance_host(instance)
    if command:
        if not quiet:
            print(f"{host}$ {command}", file=sys.stderr)
        # Quote to work around:
        # https://github.com/aws/aws-ec2-instance-connect-cli/pull/26
        command = shlex.quote(command)
    else:
        print(f"$ mssh {host}")

    result = subprocess.run(
        [
            *SSH_COMMAND,
            *extra_ssh_args,
            host,
            command,
        ],
        input=input,
        stdout=subprocess.DEVNULL if quiet else None,
        stderr=subprocess.DEVNULL if quiet else None,
    )
    # Exit code 130 = SIGINT (Ctrl-C) — exit cleanly
    if result.returncode == 130:
        sys.exit(130)
    if result.returncode != 0:
        raise subprocess.CalledProcessError(result.returncode, result.args)


def msftp(