mirror of
https://github.com/Chia-Network/chia-blockchain.git
synced 2026-09-05 02:24:21 -05:00
neurosis69andGitHub
157f462598
* first chia db backup iteration * add required attribute * fix type mismatch * added no_indexes option * black, flake8 * fix typo * add missing detach command * add error message and sqlite version handling * black
81 lines
2.8 KiB
Python
81 lines
2.8 KiB
Python
from pathlib import Path
|
|
import click
|
|
from chia.cmds.db_upgrade_func import db_upgrade_func
|
|
from chia.cmds.db_validate_func import db_validate_func
|
|
from chia.cmds.db_backup_func import db_backup_func
|
|
|
|
|
|
@click.group("db", short_help="Manage the blockchain database")
|
|
def db_cmd() -> None:
|
|
pass
|
|
|
|
|
|
@db_cmd.command("upgrade", short_help="upgrade a v1 database to v2")
|
|
@click.option("--input", default=None, type=click.Path(), help="specify input database file")
|
|
@click.option("--output", default=None, type=click.Path(), help="specify output database file")
|
|
@click.option(
|
|
"--no-update-config",
|
|
default=False,
|
|
is_flag=True,
|
|
help="don't update config file to point to new database. When specifying a "
|
|
"custom output file, the config will not be updated regardless",
|
|
)
|
|
@click.option(
|
|
"--force",
|
|
default=False,
|
|
is_flag=True,
|
|
help="force conversion despite warnings",
|
|
)
|
|
@click.pass_context
|
|
def db_upgrade_cmd(ctx: click.Context, no_update_config: bool, force: bool, **kwargs) -> None:
|
|
|
|
try:
|
|
in_db_path = kwargs.get("input")
|
|
out_db_path = kwargs.get("output")
|
|
db_upgrade_func(
|
|
Path(ctx.obj["root_path"]),
|
|
None if in_db_path is None else Path(in_db_path),
|
|
None if out_db_path is None else Path(out_db_path),
|
|
no_update_config=no_update_config,
|
|
force=force,
|
|
)
|
|
except RuntimeError as e:
|
|
print(f"FAILED: {e}")
|
|
|
|
|
|
@db_cmd.command("validate", short_help="validate the (v2) blockchain database. Does not verify proofs")
|
|
@click.option("--db", default=None, type=click.Path(), help="Specifies which database file to validate")
|
|
@click.option(
|
|
"--validate-blocks",
|
|
default=False,
|
|
is_flag=True,
|
|
help="validate consistency of properties of the encoded blocks and block records",
|
|
)
|
|
@click.pass_context
|
|
def db_validate_cmd(ctx: click.Context, validate_blocks: bool, **kwargs) -> None:
|
|
try:
|
|
in_db_path = kwargs.get("db")
|
|
db_validate_func(
|
|
Path(ctx.obj["root_path"]),
|
|
None if in_db_path is None else Path(in_db_path),
|
|
validate_blocks=validate_blocks,
|
|
)
|
|
except RuntimeError as e:
|
|
print(f"FAILED: {e}")
|
|
|
|
|
|
@db_cmd.command("backup", short_help="backup the blockchain database using VACUUM INTO command")
|
|
@click.option("--backup_file", default=None, type=click.Path(), help="Specifies the backup file")
|
|
@click.option("--no_indexes", default=False, is_flag=True, help="Create backup without indexes")
|
|
@click.pass_context
|
|
def db_backup_cmd(ctx: click.Context, no_indexes: bool, **kwargs) -> None:
|
|
try:
|
|
db_backup_file = kwargs.get("backup_file")
|
|
db_backup_func(
|
|
Path(ctx.obj["root_path"]),
|
|
None if db_backup_file is None else Path(db_backup_file),
|
|
no_indexes=no_indexes,
|
|
)
|
|
except RuntimeError as e:
|
|
print(f"FAILED: {e}")
|