mirror of
https://github.com/Chia-Network/chia-blockchain.git
synced 2026-08-29 02:24:35 -05:00
Matt HauffandGitHub
40db4635a8
* Enable PEP604 Ruff rules * Fix harcoded signature in test * Hack CLVMStreamable test with note to fast follow
118 lines
3.9 KiB
Python
118 lines
3.9 KiB
Python
from __future__ import annotations
|
|
|
|
import os
|
|
import pathlib
|
|
import sys
|
|
from multiprocessing import freeze_support
|
|
from typing import Any
|
|
|
|
from chia_rs import ConsensusConstants
|
|
|
|
from chia.apis import StubMetadataRegistry
|
|
from chia.consensus.constants import replace_str_to_bytes
|
|
from chia.consensus.default_constants import DEFAULT_CONSTANTS, update_testnet_overrides
|
|
from chia.protocols.outbound_message import NodeType
|
|
from chia.server.resolve_peer_info import get_unresolved_peer_infos
|
|
from chia.server.signal_handlers import SignalHandlers
|
|
from chia.server.start_service import RpcInfo, Service, async_run
|
|
from chia.util.chia_logging import initialize_service_logging
|
|
from chia.util.config import load_config, load_config_cli
|
|
from chia.util.default_root import resolve_root_path
|
|
from chia.util.keychain import Keychain
|
|
from chia.util.task_timing import maybe_manage_task_instrumentation
|
|
from chia.wallet.wallet_node import WalletNode
|
|
|
|
# See: https://bugs.python.org/issue29288
|
|
from chia.wallet.wallet_node_api import WalletNodeAPI
|
|
from chia.wallet.wallet_rpc_api import WalletRpcApi
|
|
from chia.wallet.wallet_service import WalletService
|
|
|
|
"".encode("idna")
|
|
|
|
SERVICE_NAME = "wallet"
|
|
|
|
|
|
def create_wallet_service(
|
|
root_path: pathlib.Path,
|
|
config: dict[str, Any],
|
|
consensus_constants: ConsensusConstants,
|
|
keychain: Keychain | None = None,
|
|
connect_to_daemon: bool = True,
|
|
) -> WalletService:
|
|
service_config = config[SERVICE_NAME]
|
|
|
|
network_id = service_config["selected_network"]
|
|
overrides = service_config["network_overrides"]["constants"][network_id]
|
|
update_testnet_overrides(network_id, overrides)
|
|
updated_constants = replace_str_to_bytes(consensus_constants, **overrides)
|
|
service_config.setdefault("short_sync_blocks_behind_threshold", 20)
|
|
|
|
node = WalletNode(
|
|
service_config,
|
|
root_path,
|
|
constants=updated_constants,
|
|
local_keychain=keychain,
|
|
)
|
|
peer_api = WalletNodeAPI(node)
|
|
|
|
rpc_info: RpcInfo[WalletRpcApi] | None = None
|
|
if service_config.get("start_rpc_server", True):
|
|
rpc_info = (WalletRpcApi, service_config["rpc_port"])
|
|
|
|
return Service(
|
|
root_path=root_path,
|
|
config=config,
|
|
node=node,
|
|
peer_api=peer_api,
|
|
node_type=NodeType.WALLET,
|
|
advertised_port=None,
|
|
service_name=SERVICE_NAME,
|
|
connect_peers=get_unresolved_peer_infos(service_config, NodeType.FULL_NODE),
|
|
on_connect_callback=node.on_connect,
|
|
network_id=network_id,
|
|
rpc_info=rpc_info,
|
|
connect_to_daemon=connect_to_daemon,
|
|
stub_metadata_for_type=StubMetadataRegistry,
|
|
)
|
|
|
|
|
|
async def async_main(root_path: pathlib.Path) -> int:
|
|
# TODO: refactor to avoid the double load
|
|
config = load_config(root_path, "config.yaml")
|
|
service_config = load_config_cli(root_path, "config.yaml", SERVICE_NAME)
|
|
config[SERVICE_NAME] = service_config
|
|
|
|
# This is simulator
|
|
local_test = service_config.get("testing", False)
|
|
if local_test is True:
|
|
from chia.simulator.block_tools import test_constants
|
|
|
|
constants = test_constants
|
|
current = service_config["database_path"]
|
|
service_config["database_path"] = f"{current}_simulation"
|
|
service_config["selected_network"] = "testnet0"
|
|
else:
|
|
constants = DEFAULT_CONSTANTS
|
|
initialize_service_logging(service_name=SERVICE_NAME, config=config, root_path=root_path)
|
|
|
|
service = create_wallet_service(root_path, config, constants)
|
|
async with SignalHandlers.manage() as signal_handlers:
|
|
await service.setup_process_global_state(signal_handlers=signal_handlers)
|
|
await service.run()
|
|
|
|
return 0
|
|
|
|
|
|
def main() -> int:
|
|
freeze_support()
|
|
root_path = resolve_root_path(override=None)
|
|
|
|
with maybe_manage_task_instrumentation(
|
|
enable=os.environ.get(f"CHIA_INSTRUMENT_{SERVICE_NAME.upper()}") is not None
|
|
):
|
|
return async_run(coro=async_main(root_path=root_path))
|
|
|
|
|
|
if __name__ == "__main__":
|
|
sys.exit(main())
|