fixes issue where no range_key_name is defined https://github.com/ansible/ansible-modules-extras/issues/841

9 years ago · e5b6d47a54
parent b211bacb40
commit e5b6d47a54
1 changed files with 9 additions and 4 deletions
--- a/cloud/amazon/dynamodb_table.py
+++ b/cloud/amazon/dynamodb_table.py
@ -143,10 +143,15 @@ def create_or_update_dynamo_table(connection, module):
    read_capacity = module.params.get('read_capacity')
    write_capacity = module.params.get('write_capacity')

-    schema = [
-        HashKey(hash_key_name, DYNAMO_TYPE_MAP.get(hash_key_type)),
-        RangeKey(range_key_name, DYNAMO_TYPE_MAP.get(range_key_type))
-    ]
+    if range_key_name:
+        schema = [
+            HashKey(hash_key_name, DYNAMO_TYPE_MAP.get(hash_key_type)),
+            RangeKey(range_key_name, DYNAMO_TYPE_MAP.get(range_key_type))
+        ]
+    else:
+        schema = [
+            HashKey(hash_key_name, DYNAMO_TYPE_MAP.get(hash_key_type))
+        ]
    throughput = {
        'read': read_capacity,
        'write': write_capacity