chia-blockchain 2.4.4__py3-none-any.whl

chia/cmds/plots.py

@@ -0,0 +1,222 @@
+from __future__ import annotations
+
+import asyncio
+import logging
+import sys
+from pathlib import Path
+
+import click
+
+from chia.plotting.util import add_plot_directory, validate_plot_size
+
+log = logging.getLogger(__name__)
+
+
+def show_plots(root_path: Path) -> None:
+    from chia.plotting.util import get_plot_directories
+
+    print("Directories where plots are being searched for:")
+    print("Note that subdirectories must be added manually")
+    print(
+        "Add with 'chia plots add -d [dir]' and remove with"
+        + " 'chia plots remove -d [dir]'"
+        + " Scan and check plots with 'chia plots check'"
+    )
+    print()
+    for str_path in get_plot_directories(root_path):
+        print(f"{str_path}")
+
+
+@click.group("plots", help="Manage your plots")
+@click.pass_context
+def plots_cmd(ctx: click.Context) -> None:
+    """Create, add, remove and check your plots"""
+    from chia.util.chia_logging import initialize_logging
+
+    root_path: Path = ctx.obj["root_path"]
+    if not root_path.is_dir():
+        raise RuntimeError("Please initialize (or migrate) your config directory with 'chia init'")
+    initialize_logging("", {"log_level": "INFO", "log_stdout": True}, root_path)
+
+
+@plots_cmd.command("create", help="Create plots")
+@click.option("-k", "--size", help="Plot size", type=int, default=32, show_default=True)
+@click.option("--override-k", help="Force size smaller than 32", default=False, show_default=True, is_flag=True)
+@click.option("-n", "--num", help="Number of plots or challenges", type=int, default=1, show_default=True)
+@click.option("-b", "--buffer", help="Megabytes for sort/plot buffer", type=int, default=3389, show_default=True)
+@click.option("-r", "--num_threads", help="Number of threads to use", type=int, default=2, show_default=True)
+@click.option("-u", "--buckets", help="Number of buckets", type=int, default=128, show_default=True)
+@click.option(
+    "-a",
+    "--alt_fingerprint",
+    type=int,
+    default=None,
+    help="Enter the alternative fingerprint of the key you want to use",
+)
+@click.option(
+    "-c",
+    "--pool_contract_address",
+    type=str,
+    default=None,
+    help="Address of where the pool reward will be sent to. Only used if alt_fingerprint and pool public key are None",
+)
+@click.option("-f", "--farmer_public_key", help="Hex farmer public key", type=str, default=None)
+@click.option("-p", "--pool_public_key", help="Hex public key of pool", type=str, default=None)
+@click.option(
+    "-t",
+    "--tmp_dir",
+    help="Temporary directory for plotting files",
+    type=click.Path(),
+    default=Path("."),
+    show_default=True,
+)
+@click.option("-2", "--tmp2_dir", help="Second temporary directory for plotting files", type=click.Path(), default=None)
+@click.option(
+    "-d",
+    "--final_dir",
+    help="Final directory for plots (relative or absolute)",
+    type=click.Path(),
+    default=Path("."),
+    show_default=True,
+)
+@click.option("-i", "--plotid", help="PlotID in hex for reproducing plots (debugging only)", type=str, default=None)
+@click.option("-m", "--memo", help="Memo in hex for reproducing plots (debugging only)", type=str, default=None)
+@click.option("-e", "--nobitfield", help="Disable bitfield", default=False, is_flag=True)
+@click.option(
+    "-x", "--exclude_final_dir", help="Skips adding [final dir] to harvester for farming", default=False, is_flag=True
+)
+@click.option(
+    "-D",
+    "--connect_to_daemon",
+    help="Connects to the daemon for keychain operations",
+    default=False,
+    is_flag=True,
+    hidden=True,  # -D is only set when launched by the daemon
+)
+@click.pass_context
+def create_cmd(
+    ctx: click.Context,
+    size: int,
+    override_k: bool,
+    num: int,
+    buffer: int,
+    num_threads: int,
+    buckets: int,
+    alt_fingerprint: int,
+    pool_contract_address: str,
+    farmer_public_key: str,
+    pool_public_key: str,
+    tmp_dir: str,
+    tmp2_dir: str,
+    final_dir: str,
+    plotid: str,
+    memo: str,
+    nobitfield: bool,
+    exclude_final_dir: bool,
+    connect_to_daemon: bool,
+) -> None:
+    from chia.plotting.create_plots import create_plots, resolve_plot_keys
+    from chia.plotting.util import Params
+
+    params = Params(
+        size=size,
+        num=num,
+        buffer=buffer,
+        num_threads=num_threads,
+        buckets=buckets,
+        tmp_dir=Path(tmp_dir),
+        tmp2_dir=Path(tmp2_dir) if tmp2_dir else None,
+        final_dir=Path(final_dir),
+        plotid=plotid,
+        memo=memo,
+        nobitfield=nobitfield,
+    )
+
+    root_path: Path = ctx.obj["root_path"]
+    try:
+        validate_plot_size(root_path, size, override_k)
+    except ValueError as e:
+        print(e)
+        sys.exit(1)
+
+    plot_keys = asyncio.run(
+        resolve_plot_keys(
+            farmer_public_key,
+            alt_fingerprint,
+            pool_public_key,
+            pool_contract_address,
+            root_path,
+            log,
+            connect_to_daemon,
+        )
+    )
+
+    asyncio.run(create_plots(params, plot_keys))
+    if not exclude_final_dir:
+        try:
+            add_plot_directory(root_path, final_dir)
+        except ValueError as e:
+            print(e)
+
+
+@plots_cmd.command("check", help="Checks plots")
+@click.option("-n", "--num", help="Number of plots or challenges", type=int, default=None)
+@click.option(
+    "-g",
+    "--grep_string",
+    help="Shows only plots that contain the string in the filename or directory name",
+    type=str,
+    default=None,
+)
+@click.option("-l", "--list_duplicates", help="List plots with duplicate IDs", default=False, is_flag=True)
+@click.option("--debug-show-memo", help="Shows memo to recreate the same exact plot", default=False, is_flag=True)
+@click.option("--challenge-start", help="Begins at a different [start] for -n [challenges]", type=int, default=None)
+@click.pass_context
+def check_cmd(
+    ctx: click.Context, num: int, grep_string: str, list_duplicates: bool, debug_show_memo: bool, challenge_start: int
+) -> None:
+    from chia.plotting.check_plots import check_plots
+
+    check_plots(ctx.obj["root_path"], num, challenge_start, grep_string, list_duplicates, debug_show_memo)
+
+
+@plots_cmd.command("add", help="Adds a directory of plots")
+@click.option(
+    "-d",
+    "--final_dir",
+    help="Final directory for plots (relative or absolute)",
+    type=click.Path(),
+    default=".",
+    show_default=True,
+)
+@click.pass_context
+def add_cmd(ctx: click.Context, final_dir: str) -> None:
+    from chia.plotting.util import add_plot_directory
+
+    try:
+        add_plot_directory(ctx.obj["root_path"], final_dir)
+        print(f"Successfully added: {final_dir}")
+    except ValueError as e:
+        print(e)
+
+
+@plots_cmd.command("remove", help="Removes a directory of plots from config.yaml")
+@click.option(
+    "-d",
+    "--final_dir",
+    help="Final directory for plots (relative or absolute)",
+    type=click.Path(),
+    default=".",
+    show_default=True,
+)
+@click.pass_context
+def remove_cmd(ctx: click.Context, final_dir: str) -> None:
+    from chia.plotting.util import remove_plot_directory
+
+    remove_plot_directory(ctx.obj["root_path"], final_dir)
+
+
+@plots_cmd.command("show", help="Shows the directory of current plots")
+@click.pass_context
+def show_cmd(ctx: click.Context) -> None:
+    show_plots(ctx.obj["root_path"])

