Chia-Network/chia-blockchain
1
0
Fork 0
mirror of https://github.com/Chia-Network/chia-blockchain.git synced 2024-09-21 00:24:37 +03:00
Code Issues Packages Projects Releases Wiki Activity

use WalletStateManager.get_wallet() in wallet_rpc_api (#15334)

Browse Source
This commit is contained in:
Kyle Altendorf 2023-05-22 19:09:55 -04:00 committed by GitHub
parent c45875d517
commit 526310374f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 6 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
chia/rpc/wallet_rpc_api.py
View File

@ -2385,13 +2385,13 @@ class WalletRpcApi:
coin_ids = []
nft_ids = []
fee = uint64(request.get("fee", 0))
nft_wallet: NFTWallet
for nft_coin in request["nft_coin_list"]:
if "nft_coin_id" not in nft_coin or "wallet_id" not in nft_coin:
log.error(f"Cannot set DID for NFT :{nft_coin}, missing nft_coin_id or wallet_id.")
continue
wallet_id = uint32(nft_coin["wallet_id"])
nft_wallet = self.service.wallet_state_manager.wallets[wallet_id]
assert isinstance(nft_wallet, NFTWallet)
nft_wallet = self.service.wallet_state_manager.get_wallet(id=wallet_id, required_type=NFTWallet)
nft_coin_id = nft_coin["nft_coin_id"]
if nft_coin_id.startswith(AddressType.NFT.hrp(self.service.config)):
nft_id = decode_puzzle_hash(nft_coin_id)
@ -2413,8 +2413,7 @@ class WalletRpcApi:
first = True
nft_wallet = None
for wallet_id, nft_list in nft_dict.items():
nft_wallet = self.service.wallet_state_manager.wallets[wallet_id]
assert isinstance(nft_wallet, NFTWallet)
nft_wallet = self.service.wallet_state_manager.get_wallet(id=wallet_id, required_type=NFTWallet)
if not first:
tx_list.extend(await nft_wallet.set_bulk_nft_did(nft_list, did_id))
else:
@ -2474,13 +2473,13 @@ class WalletRpcApi:
tx_list: List[TransactionRecord] = []
coin_ids = []
fee = uint64(request.get("fee", 0))
nft_wallet: NFTWallet
for nft_coin in request["nft_coin_list"]:
if "nft_coin_id" not in nft_coin or "wallet_id" not in nft_coin:
log.error(f"Cannot transfer NFT :{nft_coin}, missing nft_coin_id or wallet_id.")
continue
wallet_id = uint32(nft_coin["wallet_id"])
nft_wallet = self.service.wallet_state_manager.wallets[wallet_id]
assert isinstance(nft_wallet, NFTWallet)
nft_wallet = self.service.wallet_state_manager.get_wallet(id=wallet_id, required_type=NFTWallet)
nft_coin_id = nft_coin["nft_coin_id"]
if nft_coin_id.startswith(AddressType.NFT.hrp(self.service.config)):
nft_id = decode_puzzle_hash(nft_coin_id)
@ -2496,8 +2495,7 @@ class WalletRpcApi:
first = True
nft_wallet = None
for wallet_id, nft_list in nft_dict.items():
nft_wallet = self.service.wallet_state_manager.wallets[wallet_id]
assert isinstance(nft_wallet, NFTWallet)
nft_wallet = self.service.wallet_state_manager.get_wallet(id=wallet_id, required_type=NFTWallet)
if not first:
tx_list.extend(await nft_wallet.bulk_transfer_nft(nft_list, puzzle_hash))
else: