from django.shortcuts import render, get_object_or_404 from .models import Recipe, Version, Ingredient from .forms import VersionForm, IngredientFormSet from django.contrib.auth.decorators import login_required, permission_required from django.urls import reverse from django.http.response import HttpResponseRedirect from django.conf import settings @login_required @permission_required('recipes.view_recipe') def recipes(request): recipes = Recipe.objects.all() return render(request, 'recipes.html', context={'recipes': recipes}) @login_required @permission_required('recipes.view_recipe') def version(request, slug_recipe, slug_version): recipe = get_object_or_404(Recipe, slug=slug_recipe) has_multiple_versions = recipe.versions.all().count() > 1 # type: ignore version = get_object_or_404(Version, recipe=recipe, slug=slug_version) ingredients = version.ingredients.all() # type: ignore return render(request, 'version.html', context={'recipe': recipe, 'has_multiple_versions': has_multiple_versions, 'version': version, 'ingredients': ingredients}) @login_required @permission_required('recipes.view_recipe') def recipe(request, slug): recipe = get_object_or_404(Recipe, slug=slug) versions = recipe.versions.all() # type: ignore if (1 == versions.count()): if settings.ENFORCE_RECIPE_VERSION_URL: return HttpResponseRedirect(reverse('version', kwargs={'slug_recipe': slug, 'slug_version': versions.get().slug})) else: return version(request, slug, versions.get().slug) else: return render(request, 'recipe.html', context={'recipe': recipe, 'versions': versions}) VERSION_FORM_PREFIX = 'version' INGREDIENTS_FORMSET_PREFIX = 'ingredient' @login_required @permission_required('recipes.add_recipe') def add_version(request, slug_recipe): recipe = Recipe.objects.get(slug=slug_recipe) if request.method == 'POST': version_form = VersionForm(request.POST, prefix=VERSION_FORM_PREFIX) ingredients_formset = IngredientFormSet(request.POST, queryset=Ingredient.objects.none(), prefix=INGREDIENTS_FORMSET_PREFIX) version_form.recipe_id = recipe.id # type: ignore if version_form.is_valid() and ingredients_formset.is_valid(): version = version_form.save(commit=False) version.user = request.user version.recipe = recipe version.save() for ingredient in ingredients_formset.save(commit=False): ingredient.version = version ingredient.save() return HttpResponseRedirect(reverse('version', kwargs={'slug_recipe': version.recipe.slug, 'slug_version': version.slug})) else: if (recipe.versions.all().count() > 0): # type: ignore initial = {'label': '', 'slug': ''} else: initial = {} version_form = VersionForm(prefix=VERSION_FORM_PREFIX, initial=initial) version_form.fields['author'].widget.attrs['placeholder'] = request.user.first_name if request.user.first_name else request.user.username ingredients_formset = IngredientFormSet(queryset=Ingredient.objects.none(), prefix=INGREDIENTS_FORMSET_PREFIX) return render(request, 'add-version.html', {'version_form': version_form, 'ingredients_formset': ingredients_formset}) @login_required @permission_required('recipes.add_recipe') def edit_version(request, slug_recipe, slug_version): recipe = get_object_or_404(Recipe, slug=slug_recipe) version = get_object_or_404(Version, recipe=recipe, slug=slug_version) if request.method == 'POST': version_form = VersionForm(request.POST, prefix=VERSION_FORM_PREFIX, instance=version) ingredients_formset = IngredientFormSet(request.POST, queryset=version.ingredients.all(), prefix=INGREDIENTS_FORMSET_PREFIX) # type: ignore version_form.recipe_id = recipe.id # type: ignore if version_form.is_valid() and ingredients_formset.is_valid(): if version_form.has_changed(): version = version_form.save(commit=False) version.user = request.user version.recipe = recipe version.save() if ingredients_formset.has_changed(): for ingredient in ingredients_formset.save(commit=False): ingredient.version = version ingredient.save() return HttpResponseRedirect(reverse('version', kwargs={'slug_recipe': version.recipe.slug, 'slug_version': version.slug})) else: version_form = VersionForm(instance=version, prefix=VERSION_FORM_PREFIX) initial = [] for ingredient in version.ingredients.all(): # type: ignore initial.append({'text': ingredient.text}) ingredients_formset = IngredientFormSet(queryset=version.ingredients.all(), prefix=INGREDIENTS_FORMSET_PREFIX) # type: ignore return render(request, 'edit-version.html', {'version_form': version_form, 'ingredients_formset': ingredients_formset})