Skip to content

Commit 7057bd9

Browse files
authored
refactor: make transaction methods protected (#166)
1 parent fed4cb3 commit 7057bd9

File tree

5 files changed

+9
-8
lines changed

5 files changed

+9
-8
lines changed

crypto/transactions/types/abstract_transaction.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -41,7 +41,7 @@ def get_public_key(self, compact_signature, hash_):
4141
return public_key
4242

4343
def recover_sender(self):
44-
signature_with_recid = self.get_signature()
44+
signature_with_recid = self._get_signature()
4545
if not signature_with_recid:
4646
return False
4747

@@ -51,7 +51,7 @@ def recover_sender(self):
5151
self.data['senderAddress'] = Address.from_public_key(self.data['senderPublicKey'])
5252

5353
def verify(self) -> bool:
54-
signature_with_recid = self.get_signature()
54+
signature_with_recid = self._get_signature()
5555
if not signature_with_recid:
5656
return False
5757

@@ -79,7 +79,7 @@ def to_json(self) -> str:
7979
def hash(self, skip_signature: bool) -> str:
8080
return TransactionUtils.to_hash(self.data, skip_signature=skip_signature)
8181

82-
def get_signature(self):
82+
def _get_signature(self):
8383
recover_id = int(self.data.get('v', 0)) - Constants.ETHEREUM_RECOVERY_ID_OFFSET.value
8484
r = self.data.get('r')
8585
s = self.data.get('s')
@@ -90,7 +90,7 @@ def get_signature(self):
9090
return None
9191

9292
@staticmethod
93-
def decode_payload(data: dict, abi_type: ContractAbiType = ContractAbiType.CONSENSUS) -> Optional[dict]:
93+
def _decode_payload(data: dict, abi_type: ContractAbiType = ContractAbiType.CONSENSUS) -> Optional[dict]:
9494
from crypto.transactions.deserializer import Deserializer
9595

9696
return Deserializer.decode_payload(data, abi_type)

crypto/transactions/types/multipayment.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@
66

77
class Multipayment(AbstractTransaction):
88
def __init__(self, data: dict):
9-
payload = self.decode_payload(data, ContractAbiType.MULTIPAYMENT)
9+
payload = self._decode_payload(data, ContractAbiType.MULTIPAYMENT)
1010
if payload:
1111
data['pay'] = payload.get('args', [])
1212

crypto/transactions/types/username_registration.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@
66

77
class UsernameRegistration(AbstractTransaction):
88
def __init__(self, data: dict):
9-
payload = self.decode_payload(data, ContractAbiType.USERNAMES)
9+
payload = self._decode_payload(data, ContractAbiType.USERNAMES)
1010
if payload:
1111
data['username'] = payload.get('args', [None])[0] if payload.get('args') else None
1212

crypto/transactions/types/validator_registration.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6,9 +6,10 @@
66

77
class ValidatorRegistration(AbstractTransaction):
88
def __init__(self, data: dict):
9-
payload = self.decode_payload(data)
9+
payload = self._decode_payload(data)
1010
if payload:
1111
data['validatorPublicKey'] = TransactionUtils.parse_hex_from_str(payload.get('args', [None])[0]) if payload.get('args') else None
12+
1213
super().__init__(data)
1314

1415
def get_payload(self) -> str:

crypto/transactions/types/vote.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,7 @@
55

66
class Vote(AbstractTransaction):
77
def __init__(self, data: dict):
8-
payload = self.decode_payload(data)
8+
payload = self._decode_payload(data)
99
if payload:
1010
data['vote'] = payload.get('args', [None])[0] if payload.get('args') else None
1111

0 commit comments

Comments
 (0)