summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorJohn R Barker <john@johnrbarker.com>2016-09-01 20:46:08 +0100
committerGitHub <noreply@github.com>2016-09-01 20:46:08 +0100
commitf13d376f0cd8eb863d41c6c0ade99d20ea268d11 (patch)
treef5f1e52b19e76e4a29fb3a531c2b8604c56559b9
parente8a5442345b4702816814f86a4264805c1ec7753 (diff)
downloadansible-modules-extras-f13d376f0cd8eb863d41c6c0ade99d20ea268d11.tar.gz
Minor tidyup (#2828)
-rw-r--r--cloud/amazon/kinesis_stream.py33
1 files changed, 24 insertions, 9 deletions
diff --git a/cloud/amazon/kinesis_stream.py b/cloud/amazon/kinesis_stream.py
index 1ba25e69..37f20f8c 100644
--- a/cloud/amazon/kinesis_stream.py
+++ b/cloud/amazon/kinesis_stream.py
@@ -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(