From 3d10ae393d01d9653aed77fbe0d46b0a472f1457 Mon Sep 17 00:00:00 2001 From: Aayush Thapa Date: Thu, 21 Mar 2024 15:57:08 -0700 Subject: [PATCH] add case for future arn generation --- samtranslator/translator/arn_generator.py | 26 +++++++++++++---------- 1 file changed, 15 insertions(+), 11 deletions(-) diff --git a/samtranslator/translator/arn_generator.py b/samtranslator/translator/arn_generator.py index a9f6e0325..d882206c9 100644 --- a/samtranslator/translator/arn_generator.py +++ b/samtranslator/translator/arn_generator.py @@ -18,17 +18,21 @@ def _region_to_partition(region: str) -> str: # setting default partition to aws, this will be overwritten by checking the region below region_string = region.lower() if region_string.startswith("cn-"): - return "aws-cn" - if region_string.startswith("us-iso-"): - return "aws-iso" - if region_string.startswith("us-isob"): - return "aws-iso-b" - if region_string.startswith("us-gov"): - return "aws-us-gov" - if region_string.startswith("eu-isoe"): - return "aws-iso-e" - - return "aws" + partition = "aws-cn" + elif region_string.startswith("us-iso-"): + partition = "aws-iso" + elif region_string.startswith("us-isob"): + partition = "aws-iso-b" + elif region_string.startswith("us-gov"): + partition = "aws-us-gov" + elif region_string.startswith("eu-isoe"): + partition = "aws-iso-e" + elif "iso" in region_string: + partition = "{AWS::Partition}" + else: + partition = "aws" + + return partition class ArnGenerator: