diff --git a/chia/data_layer/data_layer_wallet.py b/chia/data_layer/data_layer_wallet.py index ff51c9cadaef..845f64fd9b1d 100644 --- a/chia/data_layer/data_layer_wallet.py +++ b/chia/data_layer/data_layer_wallet.py @@ -54,6 +54,7 @@ from chia.wallet.wallet import Wallet from chia.wallet.wallet_coin_record import WalletCoinRecord from chia.wallet.wallet_info import WalletInfo +from chia.wallet.wallet_protocol import WalletProtocol if TYPE_CHECKING: from chia.wallet.wallet_state_manager import WalletStateManager @@ -102,7 +103,7 @@ def from_json_dict(cls, json_dict: Dict[str, Any]) -> "Mirror": @final -class DataLayerWallet: +class DataLayerWallet(WalletProtocol): wallet_state_manager: WalletStateManager log: logging.Logger wallet_info: WalletInfo diff --git a/chia/pools/pool_wallet.py b/chia/pools/pool_wallet.py index 4187e9f5c59d..1596b2c30eb5 100644 --- a/chia/pools/pool_wallet.py +++ b/chia/pools/pool_wallet.py @@ -54,11 +54,12 @@ from chia.wallet.wallet import Wallet from chia.wallet.wallet_coin_record import WalletCoinRecord from chia.wallet.wallet_info import WalletInfo +from chia.wallet.wallet_protocol import WalletProtocol @final @dataclasses.dataclass -class PoolWallet: +class PoolWallet(WalletProtocol): MINIMUM_INITIAL_BALANCE = 1 MINIMUM_RELATIVE_LOCK_HEIGHT = 5 MAXIMUM_RELATIVE_LOCK_HEIGHT = 1000 diff --git a/chia/wallet/cat_wallet/cat_wallet.py b/chia/wallet/cat_wallet/cat_wallet.py index 3a86dd9e2718..ac1ebe8ff0e5 100644 --- a/chia/wallet/cat_wallet/cat_wallet.py +++ b/chia/wallet/cat_wallet/cat_wallet.py @@ -55,6 +55,7 @@ from chia.wallet.wallet import Wallet from chia.wallet.wallet_coin_record import WalletCoinRecord from chia.wallet.wallet_info import WalletInfo +from chia.wallet.wallet_protocol import WalletProtocol if TYPE_CHECKING: from chia.wallet.wallet_state_manager import WalletStateManager @@ -66,7 +67,7 @@ QUOTED_MOD_HASH = calculate_hash_of_quoted_mod_hash(CAT_MOD_HASH) -class CATWallet: +class CATWallet(WalletProtocol): wallet_state_manager: WalletStateManager log: logging.Logger wallet_info: WalletInfo diff --git a/chia/wallet/did_wallet/did_wallet.py b/chia/wallet/did_wallet/did_wallet.py index e9b506c8f925..51c517b7115f 100644 --- a/chia/wallet/did_wallet/did_wallet.py +++ b/chia/wallet/did_wallet/did_wallet.py @@ -44,9 +44,10 @@ from chia.wallet.wallet import CHIP_0002_SIGN_MESSAGE_PREFIX, Wallet from chia.wallet.wallet_coin_record import WalletCoinRecord from chia.wallet.wallet_info import WalletInfo +from chia.wallet.wallet_protocol import WalletProtocol -class DIDWallet: +class DIDWallet(WalletProtocol): wallet_state_manager: Any log: logging.Logger wallet_info: WalletInfo diff --git a/chia/wallet/nft_wallet/nft_wallet.py b/chia/wallet/nft_wallet/nft_wallet.py index c2f607dbac74..58b88a0613fa 100644 --- a/chia/wallet/nft_wallet/nft_wallet.py +++ b/chia/wallet/nft_wallet/nft_wallet.py @@ -55,11 +55,12 @@ from chia.wallet.wallet_coin_record import WalletCoinRecord from chia.wallet.wallet_info import WalletInfo from chia.wallet.wallet_nft_store import WalletNftStore +from chia.wallet.wallet_protocol import WalletProtocol _T_NFTWallet = TypeVar("_T_NFTWallet", bound="NFTWallet") -class NFTWallet: +class NFTWallet(WalletProtocol): wallet_state_manager: Any log: logging.Logger wallet_info: WalletInfo diff --git a/chia/wallet/wallet.py b/chia/wallet/wallet.py index b9fb92617054..51b3a7f21d56 100644 --- a/chia/wallet/wallet.py +++ b/chia/wallet/wallet.py @@ -46,6 +46,7 @@ from chia.wallet.util.wallet_types import AmountWithPuzzlehash, WalletType from chia.wallet.wallet_coin_record import WalletCoinRecord from chia.wallet.wallet_info import WalletInfo +from chia.wallet.wallet_protocol import WalletProtocol if TYPE_CHECKING: from chia.server.ws_connection import WSChiaConnection @@ -54,7 +55,7 @@ CHIP_0002_SIGN_MESSAGE_PREFIX = "Chia Signed Message" -class Wallet: +class Wallet(WalletProtocol): wallet_info: WalletInfo wallet_state_manager: Any log: logging.Logger