79 lines
2.8 KiB
Python
79 lines
2.8 KiB
Python
from pathlib import Path
|
|
|
|
import click
|
|
|
|
from iottb.main import DB_NAME, logger, CFG_FILE_PATH
|
|
from iottb.models.iottb_config import IottbConfig
|
|
|
|
|
|
@click.command()
|
|
@click.option('--file', default=DB_NAME)
|
|
@click.option('--table', type=str, default='DefaultDatabase')
|
|
@click.option('--key')
|
|
@click.option('--value')
|
|
@click.pass_context
|
|
def set_key_in_table_to(ctx, file, table, key, value):
|
|
"""Edit config or metadata files. TODO: Implement"""
|
|
click.echo(f'set_key_in_table_to invoked')
|
|
logger.warning("Unimplemented subcommand invoked.")
|
|
|
|
|
|
@click.command()
|
|
@click.confirmation_option(prompt="Are you certain that you want to delete the cfg file?")
|
|
def rm_cfg():
|
|
""" Removes the cfg file from the filesystem.
|
|
|
|
This is mostly a utility during development. Once non-standard database locations are implemented,
|
|
deleting this would lead to iottb not being able to find them anymore.
|
|
"""
|
|
Path(CFG_FILE_PATH).unlink()
|
|
click.echo(f'Iottb configuration removed at {CFG_FILE_PATH}')
|
|
|
|
|
|
@click.command()
|
|
@click.confirmation_option(prompt="Are you certain that you want to delete the databases file?")
|
|
def rm_dbs(dbs):
|
|
""" Removes ALL(!) databases from the filesystem if they're empty.
|
|
|
|
Development utility currently unfit for use.
|
|
"""
|
|
config = IottbConfig()
|
|
paths = config.get_know_database_paths()
|
|
logger.debug(f'Known db paths: {str(paths)}')
|
|
for dbs in paths:
|
|
try:
|
|
Path(dbs).rmdir()
|
|
click.echo(f'{dbs} deleted')
|
|
except Exception as e:
|
|
logger.debug(f'Failed unlinking db {dbs} with error {e}')
|
|
logger.info(f'All databases deleted')
|
|
|
|
|
|
@click.command('show-cfg', help='Show the current configuration context')
|
|
@click.option('--cfg-file', type=click.Path(), default=CFG_FILE_PATH, help='Path to the config file')
|
|
@click.option('-pp', is_flag=True, default=False, help='Pretty Print')
|
|
@click.pass_context
|
|
def show_cfg(ctx, cfg_file, pp):
|
|
logger.debug(f'Pretty print option set to {pp}')
|
|
if pp:
|
|
try:
|
|
config = IottbConfig(Path(cfg_file))
|
|
click.echo("Configuration Context:")
|
|
click.echo(f"Default Database: {config.default_database}")
|
|
click.echo(f"Default Database Path: {config.default_db_location}")
|
|
click.echo("Database Locations:")
|
|
for db_name, db_path in config.db_path_dict.items():
|
|
click.echo(f" - {db_name}: {db_path}")
|
|
except Exception as e:
|
|
logger.error(f"Error loading configuration: {e}")
|
|
click.echo(f"Failed to load configuration from {cfg_file}")
|
|
else:
|
|
path = Path(cfg_file)
|
|
|
|
if path.is_file():
|
|
with path.open('r') as file:
|
|
content = file.read()
|
|
click.echo(content)
|
|
else:
|
|
click.echo(f"Configuration file not found at {cfg_file}")
|