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)
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'))
