使用 subprocess 和 aws ssm 在 EC2 上执行命令

Question

我正在使用部署在子网中的一组 EC2 实例。

我通常使用 aws ssm 命令

aws ssm start-session --region us-east-2 --target i-01234567abcdef --profile profile-one

连接到该 ec2 实例并运行命令，例如

ssh mirrora

到 go 到子网中与我连接的不同的 ec2 实例

然后我可以在 mirrora 上运行 ls 来查看文件等。

现在我正在尝试使用 python 自动执行相同的操作

我是 python 的新手（也是第一次使用子进程）。 我研究了一下，了解到我可以使用 Popen 打开一个子进程，并将标准输入和标准输出 arguments 设置为 subprocess.PIPE

这是我尝试执行的代码示例

    command = "aws ssm start-session --region us-east-2 --target i-0123456abcdef --profile profile-one"
    
    ssh = subprocess.Popen(command.split(), stdin=subprocess.PIPE, stdout=subprocess.PIPE, bufsize=0, shell= True, universal_newlines=True)
    
    ssh.stdin.write("ssh mirrora")
    ssh.stdin.write("ls")
    # output = ssh.stdout.read()
    # output , err = ssh.communicate()
    # print("output:" + output)
    # print("err:" + err)
    # output = ssh.stdout.read()
    # print("Output of ls is:", output)

    

    print("HERE")

如果我使用 stdout.read 或通信方法，程序似乎卡住了，没有任何效果。 我什至不能 ctrl + c 终端。

如果我在没有任何标准输出或通信的情况下运行它，它会在终端中打印“HERE”（几乎是瞬间），并且服务器端实际上似乎没有发生任何事情（因为我试图运行关闭命令来关闭 mirrora，但它是程序在没有标准输出或通信的情况下退出后仍在运行）。

我究竟做错了什么？

理想情况下，我想在 aws ssm 打开的子进程中运行我想要的命令，并在其中一个 ec2 实例上运行某些命令，并在某些 stdin.write 之后使用 stdout.read 为一个或多个命令获取 output

任何帮助或链接将不胜感激。

谢谢

Answer 1

在我看来，您应该使用run-command而不是start-session 。

这是用于执行 static 命令和脚本的 SSM 功能，其中 session 类似于仅 ssh 连接。

我认为它应该像这样工作：

import boto3

script = f"""
ssh mirrora 'ls'
ssh mirrora 'ls; ps -aux'
"""
target = "i-01234567abcdef "

session = boto3.Session(profile_name="profile-one")
ssm_client = session.client("ssm")


response = ssm_client.send_command(
    InstanceIds=[target],
    DocumentName="AWS-RunShellScript",  # https://eu-central-1.console.aws.amazon.com/systems-manager/documents/AWS-RunShellScript/description?region=eu-central-1
    TimeoutSeconds=20,
    Comment="Example of send command",
    Parameters={
        'commands': [
            script,
        ],
        'workingDirectory': "/home/user1",
    },
)

在这里您可以找到发送命令 API 的文档。 https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/ssm.html#SSM.Client.send_command

如果您想查看 python 脚本中的日志，那么您需要扩展它 - 可能最简单的方法是在send_command中设置一个 s3 存储桶（它会将您的stdout和stderr重定向到提供的 s3 存储桶），下载保存在 s3 文件中，并 cat/打印它们。