Yasuke/moto
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

move dynamodb responses from camelcase

Browse Source
This commit is contained in:
Steve Pulec 2013-03-16 00:33:57 -04:00
parent 53dc766dad
commit 893b446890
1 changed files with 14 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

27
moto/dynamodb/responses.py
View File

@ -1,6 +1,6 @@
import json import json
from moto.core.utils import headers_to_dict from moto.core.utils import headers_to_dict, camelcase_to_underscores
from .models import dynamodb_backend, dynamo_json_dump from .models import dynamodb_backend, dynamo_json_dump
@ -54,11 +54,12 @@ class DynamoHandler(object):
def dispatch(self): def dispatch(self):
method = self.get_method_name(self.headers) method = self.get_method_name(self.headers)
if method: if method:
method = camelcase_to_underscores(method)
return getattr(self, method)(self.uri, self.body, self.headers) return getattr(self, method)(self.uri, self.body, self.headers)
else: else:
return "", dict(status=404) return "", dict(status=404)
def ListTables(self, uri, body, headers): def list_tables(self, uri, body, headers):
limit = body.get('Limit') limit = body.get('Limit')
if body.get("ExclusiveStartTableName"): if body.get("ExclusiveStartTableName"):
last = body.get("ExclusiveStartTableName") last = body.get("ExclusiveStartTableName")
@ -75,7 +76,7 @@ class DynamoHandler(object):
response["LastEvaluatedTableName"] = tables[-1] response["LastEvaluatedTableName"] = tables[-1]
return dynamo_json_dump(response) return dynamo_json_dump(response)
def CreateTable(self, uri, body, headers): def create_table(self, uri, body, headers):
name = body['TableName'] name = body['TableName']
key_schema = body['KeySchema'] key_schema = body['KeySchema']
@ -102,7 +103,7 @@ class DynamoHandler(object):
) )
return dynamo_json_dump(table.describe) return dynamo_json_dump(table.describe)
def DeleteTable(self, uri, body, headers): def delete_table(self, uri, body, headers):
name = body['TableName'] name = body['TableName']
table = dynamodb_backend.delete_table(name) table = dynamodb_backend.delete_table(name)
if table: if table:
@ -111,7 +112,7 @@ class DynamoHandler(object):
er = 'com.amazonaws.dynamodb.v20111205#ResourceNotFoundException' er = 'com.amazonaws.dynamodb.v20111205#ResourceNotFoundException'
return self.error(er) return self.error(er)
def UpdateTable(self, uri, body, headers): def update_table(self, uri, body, headers):
name = body['TableName'] name = body['TableName']
throughput = body["ProvisionedThroughput"] throughput = body["ProvisionedThroughput"]
new_read_units = throughput["ReadCapacityUnits"] new_read_units = throughput["ReadCapacityUnits"]
@ -119,7 +120,7 @@ class DynamoHandler(object):
table = dynamodb_backend.update_table_throughput(name, new_read_units, new_write_units) table = dynamodb_backend.update_table_throughput(name, new_read_units, new_write_units)
return dynamo_json_dump(table.describe) return dynamo_json_dump(table.describe)
def DescribeTable(self, uri, body, headers): def describe_table(self, uri, body, headers):
name = body['TableName'] name = body['TableName']
try: try:
table = dynamodb_backend.tables[name] table = dynamodb_backend.tables[name]
@ -128,7 +129,7 @@ class DynamoHandler(object):
return self.error(er) return self.error(er)
return dynamo_json_dump(table.describe) return dynamo_json_dump(table.describe)
def PutItem(self, uri, body, headers): def put_item(self, uri, body, headers):
name = body['TableName'] name = body['TableName']
item = body['Item'] item = body['Item']
result = dynamodb_backend.put_item(name, item) result = dynamodb_backend.put_item(name, item)
@ -140,7 +141,7 @@ class DynamoHandler(object):
er = 'com.amazonaws.dynamodb.v20111205#ResourceNotFoundException' er = 'com.amazonaws.dynamodb.v20111205#ResourceNotFoundException'
return self.error(er) return self.error(er)
def BatchWriteItem(self, uri, body, headers): def batch_write_item(self, uri, body, headers):
table_batches = body['RequestItems'] table_batches = body['RequestItems']
for table_name, table_requests in table_batches.iteritems(): for table_name, table_requests in table_batches.iteritems():
@ -171,7 +172,7 @@ class DynamoHandler(object):
return dynamo_json_dump(response) return dynamo_json_dump(response)
def GetItem(self, uri, body, headers): def get_item(self, uri, body, headers):
name = body['TableName'] name = body['TableName']
key = body['Key'] key = body['Key']
hash_key = key['HashKeyElement'] hash_key = key['HashKeyElement']
@ -186,7 +187,7 @@ class DynamoHandler(object):
er = 'com.amazonaws.dynamodb.v20111205#ResourceNotFoundException' er = 'com.amazonaws.dynamodb.v20111205#ResourceNotFoundException'
return self.error(er) return self.error(er)
def BatchGetItem(self, uri, body, headers): def batch_get_item(self, uri, body, headers):
table_batches = body['RequestItems'] table_batches = body['RequestItems']
results = { results = {
@ -209,7 +210,7 @@ class DynamoHandler(object):
results["Responses"][table_name] = {"Items": items, "ConsumedCapacityUnits": 1} results["Responses"][table_name] = {"Items": items, "ConsumedCapacityUnits": 1}
return dynamo_json_dump(results) return dynamo_json_dump(results)
def Query(self, uri, body, headers): def query(self, uri, body, headers):
name = body['TableName'] name = body['TableName']
hash_key = body['HashKeyValue'] hash_key = body['HashKeyValue']
range_condition = body.get('RangeKeyCondition') range_condition = body.get('RangeKeyCondition')
@ -240,7 +241,7 @@ class DynamoHandler(object):
# } # }
return dynamo_json_dump(result) return dynamo_json_dump(result)
def Scan(self, uri, body, headers): def scan(self, uri, body, headers):
name = body['TableName'] name = body['TableName']
filters = {} filters = {}
@ -272,7 +273,7 @@ class DynamoHandler(object):
# } # }
return dynamo_json_dump(result) return dynamo_json_dump(result)
def DeleteItem(self, uri, body, headers): def delete_item(self, uri, body, headers):
name = body['TableName'] name = body['TableName']
key = body['Key'] key = body['Key']
hash_key = key['HashKeyElement'] hash_key = key['HashKeyElement']