TaggingService: Allow key value field names to be overridden

viren-nadkarni · viren-nadkarni · commit 748bb1c08b59 · 2025-02-26T15:24:31.000+05:30
diff --git a/localstack-core/localstack/utils/tagging.py b/localstack-core/localstack/utils/tagging.py
@@ -2,15 +2,23 @@
 
 
 class TaggingService:
-    def __init__(self):
+    def __init__(self, key_field: str = None, value_field: str = None):
+        """
+        :param key_field: the field name representing the tag key as used by botocore specs
+        :param value_field: the field name representing the tag value as used by botocore specs
+        """
+        self.key_field = key_field or "Key"
+        self.value_field = value_field or "Value"
+
         self.tags = {}
 
     def list_tags_for_resource(self, arn: str, root_name: Optional[str] = None):
         root_name = root_name or "Tags"
+
         result = []
         if arn in self.tags:
             for k, v in self.tags[arn].items():
-                result.append({"Key": k, "Value": v})
+                result.append({self.key_field: k, self.value_field: v})
         return {root_name: result}
 
     def tag_resource(self, arn: str, tags: List[Dict[str, str]]):
@@ -19,7 +27,7 @@ def tag_resource(self, arn: str, tags: List[Dict[str, str]]):
         if arn not in self.tags:
             self.tags[arn] = {}
         for t in tags:
-            self.tags[arn][t["Key"]] = t["Value"]
+            self.tags[arn][t[self.key_field]] = t[self.value_field]
 
     def untag_resource(self, arn: str, tag_names: List[str]):
         tags = self.tags.get(arn, {})
