def locacao_new(request): form = LocacaoForm() context = {'label': 'Cadastrar', 'form': form} if request.method == 'GET': form.set_veiculo(Veiculo.objects.filter(disponivel=True), "---------") form.set_cliente(Cliente.objects.all(), "---------") return render(request, 'locacao/locacao.html', context) form = LocacaoForm(request.POST) if not form.is_valid(): form.set_veiculo(Veiculo.objects.all(), "---------") context['form'] = form return render(request, 'locacao/locacao.html', context) form.save() messages.success(request, SUCCESS_MSG) return HttpResponseRedirect(r('locacao'))
def locacao_edit(request, pk): locacao = get_object_or_404(Locacao, pk=pk) if request.method == "POST": form = LocacaoForm(request.POST, instance=locacao) if form.is_valid(): locacao = form.save(commit=False) locacao.save() messages.success(request, UPDATE_MSG) return HttpResponseRedirect(r('locacao')) else: form = LocacaoForm(instance=locacao, initial={'veiculo': locacao.veiculo.pk}) form.set_veiculo(Veiculo.objects.filter(pk=locacao.veiculo.pk)) form.set_cliente(Cliente.objects.filter(pk=locacao.cliente.pk)) context = {'label': 'Editar', 'form': form} return render(request, 'locacao/locacao.html', context)