76 lines
2.3 KiB
Python
76 lines
2.3 KiB
Python
|
import click
|
||
|
|
from grocy import RestService
|
||
|
|
from grocy.commands import Stock
|
||
|
|
from pkg_resources import iter_entry_points
|
||
|
|
import yaml
|
||
|
|
from sys import exit
|
||
|
|
from shutil import copy
|
||
|
|
from os import path, chmod, makedirs
|
||
|
|
from taskw import TaskWarriorShellout
|
||
|
|
import logging
|
||
|
|
|
||
|
|
logger = logging.getLogger(__name__)
|
||
|
|
|
||
|
|
APP_NAME = 'grocy-cli'
|
||
|
|
SAMPLE_CONFIG_FILE = 'sample.config.yml'
|
||
|
|
CONFIG_FILE = 'config.yml'
|
||
|
|
CONFIG_DIR = click.get_app_dir(APP_NAME)
|
||
|
|
PROJ_DIR = path.join(path.dirname(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_FILE)
|
||
|
|
if not path.exists(CONFIG_DIR):
|
||
|
|
click.echo('Config {} director does not exist, create...'.format(CONFIG_DIR))
|
||
|
|
makedirs(CONFIG_DIR)
|
||
|
|
|
||
|
|
copy(sample_cfg_file, user_cfg_file)
|
||
|
|
click.echo('Copying sample config to {}'.format(sample_cfg_file, user_cfg_file))
|
||
|
|
chmod(user_cfg_file, 0o664)
|
||
|
|
return user_cfg_file
|
||
|
|
|
||
|
|
|
||
|
|
@click.group()
|
||
|
|
def main():
|
||
|
|
pass
|
||
|
|
|
||
|
|
|
||
|
|
def get_config_file():
|
||
|
|
no_config_msg = 'A config file was not found'
|
||
|
|
no_config_msg+= ' and will be created under {}. Is that Ok?'
|
||
|
|
no_config_msg = no_config_msg.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()
|
||
|
|
exit(0)
|
||
|
|
fd = open(cfg_file)
|
||
|
|
parse_cfg_file = yaml.safe_load(fd)
|
||
|
|
return parse_cfg_file
|
||
|
|
|
||
|
|
|
||
|
|
@main.command()
|
||
|
|
def stock():
|
||
|
|
cfg = get_config_file()
|
||
|
|
|
||
|
|
# Get logger
|
||
|
|
if 'logger' in cfg:
|
||
|
|
log_cfg = cfg['logger']
|
||
|
|
else:
|
||
|
|
log_cfg = path.join(click.get_app_dir(APP_NAME), 'grocy.log')
|
||
|
|
log_level = 'DEBUG' if 'level' not in log_cfg else log_cfg['level']
|
||
|
|
log_filename = 'log' if 'file_location' not in log_cfg else log_cfg['file_location']
|
||
|
|
logging.basicConfig(level=log_level, filename=log_filename)
|
||
|
|
|
||
|
|
# Validate token
|
||
|
|
if hasattr(cfg, 'token') or cfg['token'] is None:
|
||
|
|
click.echo('No token was found. Please add your token to the config file', err=True)
|
||
|
|
logger.error('No token was found. Please add your token')
|
||
|
|
exit(1)
|
||
|
|
|
||
|
|
stock = Stock(**cfg)
|
||
|
|
stock_entries = stock.get_entries()
|
||
|
|
click.echo(stock_entries)
|