74 lines
1.8 KiB
Python
74 lines
1.8 KiB
Python
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)
|
|
|
|
|
|
|
|
|
|
|