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

Handle delete updates.

Browse Source
This commit is contained in:
Paul Craciunoiu 2016-01-04 16:29:02 -07:00
parent f1099dd006
commit f5406ad212
1 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
moto/dynamodb2/models.py
View File

@ -124,6 +124,9 @@ class Item(object):
def update_with_attribute_updates(self, attribute_updates): def update_with_attribute_updates(self, attribute_updates):
for attribute_name, update_action in attribute_updates.items(): for attribute_name, update_action in attribute_updates.items():
action = update_action['Action'] action = update_action['Action']
if action == 'DELETE' and not 'Value' in update_action:
del self.attrs[attribute_name]
continue
new_value = list(update_action['Value'].values())[0] new_value = list(update_action['Value'].values())[0]
if action == 'PUT': if action == 'PUT':
# TODO deal with other types # TODO deal with other types
@ -133,6 +136,8 @@ class Item(object):
self.attrs[attribute_name] = DynamoType({"M": new_value}) self.attrs[attribute_name] = DynamoType({"M": new_value})
elif update_action['Value'].keys() == ['N']: elif update_action['Value'].keys() == ['N']:
self.attrs[attribute_name] = DynamoType({"N": new_value}) self.attrs[attribute_name] = DynamoType({"N": new_value})
elif update_action['Value'].keys() == ['NULL']:
del self.attrs[attribute_name]
else: else:
self.attrs[attribute_name] = DynamoType({"S": new_value}) self.attrs[attribute_name] = DynamoType({"S": new_value})