fixup! Set Lambda function log retention (#6070)

scripts/import_cloudwatch_log_groups.py

@@ -1,6 +1,9 @@
 import json
 
 import boto3
+from more_itertools import (
+    one,
+)
 
 from azul import (
     config,
@@ -29,19 +32,21 @@ def add_prefix(name):
             name = add_prefix('chatbot')
             log_group = '/aws/chatbot/' + config.qualified_resource_name('chatbot')
             log_groups[name] = log_group
-    else:
+    elif not tf_component:
         # Lambda function log groups
         tf_path = f'{config.project_root}/terraform/api_gateway.tf.json'
         with open(tf_path, 'r') as f:
             tf_json = json.load(f)
-        for d1 in tf_json['resource']:
-            for k1, v1 in d1.items():
-                if k1 == 'aws_lambda_function':
-                    for d2 in v1:
-                        for k2, v2 in d2.items():
-                            name = add_prefix(k2 + '_lambda')
-                            log_group = '/aws/lambda/' + v2['function_name']
-                            log_groups[name] = log_group
+        for resources in tf_json['resource']:
+            resource_type = one(list(resources.keys()))
+            if resource_type == 'aws_lambda_function':
+                for resource in resources[resource_type]:
+                    for resource_name, resource_def in resource.items():
+                        name = add_prefix(resource_name + '_lambda')
+                        log_group = '/aws/lambda/' + resource_def['function_name']
+                        log_groups[name] = log_group
+    else:
+        pass
 
     log_client = boto3.client('logs')
     paginator = log_client.get_paginator('describe_log_groups')