def make_validated_form(self, **kwargs):
cliente = Cliente.objects.create(nome='lucas', tipo_cnh='B', cpf='12345678901')
categoria = CategoriaVeiculo.objects.create(nome='carro', tipo_cnh='B')
veiculo = Veiculo.objects.create(modelo='Palio', quilometragem=10, disponivel=True, categoria=categoria)
data = dict(cliente=cliente.pk, veiculo=veiculo.pk,
data_inicial='2015-01-15',data_final='2015-01-20',
valor=20,devolvido=False)
valid = dict(data)
data = dict(valid, **kwargs)
form = LocacaoForm(data)
form.is_valid()
return form
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'))
