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

#1834 - Bugfix when removing item in double nested maps

Browse Source
This commit is contained in:
Bert Blommers 2019-10-05 11:33:34 +01:00
parent d86dcb2ee9
commit 71e86ab417
1 changed files with 6 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
moto/dynamodb2/models.py
View File

@ -197,15 +197,18 @@ class Item(BaseModel):
# Hack but it'll do, traverses into a dict # Hack but it'll do, traverses into a dict
last_val_type = list(last_val.keys()) last_val_type = list(last_val.keys())
if last_val_type and last_val_type[0] == 'M': if last_val_type and last_val_type[0] == 'M':
last_val = last_val['M'] last_val = last_val['M']
if key_part not in last_val: if key_part not in last_val:
last_val[key_part] = {'M': {}} last_val[key_part] = {'M': {}}
last_val = last_val[key_part] last_val = last_val[key_part]
last_val.pop(key_parts[-1], None) last_val_type = list(last_val.keys())
self.attrs.pop(value, None) if last_val_type and last_val_type[0] == 'M':
last_val['M'].pop(key_parts[-1], None)
else:
last_val.pop(key_parts[-1], None)
elif action == 'SET': elif action == 'SET':
key, value = value.split("=", 1) key, value = value.split("=", 1)
key = key.strip() key = key.strip()