ecs_attribute: new module. (#20618)
ecs_attribute: fixed docs. ecs_attribute: changed state to 'present' by default. ecs_attribute: pep8 fix. ecs_attribute: improved attribute iteration. ecs_attribute: fixed absent EC2_REGION error message. ecs_attribute: fixed exception handling on invalid attribute (e.g. 2+2). ecs_attribute: pep8 fix. ecs_attribute: docs and python3 fixes. ecs_attribute: fixed class declaration for python 2-3 compatibility.
This commit is contained in:
Andrej Svenke
Sloane Hertel
parent
e2b2ba6bc7
commit
ef55530d67
1 changed files with 311 additions and 0 deletions
311
lib/ansible/modules/cloud/amazon/ecs_attribute.py
Normal file
311
lib/ansible/modules/cloud/amazon/ecs_attribute.py
Normal file
|
|
@ -0,0 +1,311 @@
|
|||
#!/usr/bin/python
|
||||
# This file is part of Ansible
|
||||
#
|
||||
# Ansible is free software: you can redistribute it and/or modify
|
||||
# it under the terms of the GNU General Public License as published by
|
||||
# the Free Software Foundation, either version 3 of the License, or
|
||||
# (at your option) any later version.
|
||||
#
|
||||
# Ansible is distributed in the hope that it will be useful,
|
||||
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||
# GNU General Public License for more details.
|
||||
#
|
||||
# You should have received a copy of the GNU General Public License
|
||||
# along with Ansible. If not, see <http://www.gnu.org/licenses/>.
|
||||
|
||||
ANSIBLE_METADATA = {'status': ['preview'],
|
||||
'supported_by': 'community',
|
||||
'metadata_version': '1.0'}
|
||||
|
||||
DOCUMENTATION = '''
|
||||
---
|
||||
module: ecs_attribute
|
||||
short_description: manage ecs attributes
|
||||
description:
|
||||
- Create, update or delete ECS container instance attributes.
|
||||
version_added: "2.4"
|
||||
author: Andrej Svenke (@anryko)
|
||||
requirements: [ botocore, boto3 ]
|
||||
options:
|
||||
cluster:
|
||||
description:
|
||||
- The short name or full Amazon Resource Name (ARN) of the cluster
|
||||
that contains the resource to apply attributes.
|
||||
required: true
|
||||
state:
|
||||
description:
|
||||
- The desired state of the attributes.
|
||||
required: false
|
||||
default: present
|
||||
choices: ['present', 'absent']
|
||||
attributes:
|
||||
description:
|
||||
- List of attributes.
|
||||
required: true
|
||||
suboptions:
|
||||
name:
|
||||
description:
|
||||
- The name of the attribute. Up to 128 letters (uppercase and lowercase),
|
||||
numbers, hyphens, underscores, and periods are allowed.
|
||||
required: true
|
||||
value:
|
||||
description:
|
||||
- The value of the attribute. Up to 128 letters (uppercase and lowercase),
|
||||
numbers, hyphens, underscores, periods, at signs (@), forward slashes, colons,
|
||||
and spaces are allowed.
|
||||
required: false
|
||||
ec2_instance_id:
|
||||
description:
|
||||
- EC2 instance ID of ECS cluster container instance.
|
||||
required: true
|
||||
extends_documentation_fragment:
|
||||
- aws
|
||||
- ec2
|
||||
'''
|
||||
|
||||
EXAMPLES = '''
|
||||
# Note: These examples do not set authentication details, see the AWS Guide for details.
|
||||
|
||||
# Set attributes
|
||||
- ecs_attribute:
|
||||
state: present
|
||||
cluster: test-cluster
|
||||
ec2_instance_id: "{{ ec2_id }}"
|
||||
attributes:
|
||||
- flavor: test
|
||||
- migrated
|
||||
delegate_to: localhost
|
||||
|
||||
# Delete attributes
|
||||
- ecs_attribute:
|
||||
state: absent
|
||||
cluster: test-cluster
|
||||
ec2_instance_id: "{{ ec2_id }}"
|
||||
attributes:
|
||||
- flavor: test
|
||||
- migrated
|
||||
delegate_to: localhost
|
||||
'''
|
||||
|
||||
RETURN = '''
|
||||
attributes:
|
||||
description: attributes
|
||||
type: complex
|
||||
returned: always
|
||||
contains:
|
||||
cluster:
|
||||
description: cluster name
|
||||
type: string
|
||||
ec2_instance_id:
|
||||
description: ec2 instance id of ecs container instance
|
||||
type: string
|
||||
attributes:
|
||||
description: list of attributes
|
||||
type: list of complex
|
||||
contains:
|
||||
name:
|
||||
description: name of the attribute
|
||||
type: string
|
||||
value:
|
||||
description: value of the attribute
|
||||
returned: if present
|
||||
type: string
|
||||
'''
|
||||
|
||||
try:
|
||||
import boto3
|
||||
from botocore.exceptions import ClientError, EndpointConnectionError
|
||||
HAS_BOTO3 = True
|
||||
except ImportError:
|
||||
HAS_BOTO3 = False
|
||||
|
||||
from ansible.module_utils.basic import AnsibleModule
|
||||
from ansible.module_utils.ec2 import boto3_conn, ec2_argument_spec, get_aws_connection_info
|
||||
|
||||
|
||||
class EcsAttributes(object):
|
||||
"""Handles ECS Cluster Attribute"""
|
||||
|
||||
def __init__(self, module, attributes):
|
||||
self.module = module
|
||||
self.attributes = attributes if self._validate_attrs(attributes) else self._parse_attrs(attributes)
|
||||
|
||||
def __bool__(self):
|
||||
return bool(self.attributes)
|
||||
|
||||
__nonzero__ = __bool__
|
||||
|
||||
def __iter__(self):
|
||||
return iter(self.attributes)
|
||||
|
||||
@staticmethod
|
||||
def _validate_attrs(attrs):
|
||||
return all(tuple(attr.keys()) in (('name', 'value'), ('value', 'name')) for attr in attrs)
|
||||
|
||||
def _parse_attrs(self, attrs):
|
||||
attrs_parsed = []
|
||||
for attr in attrs:
|
||||
if isinstance(attr, dict):
|
||||
if len(attr) != 1:
|
||||
self.module.fail_json(msg="Incorrect attribute format - %s" % str(attr))
|
||||
name, value = list(attr.items())[0]
|
||||
attrs_parsed.append({'name': name, 'value': value})
|
||||
elif isinstance(attr, str):
|
||||
attrs_parsed.append({'name': attr, 'value': None})
|
||||
else:
|
||||
self.module.fail_json(msg="Incorrect attributes format - %s" % str(attrs))
|
||||
|
||||
return attrs_parsed
|
||||
|
||||
def _setup_attr_obj(self, ecs_arn, name, value=None, skip_value=False):
|
||||
attr_obj = {'targetType': 'container-instance',
|
||||
'targetId': ecs_arn,
|
||||
'name': name}
|
||||
if not skip_value and value is not None:
|
||||
attr_obj['value'] = value
|
||||
|
||||
return attr_obj
|
||||
|
||||
def get_for_ecs_arn(self, ecs_arn, skip_value=False):
|
||||
"""
|
||||
Returns list of attribute dicts ready to be passed to boto3
|
||||
attributes put/delete methods.
|
||||
"""
|
||||
return [self._setup_attr_obj(ecs_arn, skip_value=skip_value, **attr) for attr in self.attributes]
|
||||
|
||||
def diff(self, attrs):
|
||||
"""
|
||||
Returns EcsAttributes Object containing attributes which are present
|
||||
in self but are absent in passed attrs (EcsAttributes Object).
|
||||
"""
|
||||
attrs_diff = [attr for attr in self.attributes if attr not in attrs]
|
||||
return EcsAttributes(self.module, attrs_diff)
|
||||
|
||||
|
||||
class Ec2EcsInstance(object):
|
||||
"""Handle ECS Cluster Remote Operations"""
|
||||
|
||||
def __init__(self, module, cluster, ec2_id):
|
||||
self.module = module
|
||||
self.cluster = cluster
|
||||
self.ec2_id = ec2_id
|
||||
|
||||
region, ec2_url, aws_connect_kwargs = get_aws_connection_info(module, boto3=True)
|
||||
if not region:
|
||||
module.fail_json(msg=("Region must be specified as a parameter,"
|
||||
" in EC2_REGION or AWS_REGION environment"
|
||||
" variables or in boto configuration file"))
|
||||
self.ecs = boto3_conn(module, conn_type='client', resource='ecs',
|
||||
region=region, endpoint=ec2_url, **aws_connect_kwargs)
|
||||
|
||||
self.ecs_arn = self._get_ecs_arn()
|
||||
|
||||
def _get_ecs_arn(self):
|
||||
try:
|
||||
ecs_instances_arns = self.ecs.list_container_instances(cluster=self.cluster)['containerInstanceArns']
|
||||
ec2_instances = self.ecs.describe_container_instances(cluster=self.cluster,
|
||||
containerInstances=ecs_instances_arns)['containerInstances']
|
||||
except (ClientError, EndpointConnectionError) as e:
|
||||
self.module.fail_json(msg="Can't connect to the cluster - %s" % str(e))
|
||||
|
||||
try:
|
||||
ecs_arn = next(inst for inst in ec2_instances
|
||||
if inst['ec2InstanceId'] == self.ec2_id)['containerInstanceArn']
|
||||
except StopIteration:
|
||||
self.module.fail_json(msg="EC2 instance Id not found in ECS cluster - %s" % str(self.cluster))
|
||||
|
||||
return ecs_arn
|
||||
|
||||
def attrs_put(self, attrs):
|
||||
"""Puts attributes on ECS container instance"""
|
||||
try:
|
||||
self.ecs.put_attributes(cluster=self.cluster,
|
||||
attributes=attrs.get_for_ecs_arn(self.ecs_arn))
|
||||
except ClientError as e:
|
||||
self.module.fail_json(msg=str(e))
|
||||
|
||||
def attrs_delete(self, attrs):
|
||||
"""Deletes attributes from ECS container instance."""
|
||||
try:
|
||||
self.ecs.delete_attributes(cluster=self.cluster,
|
||||
attributes=attrs.get_for_ecs_arn(self.ecs_arn, skip_value=True))
|
||||
except ClientError as e:
|
||||
self.module.fail_json(msg=str(e))
|
||||
|
||||
def attrs_get_by_name(self, attrs):
|
||||
"""
|
||||
Returns EcsAttributes object containing attributes from ECS container instance with names
|
||||
matching to attrs.attributes (EcsAttributes Object).
|
||||
"""
|
||||
attr_objs = [{'targetType': 'container-instance', 'attributeName': attr['name']}
|
||||
for attr in attrs]
|
||||
|
||||
try:
|
||||
matched_ecs_targets = [attr_found for attr_obj in attr_objs
|
||||
for attr_found in self.ecs.list_attributes(cluster=self.cluster, **attr_obj)['attributes']]
|
||||
except ClientError as e:
|
||||
self.module.fail_json(msg="Can't connect to the cluster - %s" % str(e))
|
||||
|
||||
matched_objs = [target for target in matched_ecs_targets
|
||||
if target['targetId'] == self.ecs_arn]
|
||||
|
||||
results = [{'name': match['name'], 'value': match.get('value', None)}
|
||||
for match in matched_objs]
|
||||
|
||||
return EcsAttributes(self.module, results)
|
||||
|
||||
|
||||
def main():
|
||||
argument_spec = ec2_argument_spec()
|
||||
argument_spec.update(dict(
|
||||
state=dict(required=False, default='present', choices=['present', 'absent']),
|
||||
cluster=dict(required=True, type='str'),
|
||||
ec2_instance_id=dict(required=True, type='str'),
|
||||
attributes=dict(required=True, type='list'),
|
||||
))
|
||||
|
||||
required_together = (['cluster', 'ec2_instance_id', 'attributes'])
|
||||
|
||||
module = AnsibleModule(argument_spec=argument_spec, supports_check_mode=True,
|
||||
required_together=required_together)
|
||||
|
||||
if not HAS_BOTO3:
|
||||
module.fail_json(msg='boto3 is required.')
|
||||
|
||||
cluster = module.params['cluster']
|
||||
ec2_instance_id = module.params['ec2_instance_id']
|
||||
attributes = module.params['attributes']
|
||||
|
||||
conti = Ec2EcsInstance(module, cluster, ec2_instance_id)
|
||||
attrs = EcsAttributes(module, attributes)
|
||||
|
||||
results = {'changed': False,
|
||||
'attributes': [
|
||||
{'cluster': cluster,
|
||||
'ec2_instance_id': ec2_instance_id,
|
||||
'attributes': attributes}
|
||||
]}
|
||||
|
||||
attrs_present = conti.attrs_get_by_name(attrs)
|
||||
|
||||
if module.params['state'] == 'present':
|
||||
attrs_diff = attrs.diff(attrs_present)
|
||||
if not attrs_diff:
|
||||
module.exit_json(**results)
|
||||
|
||||
conti.attrs_put(attrs_diff)
|
||||
results['changed'] = True
|
||||
|
||||
elif module.params['state'] == 'absent':
|
||||
if not attrs_present:
|
||||
module.exit_json(**results)
|
||||
|
||||
conti.attrs_delete(attrs_present)
|
||||
results['changed'] = True
|
||||
|
||||
module.exit_json(**results)
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
main()
|
||||
Loading…
Reference in a new issue