From f13d376f0cd8eb863d41c6c0ade99d20ea268d11 Mon Sep 17 00:00:00 2001 From: John R Barker Date: Thu, 1 Sep 2016 20:46:08 +0100 Subject: Minor tidyup (#2828) --- cloud/amazon/kinesis_stream.py | 33 ++++++++++++++++++++++++--------- 1 file 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( -- cgit v1.2.1