diff --git a/crypto/identity/wif.py b/crypto/identity/wif.py index ba3cb90..65b3b28 100644 --- a/crypto/identity/wif.py +++ b/crypto/identity/wif.py @@ -7,22 +7,24 @@ from crypto.configuration.network import Network -def wif_from_passphrase(passphrase: str, network_wif: Optional[str] = None): - """Get wif from passphrase +class WIF: + @staticmethod + def from_passphrase(passphrase: str, network_wif: Optional[str] = None): + """Get wif from passphrase - Args: - passphrase (str): - network_wif (str, optional): + Args: + passphrase (str): + network_wif (str, optional): - Returns: - string: wif - """ - if not network_wif: - network = Network.get_network() + Returns: + string: wif + """ + if not network_wif: + network = Network.get_network() - network_wif = network.wif() + network_wif = network.wif() - private_key = hashlib.sha256(passphrase.encode()) - seed = write_bit8(int(network_wif, 16)) + private_key.digest() + write_bit8(0x01) + private_key = hashlib.sha256(passphrase.encode()) + seed = write_bit8(int(network_wif, 16)) + private_key.digest() + write_bit8(0x01) - return b58encode_check(seed).decode() + return b58encode_check(seed).decode() diff --git a/tests/identity/test_wif.py b/tests/identity/test_wif.py index b3934df..5c550ef 100644 --- a/tests/identity/test_wif.py +++ b/tests/identity/test_wif.py @@ -1,10 +1,10 @@ from crypto.configuration.network import Network -from crypto.identity.wif import wif_from_passphrase +from crypto.identity.wif import WIF from crypto.networks.testnet import Testnet def test_wif_from_passphrase(identity): Network.set_network(Testnet()) - result = wif_from_passphrase(identity['passphrase']) + result = WIF.from_passphrase(identity['passphrase']) assert result == identity['data']['wif']