Skip to content

Commit ebac544

Browse files
committed
fixing lambda state record
1 parent f439e42 commit ebac544

File tree

1 file changed

+32
-32
lines changed
  • aws_sra_examples/solutions/genai/bedrock_org/lambda/src

1 file changed

+32
-32
lines changed

aws_sra_examples/solutions/genai/bedrock_org/lambda/src/app.py

Lines changed: 32 additions & 32 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)