def wif_to_key(wif):
private_key_bytes, compressed, version = wif_to_bytes(wif)
if version == 'main':
if compressed:
return PrivateKey.from_bytes(private_key_bytes)
else:
return PrivateKey(wif)
else:
pass
def __init__(self, wif=None):
if wif:
if isinstance(wif, str):
private_key_bytes, compressed, _ = wif_to_bytes(wif)
self._pk = ECPrivateKey(private_key_bytes)
elif isinstance(wif, ECPrivateKey):
self._pk = wif
compressed = True
else:
raise TypeError('Wallet Import Format must be a string.')
else:
self._pk = ECPrivateKey()
compressed = True
self._public_point = None
self._public_key = self._pk.public_key.format(compressed=compressed)
def test_invalid_network(self):
with pytest.raises(ValueError):
wif_to_bytes(LITECOIN_ADDRESS)
def test_compressed(self):
assert wif_to_bytes(WALLET_FORMAT_COMPRESSED_MAIN) == (PRIVATE_KEY_BYTES, True, 'main')
def test_testnet(self):
assert wif_to_bytes(WALLET_FORMAT_TEST) == (PRIVATE_KEY_BYTES, False, 'test')
def test_mainnet(self):
assert wif_to_bytes(WALLET_FORMAT_MAIN) == (PRIVATE_KEY_BYTES, False, 'main')