109 lines
4.6 KiB
Python
109 lines
4.6 KiB
Python
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': # TODO refactor with add_version
|
|
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})
|