import json import click from pathlib import Path import logging ################################################## # Import package modules ################################################# from iottb.utils.logger_config import setup_logging from iottb import definitions from iottb.contexts import IottbConfig from iottb.commands.initialize_testbed import init_db ############################################################################ # Module shortcuts for global definitions ########################################################################### APP_NAME = definitions.APP_NAME DB_NAME = definitions.DB_NAME CFG_FILE_PATH = definitions.CFG_FILE_PATH # These are (possibly) redundant when defined in definitions.py # keeping them here until refactored and tested MAX_VERBOSITY = definitions.MAX_VERBOSITY # Logger stuff loglevel = definitions.LOGLEVEL logger = logging.getLogger(__name__) @click.group() @click.option('-v', '--verbosity', count=True, type=click.IntRange(0, 3), default=0, help='Set verbosity') @click.option('-d', '--debug', is_flag=True, default=False, help='Enable debug mode') @click.option('--cfg-file', type=click.Path(), default=Path(click.get_app_dir(APP_NAME)).joinpath('iottb.cfg'), envvar='IOTTB_CONF_HOME', help='Path to iottb config file') @click.pass_context def cli(ctx, verbosity, debug, cfg_file): setup_logging(verbosity, debug) # Setup logging based on the loaded configuration and other options ctx.ensure_object(dict) # Make sure context is ready for use logger.info("Starting execution.") ctx.obj['CONFIG'] = IottbConfig(cfg_file) # Load configuration directly @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.") pass @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.option('--db', default=str(Path.home() / 'iottb.db'), help='Delete database') @click.confirmation_option(prompt="Are you certain that you want to delete the database file?") def rm_db(db): """ Removes database 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(db).unlink() click.echo(f'Iottb configuration removed at {CFG_FILE_PATH}') ################################################################################## # Add all subcommands to group here ################################################################################# cli.add_command(init_db) cli.add_command(rm_cfg) cli.add_command(set_key_in_table_to) cli.add_command(rm_db) if __name__ == '__main__': cli(auto_envvar_prefix='IOTTB')