import json import re from io import BytesIO from zipfile import ZipFile from cookbook.helper.image_processing import get_filetype 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 class NextcloudCookbook(Integration): def import_file_name_filter(self, zip_info_object): return zip_info_object.filename.endswith('.json') def get_recipe_from_file(self, file): recipe_json = json.loads(file.getvalue().decode("utf-8")) description = '' if len(recipe_json['description'].strip()) > 500 else recipe_json['description'].strip() recipe = Recipe.objects.create( name=recipe_json['name'].strip(), description=description, created_by=self.request.user, internal=True, servings=recipe_json['recipeYield'], space=self.request.space) # TODO parse times (given in PT2H3M ) # TODO parse keywords ingredients_added = False for s in recipe_json['recipeInstructions']: step = Step.objects.create( instruction=s, space=self.request.space, ) if not ingredients_added: if len(recipe_json['description'].strip()) > 500: step.instruction = recipe_json['description'].strip() + '\n\n' + step.instruction ingredients_added = True for ingredient in recipe_json['recipeIngredient']: 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, space=self.request.space, )) recipe.steps.add(step) for f in self.files: if '.zip' in f['name']: import_zip = ZipFile(f['file']) for z in import_zip.filelist: if re.match(f'^Recipes/{recipe.name}/full.jpg$', z.filename): self.import_recipe_image(recipe, BytesIO(import_zip.read(z.filename)), filetype=get_filetype(z.filename)) return recipe def get_file_from_recipe(self, recipe): raise NotImplementedError('Method not implemented in storage integration')