1212from typing import Optional
1313
1414from hdwallet import HDWallet as ext_HDWallet # type: ignore
15- from hdwallet .cryptocurrencies import Bitcoin
16- from hdwallet .derivations import CustomDerivation
17- from hdwallet .mnemonics import BIP39Mnemonic
15+ from hdwallet .symbols import BTC , BTCTEST # type: ignore
1816
1917from bitcoinutils .setup import is_mainnet
2018from bitcoinutils .keys import PrivateKey
@@ -42,14 +40,20 @@ def __init__(
4240 ):
4341 """Instantiate a hdwallet object using the corresponding library with BTC"""
4442
45- self .hdw = ext_HDWallet (cryptocurrency = Bitcoin , network = 'mainnet' if is_mainnet () else 'testnet' )
43+ symbol = None
44+ if is_mainnet ():
45+ symbol = BTC
46+ else :
47+ symbol = BTCTEST
48+
49+ self .hdw = ext_HDWallet (symbol )
4650
4751 if mnemonic :
48- self .hdw .from_mnemonic (mnemonic = BIP39Mnemonic ( mnemonic = mnemonic ) )
52+ self .hdw .from_mnemonic (mnemonic = mnemonic )
4953
5054 if xprivate_key and path :
5155 self .hdw .from_xprivate_key (xprivate_key = xprivate_key )
52- self .hdw .from_derivation ( CustomDerivation ( path ) )
56+ self .hdw .from_path ( path = path )
5357
5458 @classmethod
5559 def from_mnemonic (cls , mnemonic : str ):
@@ -68,7 +72,7 @@ def from_path(self, path: str):
6872 """Set/update the path"""
6973
7074 self .hdw .clean_derivation () # type: ignore
71- self .hdw .from_derivation ( CustomDerivation ( path ) )
75+ self .hdw .from_path ( path = path )
7276
7377 def get_private_key (self ):
7478 """Return a PrivateKey object used throughout bitcoinutils library"""
0 commit comments