80 lines
3.2 KiB
Python
80 lines
3.2 KiB
Python
from django.shortcuts import render, get_object_or_404
|
|
from .models import Recipe, Version
|
|
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, 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
|
|
|
|
if 'author' in version_form.cleaned_data:
|
|
version.author = version_form.cleaned_data['author']
|
|
|
|
version.recipe = recipe
|
|
version.save()
|
|
|
|
for ingredient_form in ingredients_formset:
|
|
if 'text' in ingredient_form.cleaned_data and ingredient_form.cleaned_data['text']:
|
|
ingredient = ingredient_form.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(prefix=INGREDIENTS_FORMSET_PREFIX)
|
|
|
|
return render(request, 'add-version.html', {'version_form': version_form, 'ingredients_formset': ingredients_formset})
|