import click from markdown import markdown from html2text import html2text from jinja2 import Environment, FileSystemLoader from grocy.meta import Meta from grocy.conf import Configuration from grocy.util import Util from grocy.recipe import Recipe from grocy.table import Table from grocy.entity import Entity from grocy.stock import Stock import yaml from sys import exit from os import path import logging APP_NAME = 'grocy-cli' SAMPLE_CONFIG_FILE = 'sample.config.yml' TMP_DIR = '/tmp/grocy' CONFIG_FILE = 'config.yml' CONFIG_DIR = click.get_app_dir(APP_NAME) PROJ_DIR = path.join(path.dirname(path.realpath(__file__))) TEMPLATE_LOADER = Environment(loader=FileSystemLoader('templates'), trim_blocks=True, lstrip_blocks=True) class GrocyGroup(click.Group): def parse_args(self, ctx, args): is_subcommand = True if args[0] in self.commands else False if is_subcommand: ctx.forward(self.commands[args[0]], args[1:]) super(GrocyGroup, self).parse_args(ctx, args) @click.group() def main(): cfg = Configuration() if not cfg.exists: no_config_msg = 'A config file was not found. A sample configuration file' no_config_msg += ' will be created under {}. Is that Ok?'.format(cfg.CONFIG_DIR) create_config_app_dir = click.confirm(no_config_msg) user_cfg_options = {} if create_config_app_dir: user_cfg_options['logger_level'] = click.prompt('Enter logger level', default='DEBUG') user_cfg_options['logger_file_location'] = click.prompt('Enter location for logger', default=path.expanduser('~/.config/grocy/log')) user_cfg_options['api'] = click.prompt('Enter the grocy api url', default='https://demo-en.grocy.info/api') user_cfg_options['token'] = click.prompt('Enter the grocy token ', default='') user_cfg_options['col_format'] = click.prompt('Enter the col position for rendering tables', default='col') user_cfg_options['table_format'] = click.prompt('Enter the table format', default='simple') cfg.create(user_cfg_options) else: exit(0) else: cfg.load() logging.basicConfig(level=cfg.logger_level, filename=cfg.logger_file_location) @main.command() @click.pass_context def stock(ctx): logger = logging.getLogger('cli.stock') if ctx.invoked_subcommand is None: try: stock = Stock() table = Table(stocks=stock.products) click.echo(table.stock) except Exception as e: logger.error(e) raise e @main.group() def product(): pass @product.command() @click.pass_context @click.argument('product_id', required=False) def view(ctx, product_id): logger = logging.getLogger('cli.product') try: if product_id: stock = Stock() product = stock.get_product(product_id) table = Table(entry=product) click.echo(table.product) else: click.echo(ctx.get_help()) except Exception as e: logger.error(e) raise e @product.command() @click.option('--name', '-n', 'name') @click.argument('product_id', required=False) def edit(product_id, name): logger = logging.getLogger('cli.product.edit') try: cfg = Configuration() util = Util(cfg=cfg) cfg.load() loaded_template = cfg.templates('product/edit') entity = Entity(name='products') if product_id: product = entity.get(id=product_id) edited_product = click.edit(loaded_template.render(product)) edited_product.update() elif name: # Convert name args to a single string string_name_arg = ' '.join(name) if type(name) == list else name products = entity.find({'name': string_name_arg}) if products is None: click.echo('Could not find product') return edited_products = click.edit(loaded_template.render(products=products), extension='.yml') if edited_products is None: return parsed_edited_products = Util.load_yaml(edited_products) schema = entity.schema for index, edited_product in enumerate(parsed_edited_products): edited_product['id'] = products[index]['id'] Util.verify_integrity(edited_product, schema) entity.update(edited_product, id=products[index]['id']) else: raise click.BadParameter('Missing PRODUCT_ID or QUERY') except Exception as e: logger.error(e) raise e @product.command() @click.option('--name', '-n', 'name') @click.option('-t', 'template') def list(name, template): logger = logging.getLogger('cli.product.list') cfg = Configuration() cfg.load() product_entity = Entity(name='products') qu_entity = Entity(name='quantity_units') location_entity = Entity(name='locations') product_group_entity = Entity(name='product_groups') try: if name: # Convert name args to a single string string_name_arg = ' '.join(name) if type(name) == list else name products = product_entity.find({'name': string_name_arg}) else: products = product_entity.get() entries = {} entries['quantity_units'] = qu_entity.get() entries['product_groups'] = product_group_entity.get() entries['locations'] = location_entity.get() entries['products'] = products table = Table(entries=entries) click.echo_via_pager(table.products) except Exception as e: logger.error(e) raise e @product.command() @click.argument('product_id') def browse(product_id): logger = logging.getLogger('cli.product.browse') try: cfg = Configuration() cfg.load() url = '{domain}/product/{product_id}'.format(domain=cfg.domain, product_id=product_id) click.launch(url, wait=False) except Exception as e: logger.error(e) raise e @product.command() @click.option('-t', 'template') def add(template): logger = logging.getLogger('cli.product.add') try: cfg = Configuration() cfg.load() logger = logging.getLogger('cli.product.add') meta = Meta() # Get product_groups meta.add(type='entities', name='product_groups') # Get locations meta.add(type='entities', name='locations') # Get quantity_units meta.add(type='entities', name='quantity_units') if template: loaded_template = cfg.templates(template) else: loaded_template = cfg.templates('product/add') new_product = click.edit(loaded_template.render(grocy=meta.generate()), extension='.yml') if not new_product: return parsed_new_product = yaml.safe_load(new_product) if template == 'debug': click.echo(parsed_new_product) return entity = Entity(name='products') entity.create(parsed_new_product) except Exception as e: logger.error(e) raise e @main.group() @click.pass_context def recipe(ctx): pass @recipe.command() @click.pass_context @click.argument('recipe_id', required=False) @click.option('-t', 'template') def view(ctx, recipe_id, template): logger = logging.getLogger('cli.recipes.view') try: cfg = Configuration() cfg.load() data = {'fields': {}} if recipe_id: entity = Entity(name='recipes') data['fields'] = entity.get(id=recipe_id) # Change html markup to plain text html_markup_description = data['fields']['description'] plain_text_description = html2text(html_markup_description) data['fields']['description'] = plain_text_description recipe = Recipe(id=recipe_id) data['fields']['fulfillment'] = recipe.get_fulfillments() if template: loaded_template = cfg.templates(template) else: loaded_template = cfg.templates('recipe/view') entity = Entity(name='recipes') click.echo(loaded_template.render(grocy=data)) else: click.echo(ctx.get_help()) except Exception as e: logger.error(e) raise e #@recipe.command() #def list(): # logger = logging.getLogger('cli.recipe') # try: # entity = Entity(name='recipes') # recipes = entity.get() # recipe = Recipe( # ingredient_reqs = recipe.get_ingredient_requirements() # table = Table(recipes=recipes, recipes_reqs=recipes_reqs, ingredient_reqs=ingredient_reqs) # click.echo(table.recipe) # except Exception as e: # logger.error(e) # raise e ## # #@main.command() #def shopping(): # logger = logging.getLogger('cli.shopping') # try: # entity = Entity(name='shopping_list') # shopping_list = entity.get() # table = Table(shopping_list=shopping_list) # click.echo(table.shopping_list) # except Exception as e: # logger.error(e) # raise e # # #@main.command() #@main.group() #def ingredient(): # pass # # #@ingredient.command('add') #@click.argument('query') #@click.argument('recipe_id') #@click.option('--amount', '-a', 'amount', default=1, type=int) #@click.option('--group', '-g', type=str, default='') #@click.option('--variable-amount', '--va', 'variable_amount', default=None, type=float) #@click.option('--in-stock', '--is', 'in_stock', default=False) #@click.option('--disable-fulfillment', '--df', 'disable_fulfillment', default=False) #@click.option('--note', '-n', 'note', multiple=True, default='', type=str) #@click.option('--no-edit', '--ne', 'no_edit', default=False) #def add(query, recipe_id, amount, group, variable_amount, in_stock, disable_fulfillment, note, no_edit): # logger = logging.getLogger('cli.ingredient.add') # # try: # loaded_template = TEMPLATE_LOADER.get_template('ingredient_add.yml') # new_ingredient = {} # # entity = Entity(name='recipes') # recipe = entity.get(id=recipe_id) # # if not recipe: # raise click.BadParameter(message='recipe {id} does not exist', param='recipe_id', # param_hint='Use `grocy recipes ls` to get a list of recipes') # # entity = Entity(name='products') # product = entity.findOne(query) # new_ingredient['product_id'] = product['id'] # # new_ingredient['amount'] = amount # new_ingredient['group'] = group # new_ingredient['variable_amount'] = variable_amount # new_ingredient['only_check_single_unit_in_stock'] = "1" if in_stock else "0" # new_ingredient['not_check_stock_fulfillment'] = "1" if disable_fulfillment else "0" # new_ingredient['note'] = note # # if not no_edit: # new_ingredient = click.edit(loaded_template.render(new_ingredient)) # # parsed_new_ingredient = yaml.safe_load(new_ingredient) # entity = Entity(name='recipes_pos') # #entity.create(parsed_new_ingredient) # # except Exception as e: # logger.error(e) # raise e # # #@recipe.command('edit') #@click.argument('recipe_id') #def edit(recipe_id): # logger = logging.getLogger('cli.recipe.edit') # try: # if recipe_id: # entity = Entity(name='recipes') # recipe = entity.get(id=recipe_id) # loaded_template = TEMPLATE_LOADER.get_template('recipe_edit.yml') # edited_recipe = click.edit(loaded_template.render(recipe)) # if edited_recipe is not None: # parsed_edited_recipe = yaml.safe_load(edited_recipe) # parsed_edited_recipe['description'] = markdown(parsed_edited_recipe['description']) # recipe.__dict__.update(parsed_edited_recipe) # recipe.update() # except Exception as e: # logger.error(e) # logger.error('Could not edit recipe {}'.format(recipe_id)) # raise e # # #@recipe.command('create') #@click.pass_context #def create(ctx): # logger = logging.getLogger('cli.recipe.create') # # try: # recipe = Entity(name='recipes') # loaded_template = TEMPLATE_LOADER.get_template('recipe_add.yml') # new_recipe = click.edit(loaded_template.render()) # if new_recipe is not None: # parsed_new_recipe = yaml.safe_load(new_recipe) # parsed_new_recipe['description'] = markdown(parsed_new_recipe['description']) # recipe.__dict__.update(parsed_new_recipe) # recipe.create() # except Exception as e: # logger.error(e) # raise e # #@main.command() #@click.pass_context #def chore(ctx): # cfg = ctx.obj['cfg'] # # chore = Chore(**cfg) # chores = chore.get_list() # click.echo(chores) # #@main.command() #@click.pass_context #def task(ctx): # cfg = ctx.obj['cfg'] # # task = Task(**cfg) # tasks = task.get_list() # click.echo(tasks) # click.echo(tasks)