@@ -1649,40 +1649,40 @@ def deploy_lambda_function(account_id: str, rule_name: str, role_arn: str, regio
16491649 LOGGER .info (f"{ rule_name } already exists in { account_id } . Search result: { lambda_function_search } " )
16501650 lambda_arn = lambda_function_search ["Configuration" ]["FunctionArn" ]
16511651
1652- # Lambda state table record
1653- # TODO(liamschn): move dynamodb resource to the dynamo class object/module
1654- dynamodb_resource = sts .assume_role_resource (ssm_params .SRA_SECURITY_ACCT , sts .CONFIGURATION_ROLE , "dynamodb" , sts .HOME_REGION )
1652+ # Lambda state table record
1653+ # TODO(liamschn): move dynamodb resource to the dynamo class object/module
1654+ dynamodb_resource = sts .assume_role_resource (ssm_params .SRA_SECURITY_ACCT , sts .CONFIGURATION_ROLE , "dynamodb" , sts .HOME_REGION )
16551655
1656- item_found , find_result = dynamodb .find_item (
1657- STATE_TABLE ,
1658- dynamodb_resource ,
1659- SOLUTION_NAME ,
1660- {
1661- "arn" : lambda_arn ,
1662- },
1663- )
1664- if item_found is False :
1665- lambda_record_id , lambda_date_time = dynamodb .insert_item (STATE_TABLE , dynamodb_resource , SOLUTION_NAME )
1666- else :
1667- lambda_record_id = find_result ["record_id" ]
1656+ item_found , find_result = dynamodb .find_item (
1657+ STATE_TABLE ,
1658+ dynamodb_resource ,
1659+ SOLUTION_NAME ,
1660+ {
1661+ "arn" : lambda_arn ,
1662+ },
1663+ )
1664+ if item_found is False :
1665+ lambda_record_id , lambda_date_time = dynamodb .insert_item (STATE_TABLE , dynamodb_resource , SOLUTION_NAME )
1666+ else :
1667+ lambda_record_id = find_result ["record_id" ]
16681668
1669- dynamodb .update_item (
1670- STATE_TABLE ,
1671- dynamodb_resource ,
1672- SOLUTION_NAME ,
1673- lambda_record_id ,
1674- {
1675- "aws_service" : "lambda" ,
1676- "component_state" : "implemented" ,
1677- "account" : account_id ,
1678- "description" : "lambda for config rule" ,
1679- "component_region" : region ,
1680- "component_type" : "lambda" ,
1681- "component_name" : rule_name ,
1682- "arn" : lambda_arn ,
1683- "date_time" : dynamodb .get_date_time (),
1684- },
1685- )
1669+ dynamodb .update_item (
1670+ STATE_TABLE ,
1671+ dynamodb_resource ,
1672+ SOLUTION_NAME ,
1673+ lambda_record_id ,
1674+ {
1675+ "aws_service" : "lambda" ,
1676+ "component_state" : "implemented" ,
1677+ "account" : account_id ,
1678+ "description" : "lambda for config rule" ,
1679+ "component_region" : region ,
1680+ "component_type" : "lambda" ,
1681+ "component_name" : rule_name ,
1682+ "arn" : lambda_arn ,
1683+ "date_time" : dynamodb .get_date_time (),
1684+ },
1685+ )
16861686
16871687
16881688 return lambda_arn
0 commit comments