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
64 lines
1.9 KiB
Python
64 lines
1.9 KiB
Python
"""
|
|
NOTE: This contains duplicate code from `chia.cmds.plots`.
|
|
After `chia plots create` becomes obsolete, consider removing it from there.
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
import asyncio
|
|
import importlib.metadata
|
|
import logging
|
|
from argparse import Namespace
|
|
from pathlib import Path
|
|
from typing import Any
|
|
|
|
from chia.plotting.create_plots import create_plots, resolve_plot_keys
|
|
from chia.plotting.util import Params, add_plot_directory, validate_plot_size
|
|
|
|
log = logging.getLogger(__name__)
|
|
|
|
|
|
def get_chiapos_install_info() -> dict[str, Any] | None:
|
|
chiapos_version = importlib.metadata.version("chiapos")
|
|
return {"display_name": "Chia Proof of Space", "version": chiapos_version, "installed": True}
|
|
|
|
|
|
def plot_chia(args: Namespace, root_path: Path) -> None:
|
|
try:
|
|
validate_plot_size(root_path, args.size, args.override)
|
|
except ValueError as e:
|
|
print(e)
|
|
return
|
|
|
|
plot_keys = asyncio.run(
|
|
resolve_plot_keys(
|
|
None if args.farmerkey == b"" else args.farmerkey.hex(),
|
|
args.alt_fingerprint,
|
|
None if args.pool_key == b"" else args.pool_key.hex(),
|
|
None if args.contract == "" else args.contract,
|
|
root_path,
|
|
log,
|
|
args.connect_to_daemon,
|
|
)
|
|
)
|
|
params = Params(
|
|
size=args.size,
|
|
num=args.count,
|
|
buffer=args.buffer,
|
|
num_threads=args.threads,
|
|
buckets=args.buckets,
|
|
stripe_size=args.stripes,
|
|
tmp_dir=Path(args.tmpdir),
|
|
tmp2_dir=Path(args.tmpdir2) if args.tmpdir2 else None,
|
|
final_dir=Path(args.finaldir),
|
|
plotid=args.id,
|
|
memo=args.memo,
|
|
nobitfield=args.nobitfield,
|
|
)
|
|
asyncio.run(create_plots(params, plot_keys))
|
|
if not args.exclude_final_dir:
|
|
try:
|
|
add_plot_directory(root_path, args.finaldir)
|
|
except ValueError as e:
|
|
print(e)
|