hotfix: client_args_map set to None caused errors

Author: Erik Steringer

principalmapper/graphing/autoscaling_edges.py

@@ -40,7 +40,10 @@ def return_edges(self, nodes: List[Node], region_allow_list: Optional[List[str]]
 
         logger.info('Generating Edges based on EC2 Auto Scaling.')
 
-        asargs = client_args_map.get('autoscaling', {})
+        if client_args_map is None:
+            asargs = {}
+        else:
+            asargs = client_args_map.get('autoscaling', {})
 
         # Gather projects information for each region
         autoscaling_clients = []

principalmapper/graphing/cloudformation_edges.py

@@ -42,7 +42,10 @@ def return_edges(self, nodes: List[Node], region_allow_list: Optional[List[str]]
 
         logger.info('Pulling data on CloudFormation stacks.')
 
-        cfargs = client_args_map.get('cloudformation', {})
+        if client_args_map is None:
+            cfargs = {}
+        else:
+            cfargs = client_args_map.get('cloudformation', {})
 
         # Grab existing stacks in each region
         cloudformation_clients = []

principalmapper/graphing/codebuild_edges.py

@@ -42,7 +42,10 @@ def return_edges(self, nodes: List[Node], region_allow_list: Optional[List[str]]
 
         # Gather projects information for each region
 
-        cbargs = client_args_map.get('codebuild', {})
+        if client_args_map is None:
+            cbargs = {}
+        else:
+            cbargs = client_args_map.get('codebuild', {})
 
         codebuild_clients = []
         if self.session is not None:

principalmapper/graphing/lambda_edges.py

@@ -42,7 +42,10 @@ def return_edges(self, nodes: List[Node], region_allow_list: Optional[List[str]]
 
         logger.info('Pulling data on Lambda functions')
 
-        lambdaargs = client_args_map.get('lambda', {})
+        if client_args_map is None:
+            lambdaargs = {}
+        else:
+            lambdaargs = client_args_map.get('lambda', {})
 
         lambda_clients = []
         if self.session is not None: