Fixed dynamodb2 put_item ValidationException

This commit is contained in:
Andrea Amorosi 2020-03-28 17:59:42 +00:00
commit 349b381390
3 changed files with 55 additions and 1 deletions

View file

@ -800,13 +800,19 @@ class Table(BaseModel):
overwrite=False,
):
if self.hash_key_attr not in item_attrs.keys():
raise ValueError(
raise KeyError(
"One or more parameter values were invalid: Missing the key "
+ self.hash_key_attr
+ " in the item"
)
hash_value = DynamoType(item_attrs.get(self.hash_key_attr))
if self.has_range_key:
if self.range_key_attr not in item_attrs.keys():
raise KeyError(
"One or more parameter values were invalid: Missing the key "
+ self.range_key_attr
+ " in the item"
)
range_value = DynamoType(item_attrs.get(self.range_key_attr))
else:
range_value = None