41 lines
997 B
Python
41 lines
997 B
Python
from layercake.dateutils import now
|
|
from layercake.dynamodb import DynamoDBPersistenceLayer, KeyPair, TransactItems
|
|
|
|
|
|
def update_policies(
|
|
id: str,
|
|
/,
|
|
payment_policy: dict = {},
|
|
billing_policy: dict = {},
|
|
*,
|
|
persistence_layer: DynamoDBPersistenceLayer,
|
|
):
|
|
now_ = now()
|
|
transact = TransactItems(persistence_layer.table_name)
|
|
|
|
if payment_policy:
|
|
transact.put(
|
|
item={
|
|
'id': id,
|
|
'sk': 'payment_policy',
|
|
'create_date': now_,
|
|
}
|
|
| payment_policy
|
|
)
|
|
else:
|
|
transact.delete(key=KeyPair(id, 'payment_policy'))
|
|
|
|
if billing_policy:
|
|
transact.put(
|
|
item={
|
|
'id': id,
|
|
'sk': 'billing_policy',
|
|
'create_date': now_,
|
|
}
|
|
| billing_policy
|
|
)
|
|
else:
|
|
transact.delete(key=KeyPair(id, 'billing_policy'))
|
|
|
|
return persistence_layer.transact_write_items(transact)
|