使用Python脚本循环创建多个EC2实例

44次阅读

问题描述

在使用Python脚本循环创建多个EC2实例时遇到了问题。他有一个JSON文件，其中包含了邮箱地址和实例数量的信息。他的目标是根据JSON文件中的用户名和实例数量创建相应数量的EC2实例。但是他在运行脚本后发现，当JSON文件中包含多个条目时，只有最后一个创建的EC2实例保持运行，之前的实例都被终止了。他怀疑问题出在代码中，希望能够找到解决方法。

解决方案

问题出在于用户在每次循环迭代中都调用了Terraform，导致每次迭代都会创建一个新的状态文件，使得实例的管理变得困难。下面是一个解决方案，通过修改Python脚本和Terraform配置来解决这个问题。

首先，我们要修改Python脚本，以便在一次循环迭代中获取所有需要创建实例的用户名和数量。然后，我们将这些信息一次性传递给Terraform，以确保只调用一次Terraform。以下是修改后的Python脚本示例：

#!/bin/python
import json
from python_terraform import *

def myfunc(count, users):
  tf = Terraform(working_dir='/home/ja/terraform-course/demo-2b', variables={'count': count, 'INSTANCE_USERS': users})
  tf.plan(no_color=IsFlagged, refresh=False, capture_output=False)
  approve = {"auto-approve": True}
  print(tf.plan())
  print(tf.apply(**approve))
  return

json_data = open('./my.json')
data = json.load(json_data)
json_data.close()

users = []
count = 0

for customer in data['customers']:
    username = customer['email'].split('@')[0]
    instances = customer['instances']
    count += instances
    users.extend([username] * instances)

myfunc(count, users)

接下来，我们需要修改Terraform配置文件（windows.tf），以便正确地根据传递的用户名列表创建相应数量的实例。我们将使用count.index来获取正确的用户名，并保持实例的一致性。以下是修改后的Terraform配置文件示例：

resource "aws_key_pair" "mykey" {
  key_name   = "mykey"
  public_key = "${file("${var.PATH_TO_PUBLIC_KEY}")}"
}

resource "aws_instance" "win-example" {
  ami           = "${lookup(var.WIN_AMIS, var.AWS_REGION)}"
  instance_type = "t2.medium"
  count         = "${var.count}"
  key_name      = "${aws_key_pair.mykey.key_name}"
  user_data = <<EOF
<powershell>
net user ${var.INSTANCE_USERS[count.index]} '${var.INSTANCE_PASSWORD}' /add /y
net localgroup administrators ${var.INSTANCE_USERS[count.index]} /add
winrm quickconfig -q
winrm set winrm/config/winrs '@{MaxMemoryPerShellMB="300"}'
winrm set winrm/config '@{MaxTimeoutms="1800000"}'
winrm set winrm/config/service '@{AllowUnencrypted="true"}'
winrm set winrm/config/service/auth '@{Basic="true"}'
netsh advfirewall firewall add rule name="WinRM 5985" protocol=TCP dir=in localport=5985 action=allow
netsh advfirewall firewall add rule name="WinRM 5986" protocol=TCP dir=in localport=5986 action=allow
net stop winrm
sc.exe config winrm start=auto
net start winrm
</powershell>
EOF

  provisioner "file" {
    source      = "test.txt"
    destination = "C:/test.txt"
  }

  connection {
    type     = "winrm"
    timeout  = "10m"
    user     = "${var.INSTANCE_USERS[count.index]}"
    password = "${var.INSTANCE_PASSWORD}"
  }

  tags {
    Name = "${var.INSTANCE_USERS[count.index]}"
  }
}

上述配置文件中，我们使用了${var.INSTANCE_USERS[count.index]}来获取正确的用户名。这样做可以保证每个实例都使用对应的用户名。

请注意，count.index用于获取当前迭代的索引，从而正确地匹配用户名和实例数量。这样，每个用户都会获得所需数量的实例，并且实例之间不会被终止。

这个解决方案的关键是将所有的用户名和实例数量一次性传递给Terraform，以确保所有实例都在同一个Terraform调用中进行管理。

如果您需要进一步了解count.index的用法，可以阅读Terraform官方文档。

希望这个解决方案能帮助您解决问题。如果还有疑问，请随时提问。

正文完