Skip to content
Snippets Groups Projects
Unverified Commit 3d0b5e9c authored by Katleho Madisa's avatar Katleho Madisa
Browse files
parent c3aac0e8
Branches
Tags
No related merge requests found
...@@ -72,10 +72,7 @@ class Transaction: ...@@ -72,10 +72,7 @@ class Transaction:
self._params = params self._params = params
self._transaction_id_key = transaction_id_key self._transaction_id_key = transaction_id_key
self._transaction_id = transaction_id if self._is_valid_id(transaction_id) else "" self._transaction_id = self._get_id_from_params_or_generate_new_id(transaction_id)
if not self._transaction_id:
self._transaction_id = self._get_from_params_or_generate_new_id()
if transaction_id and params.get(self._transaction_id_key): if transaction_id and params.get(self._transaction_id_key):
self.logger.info( self.logger.info(
...@@ -100,20 +97,27 @@ class Transaction: ...@@ -100,20 +97,27 @@ class Transaction:
if exc_type: if exc_type:
raise raise
def _get_from_params_or_generate_new_id(self): def _get_id_from_params_or_generate_new_id(self, transaction_id):
"""Use the transaction_id_key to get the transaction ID from the parameters """At first use the transaction_id passed or use the transaction_id_key to get the
or generate a new one if it's not there. transaction ID from the parameters or generate a new one if it's not there.
Parameters
----------
transaction_id : [String]
[The transaction ID]
Returns Returns
------- -------
[String] [String]
[transaction ID] [transaction ID]
""" """
transaction_id = self._params.get(self._transaction_id_key) _transaction_id = (
if not self._is_valid_id(transaction_id): transaction_id if transaction_id else self._params.get(self._transaction_id_key)
transaction_id = self._generate_new_id() )
self.logger.info(f"Generated transaction ID {transaction_id}") if not self._is_valid_id(_transaction_id):
return transaction_id _transaction_id = self._generate_new_id()
self.logger.info(f"Generated transaction ID {_transaction_id}")
return _transaction_id
def _is_valid_id(self, transaction_id): def _is_valid_id(self, transaction_id):
"""Check if the ID is valid """Check if the ID is valid
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment