add table delete and update. add item ops, including querying.
This commit is contained in:
Steve Pulec
parent
1f0fd7fac7
commit
621ac79f33
3 changed files with 382 additions and 16 deletions
|
|
@ -1,11 +1,39 @@
|
|||
import datetime
|
||||
|
||||
from collections import OrderedDict
|
||||
from collections import defaultdict, OrderedDict
|
||||
|
||||
from moto.core import BaseBackend
|
||||
from .comparisons import get_comparison_func
|
||||
from .utils import unix_time
|
||||
|
||||
|
||||
class Item(object):
|
||||
def __init__(self, hash_key, hash_key_type, range_key, range_key_type, attrs):
|
||||
self.hash_key = hash_key
|
||||
self.hash_key_type = hash_key_type
|
||||
self.range_key = range_key
|
||||
self.range_key_type = range_key_type
|
||||
self.attrs = attrs
|
||||
|
||||
@property
|
||||
def describe(self):
|
||||
return {
|
||||
"Attributes": self.attrs
|
||||
}
|
||||
|
||||
def describe_attrs(self, attributes):
|
||||
if attributes:
|
||||
included = {}
|
||||
for key, value in self.attrs.iteritems():
|
||||
if key in attributes:
|
||||
included[key] = value
|
||||
else:
|
||||
included = self.attrs
|
||||
return {
|
||||
"Item": included
|
||||
}
|
||||
|
||||
|
||||
class Table(object):
|
||||
|
||||
def __init__(self, name, hash_key_attr=None, hash_key_type=None,
|
||||
|
|
@ -19,6 +47,7 @@ class Table(object):
|
|||
self.read_capacity = read_capacity
|
||||
self.write_capacity = write_capacity
|
||||
self.created_at = datetime.datetime.now()
|
||||
self.items = defaultdict(dict)
|
||||
|
||||
@property
|
||||
def describe(self):
|
||||
|
|
@ -41,11 +70,50 @@ class Table(object):
|
|||
},
|
||||
"TableName": self.name,
|
||||
"TableStatus": "ACTIVE",
|
||||
"ItemCount": 0,
|
||||
"ItemCount": len(self),
|
||||
"TableSizeBytes": 0,
|
||||
}
|
||||
}
|
||||
|
||||
def __len__(self):
|
||||
count = 0
|
||||
for key, value in self.items.iteritems():
|
||||
count += len(value)
|
||||
return count
|
||||
|
||||
def __nonzero__(self):
|
||||
return True
|
||||
|
||||
def put_item(self, item_attrs):
|
||||
hash_value = item_attrs.get(self.hash_key_attr).values()[0]
|
||||
range_value = item_attrs.get(self.range_key_attr).values()[0]
|
||||
item = Item(hash_value, self.hash_key_type, range_value, self.range_key_type, item_attrs)
|
||||
self.items[hash_value][range_value] = item
|
||||
return item
|
||||
|
||||
def get_item(self, hash_key, range_key):
|
||||
try:
|
||||
return self.items[hash_key][range_key]
|
||||
except KeyError:
|
||||
return None
|
||||
|
||||
def query(self, hash_key, range_comparison, range_value):
|
||||
results = []
|
||||
last_page = True # Once pagination is implemented, change this
|
||||
|
||||
possible_results = self.items.get(hash_key, [])
|
||||
comparison_func = get_comparison_func(range_comparison)
|
||||
for result in possible_results.values():
|
||||
if comparison_func(result.range_key, range_value):
|
||||
results.append(result)
|
||||
return results, last_page
|
||||
|
||||
def delete_item(self, hash_key, range_key):
|
||||
try:
|
||||
return self.items[hash_key].pop(range_key)
|
||||
except KeyError:
|
||||
return None
|
||||
|
||||
|
||||
class DynamoDBBackend(BaseBackend):
|
||||
|
||||
|
|
@ -53,6 +121,46 @@ class DynamoDBBackend(BaseBackend):
|
|||
self.tables = OrderedDict()
|
||||
|
||||
def create_table(self, name, **params):
|
||||
self.tables[name] = Table(name, **params)
|
||||
table = Table(name, **params)
|
||||
self.tables[name] = table
|
||||
return table
|
||||
|
||||
def delete_table(self, name):
|
||||
return self.tables.pop(name, None)
|
||||
|
||||
def update_table_throughput(self, name, new_read_units, new_write_units):
|
||||
table = self.tables[name]
|
||||
table.read_capacity = new_read_units
|
||||
table.write_capacity = new_write_units
|
||||
return table
|
||||
|
||||
def put_item(self, table_name, item_attrs):
|
||||
table = self.tables.get(table_name)
|
||||
if not table:
|
||||
return None
|
||||
|
||||
return table.put_item(item_attrs)
|
||||
|
||||
def get_item(self, table_name, hash_key, range_key):
|
||||
table = self.tables.get(table_name)
|
||||
if not table:
|
||||
return None
|
||||
|
||||
return table.get_item(hash_key, range_key)
|
||||
|
||||
def query(self, table_name, hash_key, range_comparison, range_value):
|
||||
table = self.tables.get(table_name)
|
||||
if not table:
|
||||
return None
|
||||
|
||||
return table.query(hash_key, range_comparison, range_value)
|
||||
|
||||
def delete_item(self, table_name, hash_key, range_key):
|
||||
table = self.tables.get(table_name)
|
||||
if not table:
|
||||
return None
|
||||
|
||||
return table.delete_item(hash_key, range_key)
|
||||
|
||||
|
||||
dynamodb_backend = DynamoDBBackend()
|
||||
|
|
|
|||
|
|
@ -48,6 +48,50 @@ class DynamoHandler(object):
|
|||
response["LastEvaluatedTableName"] = tables[-1]
|
||||
return json.dumps(response)
|
||||
|
||||
def CreateTable(self, uri, body, headers):
|
||||
name = body['TableName']
|
||||
|
||||
key_schema = body['KeySchema']
|
||||
hash_hey = key_schema['HashKeyElement']
|
||||
hash_key_attr = hash_hey['AttributeName']
|
||||
hash_key_type = hash_hey['AttributeType']
|
||||
|
||||
range_hey = key_schema['RangeKeyElement']
|
||||
range_key_attr = range_hey['AttributeName']
|
||||
range_key_type = range_hey['AttributeType']
|
||||
|
||||
throughput = body["ProvisionedThroughput"]
|
||||
read_units = throughput["ReadCapacityUnits"]
|
||||
write_units = throughput["WriteCapacityUnits"]
|
||||
|
||||
table = dynamodb_backend.create_table(
|
||||
name,
|
||||
hash_key_attr=hash_key_attr,
|
||||
hash_key_type=hash_key_type,
|
||||
range_key_attr=range_key_attr,
|
||||
range_key_type=range_key_type,
|
||||
read_capacity=int(read_units),
|
||||
write_capacity=int(write_units),
|
||||
)
|
||||
return json.dumps(table.describe)
|
||||
|
||||
def DeleteTable(self, uri, body, headers):
|
||||
name = body['TableName']
|
||||
table = dynamodb_backend.delete_table(name)
|
||||
if table:
|
||||
return json.dumps(table.describe)
|
||||
else:
|
||||
er = 'com.amazonaws.dynamodb.v20111205#ResourceNotFoundException'
|
||||
return self.error(er)
|
||||
|
||||
def UpdateTable(self, uri, body, headers):
|
||||
name = body['TableName']
|
||||
throughput = 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)
|
||||
return json.dumps(table.describe)
|
||||
|
||||
def DescribeTable(self, uri, body, headers):
|
||||
name = body['TableName']
|
||||
try:
|
||||
|
|
@ -57,6 +101,66 @@ class DynamoHandler(object):
|
|||
return self.error(er)
|
||||
return json.dumps(table.describe)
|
||||
|
||||
def PutItem(self, uri, body, headers):
|
||||
name = body['TableName']
|
||||
item = body['Item']
|
||||
result = dynamodb_backend.put_item(name, item)
|
||||
item_dict = result.describe
|
||||
item_dict['ConsumedCapacityUnits'] = 1
|
||||
return json.dumps(item_dict)
|
||||
|
||||
def GetItem(self, uri, body, headers):
|
||||
name = body['TableName']
|
||||
hash_key = body['Key']['HashKeyElement'].values()[0]
|
||||
range_key = body['Key']['RangeKeyElement'].values()[0]
|
||||
attrs_to_get = body.get('AttributesToGet')
|
||||
item = dynamodb_backend.get_item(name, hash_key, range_key)
|
||||
if item:
|
||||
item_dict = item.describe_attrs(attrs_to_get)
|
||||
item_dict['ConsumedCapacityUnits'] = 0.5
|
||||
return json.dumps(item_dict)
|
||||
else:
|
||||
er = 'com.amazonaws.dynamodb.v20111205#ResourceNotFoundException'
|
||||
return self.error(er)
|
||||
|
||||
def Query(self, uri, body, headers):
|
||||
name = body['TableName']
|
||||
hash_key = body['HashKeyValue'].values()[0]
|
||||
range_condition = body['RangeKeyCondition']
|
||||
range_comparison = range_condition['ComparisonOperator']
|
||||
range_value = range_condition['AttributeValueList'][0].values()[0]
|
||||
items, last_page = dynamodb_backend.query(name, hash_key, range_comparison, range_value)
|
||||
|
||||
result = {
|
||||
"Count": len(items),
|
||||
"Items": [item.attrs for item in items],
|
||||
"ConsumedCapacityUnits": 1,
|
||||
}
|
||||
|
||||
if not last_page:
|
||||
result["LastEvaluatedKey"] = {
|
||||
"HashKeyElement": items[-1].hash_key,
|
||||
"RangeKeyElement": items[-1].range_key,
|
||||
}
|
||||
return json.dumps(result)
|
||||
|
||||
def DeleteItem(self, uri, body, headers):
|
||||
name = body['TableName']
|
||||
hash_key = body['Key']['HashKeyElement'].values()[0]
|
||||
range_key = body['Key']['RangeKeyElement'].values()[0]
|
||||
return_values = body.get('ReturnValues', '')
|
||||
item = dynamodb_backend.delete_item(name, hash_key, range_key)
|
||||
if item:
|
||||
if return_values == 'ALL_OLD':
|
||||
item_dict = item.describe
|
||||
else:
|
||||
item_dict = {'Attributes': []}
|
||||
item_dict['ConsumedCapacityUnits'] = 0.5
|
||||
return json.dumps(item_dict)
|
||||
else:
|
||||
er = 'com.amazonaws.dynamodb.v20111205#ResourceNotFoundException'
|
||||
return self.error(er)
|
||||
|
||||
|
||||
def handler(uri, body, headers):
|
||||
body = json.loads(body or '{}')
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue