Run black on moto & test directories.
This commit is contained in:
Asher Foa
parent
c820395dbf
commit
96e5b1993d
507 changed files with 52541 additions and 47814 deletions
|
|
@ -8,7 +8,6 @@ from .models import dynamodb_backend, dynamo_json_dump
|
|||
|
||||
|
||||
class DynamoHandler(BaseResponse):
|
||||
|
||||
def get_endpoint_name(self, headers):
|
||||
"""Parses request headers and extracts part od the X-Amz-Target
|
||||
that corresponds to a method of DynamoHandler
|
||||
|
|
@ -16,15 +15,15 @@ class DynamoHandler(BaseResponse):
|
|||
ie: X-Amz-Target: DynamoDB_20111205.ListTables -> ListTables
|
||||
"""
|
||||
# Headers are case-insensitive. Probably a better way to do this.
|
||||
match = headers.get('x-amz-target') or headers.get('X-Amz-Target')
|
||||
match = headers.get("x-amz-target") or headers.get("X-Amz-Target")
|
||||
if match:
|
||||
return match.split(".")[1]
|
||||
|
||||
def error(self, type_, status=400):
|
||||
return status, self.response_headers, dynamo_json_dump({'__type': type_})
|
||||
return status, self.response_headers, dynamo_json_dump({"__type": type_})
|
||||
|
||||
def call_action(self):
|
||||
self.body = json.loads(self.body or '{}')
|
||||
self.body = json.loads(self.body or "{}")
|
||||
endpoint = self.get_endpoint_name(self.headers)
|
||||
if endpoint:
|
||||
endpoint = camelcase_to_underscores(endpoint)
|
||||
|
|
@ -41,7 +40,7 @@ class DynamoHandler(BaseResponse):
|
|||
|
||||
def list_tables(self):
|
||||
body = self.body
|
||||
limit = body.get('Limit')
|
||||
limit = body.get("Limit")
|
||||
if body.get("ExclusiveStartTableName"):
|
||||
last = body.get("ExclusiveStartTableName")
|
||||
start = list(dynamodb_backend.tables.keys()).index(last) + 1
|
||||
|
|
@ -49,7 +48,7 @@ class DynamoHandler(BaseResponse):
|
|||
start = 0
|
||||
all_tables = list(dynamodb_backend.tables.keys())
|
||||
if limit:
|
||||
tables = all_tables[start:start + limit]
|
||||
tables = all_tables[start : start + limit]
|
||||
else:
|
||||
tables = all_tables[start:]
|
||||
response = {"TableNames": tables}
|
||||
|
|
@ -59,16 +58,16 @@ class DynamoHandler(BaseResponse):
|
|||
|
||||
def create_table(self):
|
||||
body = self.body
|
||||
name = body['TableName']
|
||||
name = body["TableName"]
|
||||
|
||||
key_schema = body['KeySchema']
|
||||
hash_key = key_schema['HashKeyElement']
|
||||
hash_key_attr = hash_key['AttributeName']
|
||||
hash_key_type = hash_key['AttributeType']
|
||||
key_schema = body["KeySchema"]
|
||||
hash_key = key_schema["HashKeyElement"]
|
||||
hash_key_attr = hash_key["AttributeName"]
|
||||
hash_key_type = hash_key["AttributeType"]
|
||||
|
||||
range_key = key_schema.get('RangeKeyElement', {})
|
||||
range_key_attr = range_key.get('AttributeName')
|
||||
range_key_type = range_key.get('AttributeType')
|
||||
range_key = key_schema.get("RangeKeyElement", {})
|
||||
range_key_attr = range_key.get("AttributeName")
|
||||
range_key_type = range_key.get("AttributeType")
|
||||
|
||||
throughput = body["ProvisionedThroughput"]
|
||||
read_units = throughput["ReadCapacityUnits"]
|
||||
|
|
@ -86,137 +85,131 @@ class DynamoHandler(BaseResponse):
|
|||
return dynamo_json_dump(table.describe)
|
||||
|
||||
def delete_table(self):
|
||||
name = self.body['TableName']
|
||||
name = self.body["TableName"]
|
||||
table = dynamodb_backend.delete_table(name)
|
||||
if table:
|
||||
return dynamo_json_dump(table.describe)
|
||||
else:
|
||||
er = 'com.amazonaws.dynamodb.v20111205#ResourceNotFoundException'
|
||||
er = "com.amazonaws.dynamodb.v20111205#ResourceNotFoundException"
|
||||
return self.error(er)
|
||||
|
||||
def update_table(self):
|
||||
name = self.body['TableName']
|
||||
name = self.body["TableName"]
|
||||
throughput = self.body["ProvisionedThroughput"]
|
||||
new_read_units = throughput["ReadCapacityUnits"]
|
||||
new_write_units = throughput["WriteCapacityUnits"]
|
||||
table = dynamodb_backend.update_table_throughput(
|
||||
name, new_read_units, new_write_units)
|
||||
name, new_read_units, new_write_units
|
||||
)
|
||||
return dynamo_json_dump(table.describe)
|
||||
|
||||
def describe_table(self):
|
||||
name = self.body['TableName']
|
||||
name = self.body["TableName"]
|
||||
try:
|
||||
table = dynamodb_backend.tables[name]
|
||||
except KeyError:
|
||||
er = 'com.amazonaws.dynamodb.v20111205#ResourceNotFoundException'
|
||||
er = "com.amazonaws.dynamodb.v20111205#ResourceNotFoundException"
|
||||
return self.error(er)
|
||||
return dynamo_json_dump(table.describe)
|
||||
|
||||
def put_item(self):
|
||||
name = self.body['TableName']
|
||||
item = self.body['Item']
|
||||
name = self.body["TableName"]
|
||||
item = self.body["Item"]
|
||||
result = dynamodb_backend.put_item(name, item)
|
||||
if result:
|
||||
item_dict = result.to_json()
|
||||
item_dict['ConsumedCapacityUnits'] = 1
|
||||
item_dict["ConsumedCapacityUnits"] = 1
|
||||
return dynamo_json_dump(item_dict)
|
||||
else:
|
||||
er = 'com.amazonaws.dynamodb.v20111205#ResourceNotFoundException'
|
||||
er = "com.amazonaws.dynamodb.v20111205#ResourceNotFoundException"
|
||||
return self.error(er)
|
||||
|
||||
def batch_write_item(self):
|
||||
table_batches = self.body['RequestItems']
|
||||
table_batches = self.body["RequestItems"]
|
||||
|
||||
for table_name, table_requests in table_batches.items():
|
||||
for table_request in table_requests:
|
||||
request_type = list(table_request)[0]
|
||||
request = list(table_request.values())[0]
|
||||
|
||||
if request_type == 'PutRequest':
|
||||
item = request['Item']
|
||||
if request_type == "PutRequest":
|
||||
item = request["Item"]
|
||||
dynamodb_backend.put_item(table_name, item)
|
||||
elif request_type == 'DeleteRequest':
|
||||
key = request['Key']
|
||||
hash_key = key['HashKeyElement']
|
||||
range_key = key.get('RangeKeyElement')
|
||||
item = dynamodb_backend.delete_item(
|
||||
table_name, hash_key, range_key)
|
||||
elif request_type == "DeleteRequest":
|
||||
key = request["Key"]
|
||||
hash_key = key["HashKeyElement"]
|
||||
range_key = key.get("RangeKeyElement")
|
||||
item = dynamodb_backend.delete_item(table_name, hash_key, range_key)
|
||||
|
||||
response = {
|
||||
"Responses": {
|
||||
"Thread": {
|
||||
"ConsumedCapacityUnits": 1.0
|
||||
},
|
||||
"Reply": {
|
||||
"ConsumedCapacityUnits": 1.0
|
||||
}
|
||||
"Thread": {"ConsumedCapacityUnits": 1.0},
|
||||
"Reply": {"ConsumedCapacityUnits": 1.0},
|
||||
},
|
||||
"UnprocessedItems": {}
|
||||
"UnprocessedItems": {},
|
||||
}
|
||||
|
||||
return dynamo_json_dump(response)
|
||||
|
||||
def get_item(self):
|
||||
name = self.body['TableName']
|
||||
key = self.body['Key']
|
||||
hash_key = key['HashKeyElement']
|
||||
range_key = key.get('RangeKeyElement')
|
||||
attrs_to_get = self.body.get('AttributesToGet')
|
||||
name = self.body["TableName"]
|
||||
key = self.body["Key"]
|
||||
hash_key = key["HashKeyElement"]
|
||||
range_key = key.get("RangeKeyElement")
|
||||
attrs_to_get = self.body.get("AttributesToGet")
|
||||
try:
|
||||
item = dynamodb_backend.get_item(name, hash_key, range_key)
|
||||
except ValueError:
|
||||
er = 'com.amazon.coral.validate#ValidationException'
|
||||
er = "com.amazon.coral.validate#ValidationException"
|
||||
return self.error(er, status=400)
|
||||
if item:
|
||||
item_dict = item.describe_attrs(attrs_to_get)
|
||||
item_dict['ConsumedCapacityUnits'] = 0.5
|
||||
item_dict["ConsumedCapacityUnits"] = 0.5
|
||||
return dynamo_json_dump(item_dict)
|
||||
else:
|
||||
# Item not found
|
||||
er = 'com.amazonaws.dynamodb.v20111205#ResourceNotFoundException'
|
||||
er = "com.amazonaws.dynamodb.v20111205#ResourceNotFoundException"
|
||||
return self.error(er, status=404)
|
||||
|
||||
def batch_get_item(self):
|
||||
table_batches = self.body['RequestItems']
|
||||
table_batches = self.body["RequestItems"]
|
||||
|
||||
results = {
|
||||
"Responses": {
|
||||
"UnprocessedKeys": {}
|
||||
}
|
||||
}
|
||||
results = {"Responses": {"UnprocessedKeys": {}}}
|
||||
|
||||
for table_name, table_request in table_batches.items():
|
||||
items = []
|
||||
keys = table_request['Keys']
|
||||
attributes_to_get = table_request.get('AttributesToGet')
|
||||
keys = table_request["Keys"]
|
||||
attributes_to_get = table_request.get("AttributesToGet")
|
||||
for key in keys:
|
||||
hash_key = key["HashKeyElement"]
|
||||
range_key = key.get("RangeKeyElement")
|
||||
item = dynamodb_backend.get_item(
|
||||
table_name, hash_key, range_key)
|
||||
item = dynamodb_backend.get_item(table_name, hash_key, range_key)
|
||||
if item:
|
||||
item_describe = item.describe_attrs(attributes_to_get)
|
||||
items.append(item_describe)
|
||||
results["Responses"][table_name] = {
|
||||
"Items": items, "ConsumedCapacityUnits": 1}
|
||||
"Items": items,
|
||||
"ConsumedCapacityUnits": 1,
|
||||
}
|
||||
return dynamo_json_dump(results)
|
||||
|
||||
def query(self):
|
||||
name = self.body['TableName']
|
||||
hash_key = self.body['HashKeyValue']
|
||||
range_condition = self.body.get('RangeKeyCondition')
|
||||
name = self.body["TableName"]
|
||||
hash_key = self.body["HashKeyValue"]
|
||||
range_condition = self.body.get("RangeKeyCondition")
|
||||
if range_condition:
|
||||
range_comparison = range_condition['ComparisonOperator']
|
||||
range_values = range_condition['AttributeValueList']
|
||||
range_comparison = range_condition["ComparisonOperator"]
|
||||
range_values = range_condition["AttributeValueList"]
|
||||
else:
|
||||
range_comparison = None
|
||||
range_values = []
|
||||
|
||||
items, last_page = dynamodb_backend.query(
|
||||
name, hash_key, range_comparison, range_values)
|
||||
name, hash_key, range_comparison, range_values
|
||||
)
|
||||
|
||||
if items is None:
|
||||
er = 'com.amazonaws.dynamodb.v20111205#ResourceNotFoundException'
|
||||
er = "com.amazonaws.dynamodb.v20111205#ResourceNotFoundException"
|
||||
return self.error(er)
|
||||
|
||||
result = {
|
||||
|
|
@ -234,10 +227,10 @@ class DynamoHandler(BaseResponse):
|
|||
return dynamo_json_dump(result)
|
||||
|
||||
def scan(self):
|
||||
name = self.body['TableName']
|
||||
name = self.body["TableName"]
|
||||
|
||||
filters = {}
|
||||
scan_filters = self.body.get('ScanFilter', {})
|
||||
scan_filters = self.body.get("ScanFilter", {})
|
||||
for attribute_name, scan_filter in scan_filters.items():
|
||||
# Keys are attribute names. Values are tuples of (comparison,
|
||||
# comparison_value)
|
||||
|
|
@ -248,14 +241,14 @@ class DynamoHandler(BaseResponse):
|
|||
items, scanned_count, last_page = dynamodb_backend.scan(name, filters)
|
||||
|
||||
if items is None:
|
||||
er = 'com.amazonaws.dynamodb.v20111205#ResourceNotFoundException'
|
||||
er = "com.amazonaws.dynamodb.v20111205#ResourceNotFoundException"
|
||||
return self.error(er)
|
||||
|
||||
result = {
|
||||
"Count": len(items),
|
||||
"Items": [item.attrs for item in items if item],
|
||||
"ConsumedCapacityUnits": 1,
|
||||
"ScannedCount": scanned_count
|
||||
"ScannedCount": scanned_count,
|
||||
}
|
||||
|
||||
# Implement this when we do pagination
|
||||
|
|
@ -267,19 +260,19 @@ class DynamoHandler(BaseResponse):
|
|||
return dynamo_json_dump(result)
|
||||
|
||||
def delete_item(self):
|
||||
name = self.body['TableName']
|
||||
key = self.body['Key']
|
||||
hash_key = key['HashKeyElement']
|
||||
range_key = key.get('RangeKeyElement')
|
||||
return_values = self.body.get('ReturnValues', '')
|
||||
name = self.body["TableName"]
|
||||
key = self.body["Key"]
|
||||
hash_key = key["HashKeyElement"]
|
||||
range_key = key.get("RangeKeyElement")
|
||||
return_values = self.body.get("ReturnValues", "")
|
||||
item = dynamodb_backend.delete_item(name, hash_key, range_key)
|
||||
if item:
|
||||
if return_values == 'ALL_OLD':
|
||||
if return_values == "ALL_OLD":
|
||||
item_dict = item.to_json()
|
||||
else:
|
||||
item_dict = {'Attributes': []}
|
||||
item_dict['ConsumedCapacityUnits'] = 0.5
|
||||
item_dict = {"Attributes": []}
|
||||
item_dict["ConsumedCapacityUnits"] = 0.5
|
||||
return dynamo_json_dump(item_dict)
|
||||
else:
|
||||
er = 'com.amazonaws.dynamodb.v20111205#ResourceNotFoundException'
|
||||
er = "com.amazonaws.dynamodb.v20111205#ResourceNotFoundException"
|
||||
return self.error(er)
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue