Fetch Lambda functions (part of #204)

nccgroup · Nov 6, 2017 · ed001c6 · ed001c6
1 parent 5949350
commit ed001c6
Show file tree

Hide file tree

Showing 4 changed files with 45 additions and 0 deletions.
diff --git a/AWSScout2/configs/data/metadata.json b/AWSScout2/configs/data/metadata.json
@@ -240,6 +240,16 @@
                     "path": "services.elbv2.ssl_policies"
                 }
             }
+        },
+        "lambda": {
+            "resources": {
+                "functions": {
+                    "api_call": "list_functions",
+                    "response": "Functions",
+                    "hidden": true,
+                    "path": "services.lambda.regions.id.functions"
+                }
+            }
         }
     },
     "security": {

diff --git a/AWSScout2/configs/regions.py b/AWSScout2/configs/regions.py
@@ -277,6 +277,7 @@ def store_target(self, global_params, region, target):
         target_dict[target_id] = target
 
 id_map = {
+    'functions': 'FunctionName',
     'peering_connections': 'VpcPeeringConnectionId',
     'subnet_groups': 'DBSubnetGroupName'
 }

diff --git a/AWSScout2/configs/services.py b/AWSScout2/configs/services.py
@@ -14,6 +14,7 @@
 from AWSScout2.services.elbv2 import ELBv2Config
 from AWSScout2.services.emr import EMRConfig
 from AWSScout2.services.iam import IAMConfig
+from AWSScout2.services.awslambda import LambdaConfig
 from AWSScout2.services.rds import RDSConfig
 from AWSScout2.services.redshift import RedshiftConfig
 from AWSScout2.services.route53 import Route53Config, Route53DomainsConfig
@@ -53,6 +54,7 @@ def __init__(self, metadata, thread_config = 4):
         self.elbv2 = ELBv2Config(metadata['compute']['elbv2'], thread_config)
         self.emr = EMRConfig(metadata['analytics']['emr'], thread_config)
         self.iam = IAMConfig(thread_config)
+        self.awslambda = LambdaConfig(metadata['compute']['lambda'], thread_config)
         self.redshift = RedshiftConfig(metadata['database']['redshift'], thread_config)
         self.rds = RDSConfig(metadata['database']['rds'], thread_config)
         self.route53 = Route53Config(thread_config)

diff --git a/AWSScout2/services/awslambda.py b/AWSScout2/services/awslambda.py
@@ -0,0 +1,32 @@
+# -*- coding: utf-8 -*-
+"""
+Lambda-related classes and functions
+"""
+
+from AWSScout2.configs.regions import RegionalServiceConfig, RegionConfig
+
+
+
+########################################
+# LambdaRegionConfig
+########################################
+
+class LambdaRegionConfig(RegionConfig):
+
+    pass
+
+
+
+########################################
+# LambdaConfig
+########################################
+
+class LambdaConfig(RegionalServiceConfig):
+    """
+    Lambda configuration for all AWS regions
+    """
+
+    region_config_class = LambdaRegionConfig
+
+    def __init__(self, service_metadata, thread_config = 4):
+        super(LambdaConfig, self).__init__(service_metadata, thread_config)