from django.shortcuts import render, get_object_or_404 from .models import Recipe, Version, Ingredient from .forms import RecipeForm, VersionForm, IngredientFormSet from django.contrib.auth.decorators import login_required from django.urls import reverse from django.http.response import HttpResponseRedirect from django.conf import settings def get_name_of_user(user): return user.first_name if user.first_name else user.username @login_required def recipes(request): recipes = Recipe.objects.all() return render(request, 'recipes.html', context={'recipes': recipes}) 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 def recipe(request, slug): recipe = get_object_or_404(Recipe, slug=slug) versions = recipe.versions.all() # type: ignore if (1 == versions.count()): return HttpResponseRedirect(reverse('version', kwargs={'slug_recipe': slug, 'slug_version': versions.get().slug})) else: return render(request, 'recipe.html', context={'recipe': recipe, 'versions': versions}) RECIPE_FORM_PREFIX = 'recipe' VERSION_FORM_PREFIX = 'version' INGREDIENTS_FORMSET_PREFIX = 'ingredient' @login_required def add_recipe(request): if request.method == 'POST': recipe_form = RecipeForm(request.POST, prefix=RECIPE_FORM_PREFIX) version_form = VersionForm(request.POST, prefix=VERSION_FORM_PREFIX, author_placeholder=get_name_of_user(request.user)) ingredients_formset = IngredientFormSet(request.POST, queryset=Ingredient.objects.none(), prefix=INGREDIENTS_FORMSET_PREFIX) if recipe_form.is_valid() and version_form.is_valid() and ingredients_formset.is_valid(): recipe = recipe_form.save(commit=True) 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('recipe', kwargs={'slug': recipe.slug})) else: recipe_form = RecipeForm(prefix=RECIPE_FORM_PREFIX) version_form = VersionForm(prefix=VERSION_FORM_PREFIX, author_placeholder=get_name_of_user(request.user)) ingredients_formset = IngredientFormSet(queryset=Ingredient.objects.none(), prefix=INGREDIENTS_FORMSET_PREFIX) return render(request, 'add-recipe.html', {'recipe_form': recipe_form, 'version_form': version_form, 'ingredients_formset': ingredients_formset}) @login_required def edit_recipe(request, slug): recipe = get_object_or_404(Recipe, slug=slug) if request.method == 'POST': form = RecipeForm(request.POST, instance=recipe, prefix=RECIPE_FORM_PREFIX) if form.is_valid(): if form.has_changed(): form.save() return HttpResponseRedirect(reverse('recipe', kwargs={'slug': recipe.slug})) else: form = RecipeForm(instance=recipe, prefix=RECIPE_FORM_PREFIX) return render(request, 'edit-recipe.html', {'form': form}) @login_required def add_version(request, slug): recipe = get_object_or_404(Recipe, slug=slug) if (recipe.versions.all().count() > 0): # type: ignore version_initial = {'label': '', 'slug': ''} else: version_initial = {} if request.method == 'POST': version_form = VersionForm(request.POST, prefix=VERSION_FORM_PREFIX, initial=version_initial, author_placeholder=get_name_of_user(request.user)) 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: version_form = VersionForm(prefix=VERSION_FORM_PREFIX, initial=version_initial, author_placeholder=get_name_of_user(request.user)) 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 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, author_placeholder=get_name_of_user(request.user)) 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, author_placeholder=get_name_of_user(request.user)) 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})