chia/cmds/plotters.py

@@ -0,0 +1,17 @@
+from __future__ import annotations
+
+import click
+
+from chia.plotters.plotters import call_plotters
+
+
+@click.command(
+    "plotters",
+    help="Advanced plotting options",
+    context_settings={"ignore_unknown_options": True},
+    add_help_option=False,
+)
+@click.pass_context
+@click.argument("args", nargs=-1)
+def plotters_cmd(ctx: click.Context, args: tuple[click.Argument]) -> None:
+    call_plotters(ctx.obj["root_path"], args)

chia/cmds/rpc.py

@@ -0,0 +1,188 @@
+from __future__ import annotations
+
+import asyncio
+import json
+import sys
+from typing import Any, Dict, List, Optional, TextIO
+
+import click
+from aiohttp import ClientResponseError
+
+from chia.util.config import load_config
+from chia.util.default_root import DEFAULT_ROOT_PATH
+from chia.util.ints import uint16
+
+services: List[str] = ["crawler", "daemon", "farmer", "full_node", "harvester", "timelord", "wallet", "data_layer"]
+
+
+async def call_endpoint(
+    service: str, endpoint: str, request: Dict[str, Any], config: Dict[str, Any], quiet: bool = False
+) -> Dict[str, Any]:
+    if service == "daemon":
+        return await call_daemon_command(endpoint, request, config, quiet)
+
+    return await call_rpc_service_endpoint(service, endpoint, request, config)
+
+
+async def call_rpc_service_endpoint(
+    service: str, endpoint: str, request: Dict[str, Any], config: Dict[str, Any]
+) -> Dict[str, Any]:
+    from chia.rpc.rpc_client import RpcClient
+
+    port: uint16
+    if service == "crawler":
+        # crawler config is inside the seeder config
+        port = uint16(config["seeder"][service]["rpc_port"])
+    else:
+        port = uint16(config[service]["rpc_port"])
+
+    try:
+        client = await RpcClient.create(config["self_hostname"], port, DEFAULT_ROOT_PATH, config)
+    except Exception as e:
+        raise Exception(f"Failed to create RPC client {service}: {e}")
+    result: Dict[str, Any]
+    try:
+        result = await client.fetch(endpoint, request)
+    except ClientResponseError as e:
+        if e.code == 404:
+            raise Exception(f"Invalid endpoint for {service}: {endpoint}")
+        raise
+    except Exception as e:
+        raise Exception(f"Request failed: {e}")
+    finally:
+        client.close()
+        await client.await_closed()
+    return result
+
+
+async def call_daemon_command(
+    command: str, request: Dict[str, Any], config: Dict[str, Any], quiet: bool = False
+) -> Dict[str, Any]:
+    from chia.daemon.client import connect_to_daemon_and_validate
+
+    daemon = await connect_to_daemon_and_validate(DEFAULT_ROOT_PATH, config, quiet=quiet)
+
+    if daemon is None:
+        raise Exception("Failed to connect to chia daemon")
+
+    result: Dict[str, Any]
+    try:
+        ws_request = daemon.format_request(command, request)
+        ws_response = await daemon._get(ws_request)
+        result = ws_response["data"]
+    except Exception as e:
+        raise Exception(f"Request failed: {e}")
+    finally:
+        await daemon.close()
+    return result
+
+
+def print_result(json_dict: Dict[str, Any]) -> None:
+    print(json.dumps(json_dict, indent=2, sort_keys=True))
+
+
+def get_routes(service: str, config: Dict[str, Any], quiet: bool = False) -> Dict[str, Any]:
+    return asyncio.run(call_endpoint(service, "get_routes", {}, config, quiet))
+
+
+@click.group("rpc", help="RPC Client")
+def rpc_cmd() -> None:
+    pass
+
+
+@rpc_cmd.command("endpoints", help="Print all endpoints of a service")
+@click.argument("service", type=click.Choice(services))
+def endpoints_cmd(service: str) -> None:
+    config = load_config(DEFAULT_ROOT_PATH, "config.yaml")
+    try:
+        routes = get_routes(service, config)
+        for route in routes["routes"]:
+            print(route.lstrip("/"))
+    except Exception as e:
+        print(e)
+
+
+@rpc_cmd.command("status", help="Print the status of all available RPC services")
+@click.option("--json-output", "json_output", is_flag=True, help="Output status as JSON")
+def status_cmd(json_output: bool) -> None:
+    import json
+
+    config = load_config(DEFAULT_ROOT_PATH, "config.yaml")
+
+    def print_row(c0: str, c1: str) -> None:
+        print(f"│ {c0:<12} │ {c1:<9} │")
+
+    status_data = {}
+    for service in services:
+        status = "ACTIVE"
+        try:
+            if not get_routes(service, config, quiet=True)["success"]:
+                raise Exception()
+        except Exception:
+            status = "INACTIVE"
+        status_data[service] = status
+
+    if json_output:
+        # If --json-output option is used, print the status data as JSON
+        print(json.dumps(status_data, indent=2))
+    else:
+        print("╭──────────────┬───────────╮")
+        print_row("SERVICE", "STATUS")
+        print("├──────────────┼───────────┤")
+        for service, status in status_data.items():
+            print_row(service, status)
+            if service != services[-1]:  # Don't print the separator after the last service
+                print("├──────────────┼───────────┤")
+        print("╰──────────────┴───────────╯")
+
+
+def create_commands() -> None:
+    for service in services:
+
+        @rpc_cmd.command(
+            service,
+            short_help=f"RPC client for the {service} RPC API",
+            help=(
+                f"Call ENDPOINT (RPC endpoint as as string) of the {service} "
+                "RPC API with REQUEST (must be a JSON string) as request data."
+            ),
+        )
+        @click.argument("endpoint", type=str)
+        @click.argument("request", type=str, required=False)
+        @click.option(
+            "-j",
+            "--json-file",
+            help="Optionally instead of REQUEST you can provide a json file containing the request data",
+            type=click.File("r"),
+            default=None,
+        )
+        def rpc_client_cmd(
+            endpoint: str, request: Optional[str], json_file: Optional[TextIO], service: str = service
+        ) -> None:
+            config = load_config(DEFAULT_ROOT_PATH, "config.yaml")
+            if request is not None and json_file is not None:
+                sys.exit(
+                    "Can only use one request source: REQUEST argument OR -j/--json-file option. See the help with -h"
+                )
+
+            request_json: Dict[str, Any] = {}
+            if json_file is not None:
+                try:
+                    request_json = json.load(json_file)
+                except Exception as e:
+                    sys.exit(f"Invalid JSON file: {e}")
+            if request is not None:
+                try:
+                    request_json = json.loads(request)
+                except Exception as e:
+                    sys.exit(f"Invalid REQUEST JSON: {e}")
+
+            try:
+                if endpoint[0] == "/":
+                    endpoint = endpoint[1:]
+                print_result(asyncio.run(call_endpoint(service, endpoint, request_json, config)))
+            except Exception as e:
+                sys.exit(str(e))
+
+
+create_commands()

