82 lines
2.8 KiB
Python
82 lines
2.8 KiB
Python
from grocy import RestService
|
|
from grocy.commands import products
|
|
from tabulate import tabulate
|
|
from os import path
|
|
|
|
class Recipe(object):
|
|
GET_RECIPES = '/get-objects/recipes'
|
|
GET_RECIPE = '/get-object/recipes/{0}'
|
|
GET_PRODUCT = '/get-object/products/{0}'
|
|
GET_RECIPES_POS = '/get-object/recipes_pos'
|
|
def __init__(self, id, **entries):
|
|
self.id = id
|
|
self.__dict__.update(entries)
|
|
self._init_rest_service()
|
|
self.products = []
|
|
#self._set_default_table_formats()
|
|
#if not hasattr('tablefmt', self):
|
|
# self.tablefmt = None
|
|
#if not hasattr('colalign', self):
|
|
# self.colalign = None
|
|
|
|
|
|
|
|
def _get_products_by_recipe_id(self):
|
|
recipe_products = self.rest_service.get(Recipe.GET_RECIPES_POS)
|
|
products_for_recipe = []
|
|
for recipe_product in recipe_products:
|
|
if recipe_product.get('recipe_id') == self.id:
|
|
## TODO: need to find a better way to run a batch call to get only products for recipe
|
|
product = self.rest_service.get(Recipe.GET_PRODUCT.format(recipe_product.get('product_id'))
|
|
## combined dict into single dict
|
|
product_recipie_info = {k: v for combined_dict in [product, recipe_product] for k, v in combined_dict.items()}
|
|
self.products.append(product_recipie_info)
|
|
|
|
def _set_default_table_formats(self):
|
|
if not hasattr('formats', self):
|
|
self.tablefmt = None
|
|
self.colalign = None
|
|
elif not hasattr('table', self.formats):
|
|
self.tableformat = None
|
|
elif not hasattr('col', self.formats):
|
|
self.colalign = None
|
|
|
|
|
|
|
|
def _init_rest_service(self):
|
|
if self.api.startswith == '/':
|
|
self.api = self.api[1:]
|
|
if self.api.endswith == '/':
|
|
self.api = self.api[1:-1]
|
|
self.rest_service = RestService(self.api, json=True)
|
|
self.rest_service.addHeader('Content-Type', 'application/json')
|
|
self.rest_service.addToken(self.token)
|
|
|
|
|
|
def get_list(self):
|
|
try:
|
|
recipes = self.rest_service.get(Recipe.GET_RECIPES)
|
|
table_headers = ['#', 'Name']
|
|
table_entries = []
|
|
for recipe in recipes:
|
|
table_entry = [recipe.get('id'), recipe.get('name')]
|
|
table_entries.append(table_entry)
|
|
|
|
except Exception as e:
|
|
raise e
|
|
# Generate stock overview table
|
|
return tabulate(table_entries, headers=table_headers)
|
|
|
|
def get(self, include_products=False):
|
|
try:
|
|
recipe = self.rest_service.get(Recipe.GET_RECIPE.format(self.id))
|
|
if include_products:
|
|
self.products = self._get_products_by_recipe_id()
|
|
|
|
except Exception as e:
|
|
raise e
|
|
|
|
return recipe.to_json()
|
|
|
|
|