Minor tidyup (#2828)
This commit is contained in:
John R Barker
Matt Clay
parent
7e3af74fae
commit
b7afa14f64
1 changed files with 24 additions and 9 deletions
|
|
@ -51,12 +51,12 @@ options:
|
|||
choices: [ 'present', 'absent' ]
|
||||
wait:
|
||||
description:
|
||||
- Wait for operation to complete before returning
|
||||
- Wait for operation to complete before returning.
|
||||
required: false
|
||||
default: true
|
||||
wait_timeout:
|
||||
description:
|
||||
- How many seconds to wait for an operation to complete before timing out
|
||||
- How many seconds to wait for an operation to complete before timing out.
|
||||
required: false
|
||||
default: 300
|
||||
tags:
|
||||
|
|
@ -158,6 +158,7 @@ import datetime
|
|||
import time
|
||||
from functools import reduce
|
||||
|
||||
|
||||
def convert_to_lower(data):
|
||||
"""Convert all uppercase keys in dict with lowercase_
|
||||
Args:
|
||||
|
|
@ -195,6 +196,7 @@ def convert_to_lower(data):
|
|||
results[key] = val
|
||||
return results
|
||||
|
||||
|
||||
def make_tags_in_proper_format(tags):
|
||||
"""Take a dictionary of tags and convert them into the AWS Tags format.
|
||||
Args:
|
||||
|
|
@ -216,6 +218,7 @@ def make_tags_in_proper_format(tags):
|
|||
|
||||
return formatted_tags
|
||||
|
||||
|
||||
def make_tags_in_aws_format(tags):
|
||||
"""Take a dictionary of tags and convert them into the AWS Tags format.
|
||||
Args:
|
||||
|
|
@ -247,6 +250,7 @@ def make_tags_in_aws_format(tags):
|
|||
|
||||
return formatted_tags
|
||||
|
||||
|
||||
def get_tags(client, stream_name, check_mode=False):
|
||||
"""Retrieve the tags for a Kinesis Stream.
|
||||
Args:
|
||||
|
|
@ -289,6 +293,7 @@ def get_tags(client, stream_name, check_mode=False):
|
|||
|
||||
return success, err_msg, results
|
||||
|
||||
|
||||
def find_stream(client, stream_name, check_mode=False):
|
||||
"""Retrieve a Kinesis Stream.
|
||||
Args:
|
||||
|
|
@ -338,6 +343,7 @@ def find_stream(client, stream_name, check_mode=False):
|
|||
|
||||
return success, err_msg, results
|
||||
|
||||
|
||||
def wait_for_status(client, stream_name, status, wait_timeout=300,
|
||||
check_mode=False):
|
||||
"""Wait for the the status to change for a Kinesis Stream.
|
||||
|
|
@ -398,6 +404,7 @@ def wait_for_status(client, stream_name, status, wait_timeout=300,
|
|||
|
||||
return status_achieved, err_msg, stream
|
||||
|
||||
|
||||
def tags_action(client, stream_name, tags, action='create', check_mode=False):
|
||||
"""Create or delete multiple tags from a Kinesis Stream.
|
||||
Args:
|
||||
|
|
@ -451,6 +458,7 @@ def tags_action(client, stream_name, tags, action='create', check_mode=False):
|
|||
|
||||
return success, err_msg
|
||||
|
||||
|
||||
def recreate_tags_from_list(list_of_tags):
|
||||
"""Recreate tags from a list of tuples into the Amazon Tag format.
|
||||
Args:
|
||||
|
|
@ -483,6 +491,7 @@ def recreate_tags_from_list(list_of_tags):
|
|||
)
|
||||
return tags
|
||||
|
||||
|
||||
def update_tags(client, stream_name, tags, check_mode=False):
|
||||
"""Update tags for an amazon resource.
|
||||
Args:
|
||||
|
|
@ -562,6 +571,7 @@ def update_tags(client, stream_name, tags, check_mode=False):
|
|||
|
||||
return success, changed, err_msg
|
||||
|
||||
|
||||
def stream_action(client, stream_name, shard_count=1, action='create',
|
||||
timeout=300, check_mode=False):
|
||||
"""Create or Delete an Amazon Kinesis Stream.
|
||||
|
|
@ -615,6 +625,7 @@ def stream_action(client, stream_name, shard_count=1, action='create',
|
|||
|
||||
return success, err_msg
|
||||
|
||||
|
||||
def retention_action(client, stream_name, retention_period=24,
|
||||
action='increase', check_mode=False):
|
||||
"""Increase or Decreaste the retention of messages in the Kinesis stream.
|
||||
|
|
@ -678,6 +689,7 @@ def retention_action(client, stream_name, retention_period=24,
|
|||
|
||||
return success, err_msg
|
||||
|
||||
|
||||
def update(client, current_stream, stream_name, retention_period=None,
|
||||
tags=None, wait=False, wait_timeout=300, check_mode=False):
|
||||
"""Update an Amazon Kinesis Stream.
|
||||
|
|
@ -805,6 +817,7 @@ def update(client, current_stream, stream_name, retention_period=None,
|
|||
|
||||
return success, changed, err_msg
|
||||
|
||||
|
||||
def create_stream(client, stream_name, number_of_shards=1, retention_period=None,
|
||||
tags=None, wait=False, wait_timeout=300, check_mode=False):
|
||||
"""Create an Amazon Kinesis Stream.
|
||||
|
|
@ -941,6 +954,7 @@ def create_stream(client, stream_name, number_of_shards=1, retention_period=None
|
|||
|
||||
return success, changed, err_msg, results
|
||||
|
||||
|
||||
def delete_stream(client, stream_name, wait=False, wait_timeout=300,
|
||||
check_mode=False):
|
||||
"""Delete an Amazon Kinesis Stream.
|
||||
|
|
@ -1001,17 +1015,18 @@ def delete_stream(client, stream_name, wait=False, wait_timeout=300,
|
|||
|
||||
return success, changed, err_msg, results
|
||||
|
||||
|
||||
def main():
|
||||
argument_spec = ec2_argument_spec()
|
||||
argument_spec.update(
|
||||
dict(
|
||||
name = dict(default=None, required=True),
|
||||
shards = dict(default=None, required=False, type='int'),
|
||||
retention_period = dict(default=None, required=False, type='int'),
|
||||
tags = dict(default=None, required=False, type='dict', aliases=['resource_tags']),
|
||||
wait = dict(default=True, required=False, type='bool'),
|
||||
wait_timeout = dict(default=300, required=False, type='int'),
|
||||
state = dict(default='present', choices=['present', 'absent']),
|
||||
name=dict(default=None, required=True),
|
||||
shards=dict(default=None, required=False, type='int'),
|
||||
retention_period=dict(default=None, required=False, type='int'),
|
||||
tags=dict(default=None, required=False, type='dict', aliases=['resource_tags']),
|
||||
wait=dict(default=True, required=False, type='bool'),
|
||||
wait_timeout=dict(default=300, required=False, type='int'),
|
||||
state=dict(default='present', choices=['present', 'absent']),
|
||||
)
|
||||
)
|
||||
module = AnsibleModule(
|
||||
|
|
|
|||
Loading…
Reference in a new issue