taskwarrior-grocy/cli.py

import click
from pydoc import importfile, ErrorDuringImport
import yaml
from site import exit
from shutil import copy 
from os import path, chmod, makedirs
import stat

APP_NAME = 'twservices'
SAMPLE_CONFIG_FILE = 'sample.config.yml'
CONFIG_FILE = 'config.yml'
CONFIG_DIR = click.get_app_dir(APP_NAME)
PROJ_DIR = os.path.dirname(os.path.realpath(__file__))

def __create_config_file():
    user_cfg_file = path.join(CONFIG_DIR, CONFIG_FILE)
    sample_cfg_file = path.join(PROJ_DIR, SAMPLE_CONFIG)
    if not os.path.exists(CONFIG_DIR): 
        print('Config {} director does not exist, create...'.format(CONFIG_DIR))
        makedirs(CONFIG_DIR)

    copy(sample_cfg_file, user_cfg_file)
    print('Copying {} to {}'.format(sample_cfg_file, user_cfg_file))
    chmod(user_cfg_file, 0o664) 

    return user_cfg_file
    
    

    



    

@click.command()
@click.group()
def main():
    pass

def get_config_file()
    no_config_msg = 'A config file was not found ' + 
    'and will be created under {0}. Is that Ok?'.format(click.get_app_dir(APP_NAME))
    cfg_file = path.join(click.get_app_dir(APP_NAME), 'config.yml')
    if not path.exists(cfg_file):
        create_config_app_dir = click.confirm(no_config_msg)
        if create_config_app_dir:
            cfg_file = __create_config_file()
        else:
            exit(0)
    return cfg_file

    

@main.command()
def sync():
    cfg_file = get_config_file()
    fd = open(cfdg_file);
    parse_cfg_file = yaml.safe_load(fd)
    services = parse_cfg_file['services']
    for service in services:
        try:
            ServiceClass = importfile('{}/apps/{}.py'.format(PROJ_DIR, service['name']) 
            ServiceInstance = ServiceClass(**service)
            ServiceInstance.sync()
        except ErrorDuringImport as e:
            print(e)
            exit(1)