dotfile-automator/project/config/cli.py

import click
from project.utils import config


@click.group('config')
def cli():
    pass


@cli.command('show')
@click.argument('key', nargs=-1)
def show(key):
    key = " ".join(key)
    print(config.get(key))
    return 0


@cli.command('set')
@click.argument('key', nargs=1)
@click.argument('value', nargs=-1)
def set(key, value):
    if type(value) == type(()):
        value = " ".join(value)
    status = config.set(key, value)
    if status == "SUCCESS":
        print("{} set to {}".format(key, value))
        return 0
    elif status == "NO_KEY":
        print("Key {} not found".format(key))
        return 1
    else:
        print("Something went wrong:", status)
        return 1