catalin/moto
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity

DynamoDB - DELETE item from nested sets

Browse source
This commit is contained in:
Bert Blommers 2019-11-28 13:22:20 +00:00
commit e41bc9fc58
2 changed files with 35 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

10
moto/dynamodb2/models.py
View file

@ -420,12 +420,18 @@ class Item(BaseModel):
if not dyn_value.is_set():
raise TypeError
existing = self.attrs.get(key, None)
key_head = key.split(".")[0]
key_tail = ".".join(key.split(".")[1:])
existing = self.attrs.get(key_head)
existing = existing.get(key_tail)
if existing:
if not existing.same_type(dyn_value):
raise TypeError
new_set = set(existing.value).difference(dyn_value.value)
self.attrs[key] = DynamoType({existing.type: list(new_set)})
existing.set(
key=None,
new_value=DynamoType({existing.type: list(new_set)}),
)
else:
raise NotImplementedError(
"{} update action not yet supported".format(action)