Merge pull request #86 from lejmr/patch-1

Workaround for situations when name_key is not defined
mantl · Apr 6, 2018 · c9276d7 · c9276d7
2 parents 720ae3d + 933e917
commit c9276d7
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/src/ati/terraform.py b/src/ati/terraform.py
@@ -540,7 +540,10 @@ def aws_host(resource, module_name, **kwargs):
     except KeyError:
         ssh_host_key =  'public_ip'
 
-    name = resource['primary']['attributes'][name_key]
+    try:
+      name = resource['primary']['attributes'][name_key]
+    except:
+      name = resource['primary']['attributes'][ssh_host_key]
     raw_attrs = resource['primary']['attributes']
 
     groups = []