83 lines
2.9 KiB
Python
83 lines
2.9 KiB
Python
import json
|
|
import re
|
|
from io import BytesIO
|
|
from zipfile import ZipFile
|
|
|
|
from cookbook.helper.ingredient_parser import parse, get_food, get_unit
|
|
from cookbook.integration.integration import Integration
|
|
from cookbook.models import Recipe, Step, Food, Unit, Ingredient, Keyword
|
|
|
|
|
|
class MealMaster(Integration):
|
|
|
|
def get_recipe_from_file(self, file):
|
|
print('------------ getting recipe')
|
|
servings = 1
|
|
ingredients = []
|
|
directions = []
|
|
for line in file.replace('\r', '').split('\n'):
|
|
print('testing line')
|
|
if not line.startswith('MMMMM') and line.strip != '':
|
|
if 'Title:' in line:
|
|
title = line.replace('Title:', '').strip()
|
|
else:
|
|
if 'Categories:' in line:
|
|
tags = line.replace('Categories:', '').strip()
|
|
else:
|
|
if 'Yield:' in line:
|
|
servings_text = line.replace('Yield:', '').strip()
|
|
else:
|
|
if re.match('\s{2,}([0-9])+', line):
|
|
ingredients.append(line.strip())
|
|
else:
|
|
directions.append(line.strip())
|
|
|
|
try:
|
|
servings = re.findall('([0-9])+', servings_text)[0]
|
|
except Exception as e:
|
|
print('failed parsing servings ', e)
|
|
|
|
recipe = Recipe.objects.create(name=title, servings=servings, created_by=self.request.user, internal=True, space=self.request.space)
|
|
|
|
for k in tags.split(','):
|
|
keyword, created = Keyword.objects.get_or_create(name=k.strip(), space=self.request.space)
|
|
recipe.keywords.add(keyword)
|
|
|
|
step = Step.objects.create(
|
|
instruction='\n'.join(directions) + '\n\n'
|
|
)
|
|
|
|
for ingredient in ingredients:
|
|
amount, unit, ingredient, note = parse(ingredient)
|
|
f = get_food(ingredient, self.request.space)
|
|
u = get_unit(unit, self.request.space)
|
|
step.ingredients.add(Ingredient.objects.create(
|
|
food=f, unit=u, amount=amount, note=note
|
|
))
|
|
recipe.steps.add(step)
|
|
|
|
return recipe
|
|
|
|
def get_file_from_recipe(self, recipe):
|
|
raise NotImplementedError('Method not implemented in storage integration')
|
|
|
|
def split_recipe_file(self, file):
|
|
recipe_list = []
|
|
current_recipe = ''
|
|
|
|
for fl in file.readlines():
|
|
line = fl.decode("ANSI")
|
|
if (line.startswith('MMMMM') or line.startswith('-----')) and 'meal-master' in line.lower():
|
|
if current_recipe != '':
|
|
recipe_list.append(current_recipe)
|
|
current_recipe = ''
|
|
else:
|
|
current_recipe = ''
|
|
else:
|
|
current_recipe += line + '\n'
|
|
|
|
if current_recipe != '':
|
|
recipe_list.append(current_recipe)
|
|
|
|
return recipe_list
|