django-project/recipes/views.py

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, permission_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
@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})

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
@permission_required('recipes.add_recipe')
def add_version(request, slug):
	recipe = Recipe.objects.get(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
@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': # TODO refactor with add_version
		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})