def edit(request, caso_id):
caso = Casos.objects.get(pk=caso_id)
if request.method == 'POST':
form = FormCaso(request.POST)
if form.is_valid():
caso.classificacao_iap = request.POST['iap']
caso.classificacao_iva = request.POST['iva']
caso.entorno = Entorno.objects.get(pk=request.POST['entorno'])
caso.risco = request.POST['risco']
caso.solucao_sugerida = request.POST['solucao_sugerida']
caso.save()
return redirect('/caso/')
else:
data = {
'iap': caso.classificacao_iap,
'iva': caso.classificacao_iva,
'risco': caso.risco,
'entorno': caso.entorno,
'solucao_sugerida': caso.solucao_sugerida
}
form = FormCaso(initial=data)
return render(request, 'caso/edit.html', {
'form': form,
'caso_id': caso.id
})
def add(request):
if request.method == 'POST':
form = FormCaso(request.POST)
if form.is_valid():
caso = Casos()
caso.classificacao_iap = request.POST['iap']
caso.classificacao_iva = request.POST['iva']
caso.entorno = Entorno.objects.get(pk=request.POST['entorno'])
caso.risco = request.POST['risco']
caso.solucao_sugerida = request.POST['solucao_sugerida']
caso.save()
return redirect('/caso/')
else:
form = FormCaso()
return render(request, 'caso/add.html', {'form': form})