mirror of
https://github.com/Chia-Network/chia-blockchain.git
synced 2026-08-29 18:14:26 -05:00
Matt HauffandGitHub
40db4635a8
* Enable PEP604 Ruff rules * Fix harcoded signature in test * Hack CLVMStreamable test with note to fast follow
78 lines
2.2 KiB
Python
78 lines
2.2 KiB
Python
from __future__ import annotations
|
|
|
|
import json
|
|
|
|
import click
|
|
|
|
from chia.cmds.cmd_classes import ChiaCliContext
|
|
from chia.cmds.dev.mempool_funcs import create_block_async, export_mempool_async, import_mempool_async
|
|
|
|
|
|
@click.group("mempool", help="Debug the mempool")
|
|
@click.pass_context
|
|
def mempool_cmd(ctx: click.Context) -> None:
|
|
pass
|
|
|
|
|
|
@click.command("import", help="Import mempool items from a JSON file", 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.argument("path", type=str)
|
|
@click.pass_context
|
|
def import_mempool_cmd(ctx: click.Context, rpc_port: int | None, path: str) -> None:
|
|
import asyncio
|
|
|
|
with open(path) as file:
|
|
source = file.read()
|
|
|
|
content = json.loads(source)
|
|
|
|
asyncio.run(import_mempool_async(rpc_port, ChiaCliContext.set_default(ctx).root_path, content))
|
|
|
|
|
|
@click.command("export", help="Export mempool items to a JSON file", 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.argument("path", type=str)
|
|
@click.pass_context
|
|
def export_mempool_cmd(ctx: click.Context, rpc_port: int | None, path: str) -> None:
|
|
import asyncio
|
|
|
|
asyncio.run(export_mempool_async(rpc_port, ChiaCliContext.set_default(ctx).root_path, path))
|
|
|
|
|
|
@click.command("create_block", help="Create a block bundle from the mempool, as if you farmed it")
|
|
@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.pass_context
|
|
def create_block_cmd(ctx: click.Context, rpc_port: int | None) -> None:
|
|
import asyncio
|
|
|
|
asyncio.run(create_block_async(rpc_port, ChiaCliContext.set_default(ctx).root_path))
|
|
|
|
|
|
mempool_cmd.add_command(import_mempool_cmd)
|
|
mempool_cmd.add_command(export_mempool_cmd)
|
|
mempool_cmd.add_command(create_block_cmd)
|