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

#3506 - DynamoDB - Allow StringSets to be passed to update_item() (#3519)

Browse source
This commit is contained in:
Bert Blommers 2020-12-07 01:31:53 -08:00 • committed by GitHub
commit e1fc3a9596
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 24 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

6
moto/dynamodb2/models/__init__.py
View file

@ -128,10 +128,10 @@ class Item(BaseModel):
new_value = list(update_action["Value"].values())[0]
if action == "PUT":
# TODO deal with other types
if isinstance(new_value, list):
self.attrs[attribute_name] = DynamoType({"L": new_value})
elif isinstance(new_value, set):
if set(update_action["Value"].keys()) == set(["SS"]):
self.attrs[attribute_name] = DynamoType({"SS": new_value})
elif isinstance(new_value, list):
self.attrs[attribute_name] = DynamoType({"L": new_value})
elif isinstance(new_value, dict):
self.attrs[attribute_name] = DynamoType({"M": new_value})
elif set(update_action["Value"].keys()) == set(["N"]):