chia-blockchain 2.5.4rc2__py3-none-any.whl → 2.5.5rc2__py3-none-any.whl

chia/wallet/wallet_weight_proof_handler.py

@@ -8,10 +8,9 @@ from concurrent.futures.process import ProcessPoolExecutor
 from multiprocessing.context import BaseContext
 from typing import IO, Optional
 
-from chia_rs import ConsensusConstants
+from chia_rs import BlockRecord, ConsensusConstants
 from chia_rs.sized_ints import uint32
 
-from chia.consensus.block_record import BlockRecord
 from chia.full_node.weight_proof import _validate_sub_epoch_summaries, validate_weight_proof_inner
 from chia.types.weight_proof import WeightProof
 from chia.util.setproctitle import getproctitle, setproctitle

chia/wallet/wsm_apis.py

@@ -0,0 +1,98 @@
+from __future__ import annotations
+
+import dataclasses
+from typing import TYPE_CHECKING
+
+from chia_rs.sized_ints import uint32
+
+from chia.util.streamable import Streamable, streamable
+from chia.wallet.derivation_record import DerivationRecord, StreamableDerivationRecord
+
+if TYPE_CHECKING:
+    # avoiding a circular import
+    from chia.wallet.wallet_state_manager import WalletStateManager
+
+
+@streamable
+@dataclasses.dataclass(frozen=True)
+class StreamableCreateMorePuzzleHashesResult(Streamable):
+    derivation_paths: list[StreamableDerivationRecord]
+    mark_existing_as_used: bool
+    unused: uint32
+    new_unhardened_keys: bool
+    last_index: uint32
+
+    @classmethod
+    def from_standard(cls, result: CreateMorePuzzleHashesResult) -> StreamableCreateMorePuzzleHashesResult:
+        return cls(
+            [StreamableDerivationRecord.from_standard(path) for path in result.derivation_paths],
+            result.mark_existing_as_used,
+            uint32(result.unused),
+            result.new_unhardened_keys,
+            uint32(result.last_index),
+        )
+
+    def to_standard(self) -> CreateMorePuzzleHashesResult:
+        return CreateMorePuzzleHashesResult(
+            [path.to_standard() for path in self.derivation_paths],
+            self.mark_existing_as_used,
+            self.unused,
+            self.new_unhardened_keys,
+            self.last_index,
+        )
+
+
+@dataclasses.dataclass
+class CreateMorePuzzleHashesResult:
+    derivation_paths: list[DerivationRecord]
+    mark_existing_as_used: bool
+    unused: int  # The first unused puzzle hash
+    new_unhardened_keys: bool
+    last_index: int  # The index we derived up to
+
+    async def commit(self, wallet_state_manager: WalletStateManager) -> None:
+        if len(self.derivation_paths) > 0:
+            await wallet_state_manager.puzzle_store.add_derivation_paths(self.derivation_paths)
+            await wallet_state_manager.wallet_node.new_peak_queue.subscribe_to_puzzle_hashes(
+                [
+                    record.puzzle_hash
+                    for record in self.derivation_paths
+                    if record.wallet_id == wallet_state_manager.main_wallet.id()
+                ]
+            )
+        if self.new_unhardened_keys:
+            wallet_state_manager.state_changed("new_derivation_index", data_object={"index": self.last_index - 1})
+        # By default, we'll mark previously generated unused puzzle hashes as used if we have new paths
+        if self.mark_existing_as_used and self.unused > 0 and self.new_unhardened_keys:
+            wallet_state_manager.log.info(f"Updating last used derivation index: {self.unused - 1}")
+            await wallet_state_manager.puzzle_store.set_used_up_to(uint32(self.unused - 1))
+
+
+@streamable
+@dataclasses.dataclass(frozen=True)
+class StreambleGetUnusedDerivationRecordResult(Streamable):
+    record: StreamableDerivationRecord
+    create_more_puzzle_hashes_result: StreamableCreateMorePuzzleHashesResult
+
+    @classmethod
+    def from_standard(cls, result: GetUnusedDerivationRecordResult) -> StreambleGetUnusedDerivationRecordResult:
+        return cls(
+            StreamableDerivationRecord.from_standard(result.record),
+            StreamableCreateMorePuzzleHashesResult.from_standard(result.create_more_puzzle_hashes_result),
+        )
+
+    def to_standard(self) -> GetUnusedDerivationRecordResult:
+        return GetUnusedDerivationRecordResult(
+            self.record.to_standard(),
+            self.create_more_puzzle_hashes_result.to_standard(),
+        )
+
+
+@dataclasses.dataclass
+class GetUnusedDerivationRecordResult:
+    record: DerivationRecord
+    create_more_puzzle_hashes_result: CreateMorePuzzleHashesResult
+
+    async def commit(self, wallet_state_manager: WalletStateManager) -> None:
+        await self.create_more_puzzle_hashes_result.commit(wallet_state_manager)
+        await wallet_state_manager.puzzle_store.set_used_up_to(self.record.index)

