93 lines
3.4 KiB
Python
93 lines
3.4 KiB
Python
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')
|