add delete_item
This commit is contained in:
@@ -491,7 +491,10 @@ class DynamoDBCollection:
|
|||||||
return data or default
|
return data or default
|
||||||
|
|
||||||
def put_item(
|
def put_item(
|
||||||
self, key: KeyPair, ttl: int | datetime | None = None, **kwargs: Any
|
self,
|
||||||
|
key: KeyPair,
|
||||||
|
ttl: int | datetime | None = None,
|
||||||
|
**kwargs: Any,
|
||||||
) -> bool:
|
) -> bool:
|
||||||
now_ = now(self.tz)
|
now_ = now(self.tz)
|
||||||
|
|
||||||
@@ -514,3 +517,20 @@ class DynamoDBCollection:
|
|||||||
return self.persistence_layer.put_item(
|
return self.persistence_layer.put_item(
|
||||||
item=key | {'create_date': now_} | kwargs
|
item=key | {'create_date': now_} | kwargs
|
||||||
)
|
)
|
||||||
|
|
||||||
|
def delete_item(
|
||||||
|
self,
|
||||||
|
key: KeyPair,
|
||||||
|
*,
|
||||||
|
cond_expr: str | None = None,
|
||||||
|
expr_attr_names: dict | None = None,
|
||||||
|
expr_attr_values: dict | None = None,
|
||||||
|
) -> bool:
|
||||||
|
return self.persistence_layer.delete_item(
|
||||||
|
key=key,
|
||||||
|
cond_expr=cond_expr,
|
||||||
|
expr_attr_names=expr_attr_names,
|
||||||
|
expr_attr_values=expr_attr_values,
|
||||||
|
)
|
||||||
|
|
||||||
|
def get_items(self): ...
|
||||||
|
|||||||
@@ -121,3 +121,18 @@ def test_collection_put_item(
|
|||||||
assert 'name' in data
|
assert 'name' in data
|
||||||
assert 'ttl' in data
|
assert 'ttl' in data
|
||||||
assert 'ttl_date' in data
|
assert 'ttl_date' in data
|
||||||
|
|
||||||
|
|
||||||
|
def test_collection_delete_item(
|
||||||
|
dynamodb_seeds,
|
||||||
|
dynamodb_persistence_layer: DynamoDBPersistenceLayer,
|
||||||
|
):
|
||||||
|
collect = DynamoDBCollection(dynamodb_persistence_layer)
|
||||||
|
|
||||||
|
# This item was added from seeds
|
||||||
|
assert collect.delete_item(
|
||||||
|
key=KeyPair(
|
||||||
|
'5OxmMjL-ujoR5IMGegQz',
|
||||||
|
Key('sergio@somsbeta.com.br', prefix='emails'),
|
||||||
|
)
|
||||||
|
)
|
||||||
|
|||||||
Reference in New Issue
Block a user