chia/cmds/show.py

@@ -0,0 +1,71 @@
+from __future__ import annotations
+
+from typing import Optional
+
+import click
+
+from chia.cmds.show_funcs import show_async
+
+
+@click.command("show", help="Show node information", no_args_is_help=True)
+@click.option(
+    "-p",
+    "--rpc-port",
+    help=(
+        "Set the port where the Full Node is hosting the RPC interface. "
+        "See the rpc_port under full_node in config.yaml"
+    ),
+    type=int,
+    default=None,
+)
+@click.option(
+    "-wp",
+    "--wallet-rpc-port",
+    help="Set the port where the Wallet is hosting the RPC interface. See the rpc_port under wallet in config.yaml",
+    type=int,
+    default=None,
+)
+@click.option("-f", "--fee", help="Show the fee information", is_flag=True, type=bool, default=False)
+@click.option("-s", "--state", help="Show the current state of the blockchain", is_flag=True, type=bool, default=False)
+@click.option(
+    "-c", "--connections", help="List nodes connected to this Full Node", is_flag=True, type=bool, default=False
+)
+@click.option("-a", "--add-connection", help="Connect to another Full Node by ip:port", type=str, default="")
+@click.option(
+    "-r", "--remove-connection", help="Remove a Node by the first 8 characters of NodeID", type=str, default=""
+)
+@click.option("-bh", "--block-header-hash-by-height", help="Look up a block header hash by block height", type=int)
+@click.option("-b", "--block-by-header-hash", help="Look up a block by block header hash", type=str, default="")
+@click.pass_context
+def show_cmd(
+    ctx: click.Context,
+    rpc_port: Optional[int],
+    wallet_rpc_port: Optional[int],
+    fee: bool,
+    state: bool,
+    connections: bool,
+    add_connection: str,
+    remove_connection: str,
+    block_header_hash_by_height: Optional[int],
+    block_by_header_hash: str,
+) -> None:
+    import asyncio
+
+    if connections:
+        print("'chia show -c' has been renamed to 'chia peer full_node -c' ")
+    if add_connection != "":
+        print("'chia show -a' has been renamed to 'chia peer full_node -a' ")
+    if remove_connection != "":
+        print("'chia show -r' has been renamed to 'chia peer full_node -r' ")
+    if wallet_rpc_port is not None:
+        print("'chia show -wp' is not used, please remove it from your command.")
+    asyncio.run(
+        show_async(
+            rpc_port,
+            ctx.obj["root_path"],
+            fee,
+            state,
+            block_header_hash_by_height,
+            block_by_header_hash,
+        )
+    )