fix DynamoDB TableARN which is hardcoded as "ddblocal" in DynamoDBLocal

whummer · whummer · commit 0cc2ccdf8802 · 2017-10-10T00:07:52.000-04:00
diff --git a/localstack/services/dynamodb/dynamodb_listener.py b/localstack/services/dynamodb/dynamodb_listener.py
@@ -1,3 +1,4 @@
+import re
 import json
 import random
 import logging
@@ -34,6 +35,14 @@ def return_response(self, method, path, data, headers, response):
         if data and 'TableName' in data and 'KeySchema' in data:
             TABLE_DEFINITIONS[data['TableName']] = data
 
+        if response._content:
+            # fix the table ARN (DynamoDBLocal hardcodes "ddblocal" as the region)
+            content_replaced = re.sub(r'"TableArn"\s*:\s*"arn:aws:dynamodb:ddblocal:([^"]+)"',
+                r'"TableArn": "arn:aws:dynamodb:%s:\1"' % aws_stack.get_local_region(), to_str(response._content))
+            if content_replaced != response._content:
+                response._content = content_replaced
+                fix_headers_for_updated_response(response)
+
         action = headers.get('X-Amz-Target')
         if not action:
             return
@@ -98,8 +107,7 @@ def return_response(self, method, path, data, headers, response):
                         'TableName': data['TableName']
                     }
                     response._content = json.dumps(content)
-                    response.headers['content-length'] = len(response.content)
-                    response.headers['x-amz-crc32'] = calculate_crc32(response)
+                    fix_headers_for_updated_response(response)
         elif action == '%s.DeleteItem' % ACTION_PREFIX:
             record['eventName'] = 'REMOVE'
             record['dynamodb']['Keys'] = data['Key']
@@ -131,6 +139,11 @@ def return_response(self, method, path, data, headers, response):
 UPDATE_DYNAMODB = ProxyListenerDynamoDB()
 
 
+def fix_headers_for_updated_response(response):
+    response.headers['content-length'] = len(response.content)
+    response.headers['x-amz-crc32'] = calculate_crc32(response)
+
+
 def calculate_crc32(response):
     return crc32(to_bytes(response.content)) & 0xffffffff
 
