def add(request):
if request.method == "POST":
form = FormRio(request.POST)
if form.is_valid():
rio = Rio()
rio.nome = request.POST["nome"]
rio.dimensao = request.POST["dimensao"]
rio.bacia_hidrografica = Bacia_Hidrografica.objects.get(pk=request.POST["bacia_hidrografica"])
rio.save()
return redirect("/rio/")
else:
form = FormRio()
return render(request, "rio/add.html", {"form": form})
def edit(request, rio_id):
rio = Rio.objects.get(pk=rio_id)
if request.method == "POST":
form = FormRio(request.POST)
if form.is_valid():
rio.nome = request.POST["nome"]
rio.dimensao = request.POST["dimensao"]
rio.bacia_hidrografica = Bacia_Hidrografica.objects.get(pk=request.POST["bacia_hidrografica"])
rio.save()
return redirect("/rio/")
else:
data = {"nome": rio.nome, "dimensao": rio.dimensao, "bacia_hidrografica": rio.bacia_hidrografica}
form = FormRio(initial=data)
return render(request, "rio/edit.html", {"form": form, "rio_id": rio.id})
