Fix security group id for vpc based on @samukasmk solution

pull/2574/merge
Mario Franco 12 years ago committed by Michael DeHaan
parent 29d1b08db4
commit f74316607f

@ -277,11 +277,22 @@ def main():
module.fail_json(msg = str(e)) module.fail_json(msg = str(e))
# Here we try to lookup the group name from the security group id - if group_id is set. # Here we try to lookup the group name from the security group id - if group_id is set.
if group_id and group_name:
module.fail_json(msg = str("Use only one type of parameter (group_name) or (group_id)"))
sys.exit(1)
try: try:
if group_id: # Here we try to lookup the group id from the security group name - if group is set.
if group_name:
grp_details = ec2.get_all_security_groups()
for grp in grp_details:
if str(group_name) in str(grp):
group_id = str(grp.id)
# Now we try to lookup the group id testing if group exists.
elif group_id:
grp_details = ec2.get_all_security_groups(group_ids=group_id) grp_details = ec2.get_all_security_groups(group_ids=group_id)
grp_item = grp_details[0] grp_item = grp_details[0]
group_name = grp_item.name group_name = grp_item.name
except boto.exception.NoAuthHandlerFound, e: except boto.exception.NoAuthHandlerFound, e:
module.fail_json(msg = str(e)) module.fail_json(msg = str(e))
@ -303,19 +314,25 @@ def main():
if count_remaining > 0: if count_remaining > 0:
try: try:
res = ec2.run_instances(image, key_name = key_name, params = {'image_id': image,
client_token=id, 'key_name': key_name,
min_count = count_remaining, 'client_token': id,
max_count = count_remaining, 'min_count': count_remaining,
monitoring_enabled = monitoring, 'max_count': count_remaining,
security_groups = [group_name], 'monitoring_enabled': monitoring,
placement = zone, 'instance_type': instance_type,
instance_type = instance_type, 'kernel_id': kernel,
kernel_id = kernel, 'ramdisk_id': ramdisk,
ramdisk_id = ramdisk, 'subnet_id': vpc_subnet_id,
subnet_id = vpc_subnet_id, 'private_ip_address': private_ip,
private_ip_address = private_ip, 'user_data': user_data}
user_data = user_data)
if vpc_subnet_id:
params['security_group_ids'] = [group_id]
else:
params['security_groups'] = [group_name]
res = ec2.run_instances(**params)
except boto.exception.BotoServerError, e: except boto.exception.BotoServerError, e:
module.fail_json(msg = "%s: %s" % (e.error_code, e.error_message)) module.fail_json(msg = "%s: %s" % (e.error_code, e.error_message))

Loading…
Cancel
Save