def _save_transaction(self, bucket=None):
"""Save this transaction to the object store
Args:
bucket (dict): Bucket to load data from
Returns:
None
"""
from Acquire.Accounting import Ledger as _Ledger
_Ledger.save_transaction(self, bucket=bucket)
def load_test_and_set(uid, expected_state, new_state,
bucket=None):
"""Static method to load up the Transaction record associated with
the passed UID, check that the transaction state matches
'expected_state', and if it does, to update the transaction
state to 'new_state'. This returns the loaded (and updated)
transaction
Args:
expected_state (TransactionState): State of transaction
new_state (TransactionState): State to update transaction to
bucket (dict): Bucket to load data from
Returns:
Transaction: Updated transaction
"""
if bucket is None:
from Acquire.Service import get_service_account_bucket \
as _get_service_account_bucket
bucket = _get_service_account_bucket()
from Acquire.Accounting import Ledger as _Ledger
from Acquire.ObjectStore import Mutex as _Mutex
try:
mutex = _Mutex(uid, timeout=600, lease_time=600)
except Exception as e:
raise LedgerError("Cannot secure a Ledger mutex for transaction "
"'%s'. Error = %s" % (uid, str(e)))
try:
transaction = _Ledger.load_transaction(uid, bucket)
if transaction.transaction_state() != expected_state:
raise TransactionError(
"Cannot update the state of the transaction %s from "
"%s to %s as it is not in the expected state" %
(str(transaction), expected_state.value, new_state.value))
transaction._transaction_state = new_state
except:
mutex.unlock()
raise
# now need to write anything back if the state isn't changed
if expected_state == new_state:
return transaction
# make sure we have enough time remaining on the lease to be
# able to write this result back to the object store...
if mutex.seconds_remaining_on_lease() < 100:
try:
mutex.fully_unlock()
except:
pass
return TransactionRecord.load_test_and_set(uid, expected_state,
new_state, bucket)
try:
_Ledger.save_transaction(transaction, bucket)
except:
mutex.unlock()
raise
return transaction