Module rugged.lib.cli

Functions

def print_memory_usage()

Expand source code

def print_memory_usage():
    """ Print memory usage. """
    current, peak = tracemalloc.get_traced_memory()
    tracemalloc.stop()
    log.debug(f"Peak memory allocation: {peak / 1024:,.2f}KiB")
    log.debug(f"Final memory allocation: {current / 1024:,.2f}KiB")

Print memory usage.

def register_subcommands() ‑> None

Expand source code

def register_subcommands() -> None:
    """Add sub-commands to the CLI tool."""
    try:
        rugged_cli.add_command(add_targets_cmd)
        rugged_cli.add_command(add_root_signature_cmd)
        rugged_cli.add_command(add_verification_key_cmd)
        rugged_cli.add_command(check_monitor_cmd)
        rugged_cli.add_command(config_cmd)
        rugged_cli.add_command(deploy_root_metadata_cmd)
        rugged_cli.add_command(echo_cmd)
        rugged_cli.add_command(generate_keys_cmd)
        rugged_cli.add_command(initialize_cmd)
        rugged_cli.add_command(initialize_partial_root_metadata_cmd)
        rugged_cli.add_command(initialize_partial_root_metadata_for_update_cmd)
        rugged_cli.add_command(logs_cmd)
        rugged_cli.add_command(pause_processing_cmd)
        rugged_cli.add_command(refresh_expiry_cmd)
        rugged_cli.add_command(remove_targets_cmd)
        rugged_cli.add_command(remove_verification_key_cmd)
        rugged_cli.add_command(resume_processing_cmd)
        rugged_cli.add_command(reset_processing_cmd)
        rugged_cli.add_command(reset_refreshing_cmd)
        rugged_cli.add_command(rotate_keys_cmd)
        rugged_cli.add_command(show_partial_root_metadata_cmd)
        rugged_cli.add_command(sleep_cmd)
        rugged_cli.add_command(status_cmd)
        rugged_cli.add_command(update_hashed_bins_count_cmd)
        rugged_cli.add_command(validate_cmd)
    except FileNotFoundError:
        sys.exit(os.EX_NOINPUT)
    except RuggedKeyError:
        sys.exit(os.EX_USAGE)

Add sub-commands to the CLI tool.