Allow iam_role to work with friendly policy names (#22630)

Basically obtain all the policies and create a mapping
from friendly name to ARN

Tidy up module argument specification to remove unneeded
settings.
pull/25227/head
Will Thames 8 years ago committed by Sloane Hertel
parent 2f0ce790d1
commit bc24ae88a8

@ -43,9 +43,10 @@ options:
required: false required: false
managed_policy: managed_policy:
description: description:
- A list of managed policy ARNs (can't use friendly names due to AWS API limitation) to attach to the role. To embed an inline policy, - A list of managed policy ARNs or, since Ansible 2.4, a list of either managed policy ARNs or friendly names.
use M(iam_policy). To remove existing policies, use an empty list item. To embed an inline policy, use M(iam_policy). To remove existing policies, use an empty list item.
required: true required: true
aliases: ['managed_policies']
state: state:
description: description:
- Create or remove the IAM role - Create or remove the IAM role
@ -145,15 +146,15 @@ attached_policies:
from ansible.module_utils.basic import AnsibleModule from ansible.module_utils.basic import AnsibleModule
from ansible.module_utils.ec2 import camel_dict_to_snake_dict, ec2_argument_spec, get_aws_connection_info, boto3_conn, sort_json_policy_dict from ansible.module_utils.ec2 import camel_dict_to_snake_dict, ec2_argument_spec, get_aws_connection_info, boto3_conn, sort_json_policy_dict
from ansible.module_utils.ec2 import HAS_BOTO3
import json import json
import traceback import traceback
try: try:
import boto3
from botocore.exceptions import ClientError, NoCredentialsError from botocore.exceptions import ClientError, NoCredentialsError
HAS_BOTO3 = True
except ImportError: except ImportError:
HAS_BOTO3 = False pass # caught by imported HAS_BOTO3
def compare_assume_role_policy_doc(current_policy_doc, new_policy_doc): def compare_assume_role_policy_doc(current_policy_doc, new_policy_doc):
@ -180,6 +181,22 @@ def compare_attached_role_policies(current_attached_policies, new_attached_polic
return False return False
def convert_friendly_names_to_arns(connection, module, policy_names):
if not any([not policy.startswith('arn:') for policy in policy_names]):
return policy_names
allpolicies = {}
paginator = connection.get_paginator('list_policies')
policies = paginator.paginate().build_full_result()['Policies']
for policy in policies:
allpolicies[policy['PolicyName']] = policy['Arn']
allpolicies[policy['Arn']] = policy['Arn']
try:
return [allpolicies[policy] for policy in policy_names]
except KeyError as e:
module.fail_json(msg="Couldn't find policy: " + str(e))
def create_or_update_role(connection, module): def create_or_update_role(connection, module):
params = dict() params = dict()
@ -187,6 +204,8 @@ def create_or_update_role(connection, module):
params['RoleName'] = module.params.get('name') params['RoleName'] = module.params.get('name')
params['AssumeRolePolicyDocument'] = module.params.get('assume_role_policy_document') params['AssumeRolePolicyDocument'] = module.params.get('assume_role_policy_document')
managed_policies = module.params.get('managed_policy') managed_policies = module.params.get('managed_policy')
if managed_policies:
managed_policies = convert_friendly_names_to_arns(connection, module, managed_policies)
changed = False changed = False
# Get role # Get role
@ -305,7 +324,6 @@ def destroy_role(connection, module):
def get_role(connection, module, name): def get_role(connection, module, name):
try: try:
return connection.get_role(RoleName=name)['Role'] return connection.get_role(RoleName=name)['Role']
except ClientError as e: except ClientError as e:
@ -334,18 +352,15 @@ def main():
argument_spec.update( argument_spec.update(
dict( dict(
name=dict(required=True, type='str'), name=dict(required=True, type='str'),
path=dict(default="/", required=False, type='str'), path=dict(default="/", type='str'),
assume_role_policy_document=dict(required=False, type='json'), assume_role_policy_document=dict(type='json'),
managed_policy=dict(default=None, required=False, type='list'), managed_policy=dict(type='list', aliases=['managed_policies']),
state=dict(default=None, choices=['present', 'absent'], required=True) state=dict(choices=['present', 'absent'], required=True)
) )
) )
module = AnsibleModule(argument_spec=argument_spec, module = AnsibleModule(argument_spec=argument_spec,
required_if=[ required_if=[('state', 'present', ['assume_role_policy_document'])])
('state', 'present', ['assume_role_policy_document'])
]
)
if not HAS_BOTO3: if not HAS_BOTO3:
module.fail_json(msg='boto3 required for this module') module.fail_json(msg='boto3 required for this module')

Loading…
Cancel
Save