{chia_blockchain-2.5.4rc2.dist-info → chia_blockchain-2.5.5rc2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: chia-blockchain
-Version: 2.5.4rc2
+Version: 2.5.5rc2
 Summary: Chia blockchain full node, farmer, timelord, and wallet.
 License: Apache-2.0
 Keywords: chia,blockchain,node
@@ -26,12 +26,12 @@ Requires-Dist: bitstring (>=4.1.4)
 Requires-Dist: boto3 (>=1.35.43)
 Requires-Dist: build (>=1.2.1) ; extra == "dev"
 Requires-Dist: chia-puzzles-py (>=0.20.1)
-Requires-Dist: chia_rs (>=0.21.1,<0.22)
+Requires-Dist: chia_rs (>=0.27,<0.28)
 Requires-Dist: chiabip158 (>=1.5.2)
 Requires-Dist: chiapos (>=2.0.10)
 Requires-Dist: chiavdf (>=1.1.10)
 Requires-Dist: click (>=8.1.7)
-Requires-Dist: clvm (>=0.9.11)
+Requires-Dist: clvm (>=0.9.14)
 Requires-Dist: clvm_tools (>=0.4.9)
 Requires-Dist: clvm_tools_rs (>=0.1.45)
 Requires-Dist: colorama (>=0.4.6)
@@ -48,7 +48,7 @@ Requires-Dist: importlib-resources (>=6.4.5)
 Requires-Dist: keyring (>=25.5.0)
 Requires-Dist: keyrings.cryptfile (>=1.3.9) ; extra == "legacy-keyring"
 Requires-Dist: lxml (>=5.2.2) ; extra == "dev"
-Requires-Dist: miniupnpc (>=2.2.2) ; extra == "upnp"
+Requires-Dist: miniupnpc (>=2.3.2,<3) ; extra == "upnp"
 Requires-Dist: mypy (>=1.11.1) ; extra == "dev"
 Requires-Dist: packaging (>=24.0)
 Requires-Dist: pip (>=24.2)
@@ -66,6 +66,7 @@ Requires-Dist: ruff (>=0.8.1) ; extra == "dev"
 Requires-Dist: setproctitle (>=1.3.3)
 Requires-Dist: setuptools (>=75.5.0)
 Requires-Dist: sortedcontainers (>=2.4.0)
+Requires-Dist: tach (>=0.29.0) ; extra == "dev"
 Requires-Dist: types-aiofiles (>=24.1.0.20240626) ; extra == "dev"
 Requires-Dist: types-cryptography (>=3.3.23.2) ; extra == "dev"
 Requires-Dist: types-pyyaml (>=6.0.12.20240917) ; extra == "dev"
@@ -74,9 +75,8 @@ Requires-Dist: typing-extensions (>=4.12.2)
 Requires-Dist: watchdog (>=4.0.1)
 Requires-Dist: zstd (>=1.5.5.1) ; python_version < "3.12"
 Requires-Dist: zstd (>=1.5.5.1) ; python_version == "3.12"
-Project-URL: Changelog, https://github.com/Chia-Network/chia-blockchain/blob/main/CHANGELOG.md
-Project-URL: Homepage, https://chia.net/
-Project-URL: Source, https://github.com/Chia-Network/chia-blockchain/
+Project-URL: changelog, https://github.com/Chia-Network/chia-blockchain/blob/main/CHANGELOG.md
+Project-URL: source, https://github.com/Chia-Network/chia-blockchain/
 Description-Content-Type: text/markdown
 
 # chia-blockchain