def edit(request,offset): try: d = Driver.objects.get(pk=int(offset)) if request.method == "POST": form = DriverForm(request.POST,request.FILES, instance=d) else: form = DriverForm(instance=d) if form.is_valid(): driver = form.save(commit = False) system = System.objects.get(pk=request.session['system']) driver.system = system driver.save() vehicles = form.cleaned_data['vehicle'] for v in vehicles: driver.vehicle.add(v) driver.save() return HttpResponseRedirect("/drivers/edit/finish") return render_to_response("drivers/templates/create.html",locals(),context_instance=RequestContext(request)) except ObjectDoesNotExist: return HttpResponseNotFound("O motorista solicitado não existe.")
def create(request): #checks if system can alter the vehicle's drivers try: # if request.session['system'] in map(lambda x: x['id'],Vehicle.objects.get(pk=int(offset)).system.values()): if request.method == "POST": form = DriverForm(request.POST,request.FILES) else: form = DriverForm() if form.is_valid(): driver = form.save(commit = False) system = System.objects.get(pk=request.session['system']) driver.system = system driver.save() vehicles = form.cleaned_data['vehicle'] for v in vehicles: driver.vehicle.add(v) driver.save() return HttpResponseRedirect("/drivers/create/finish") return render_to_response("drivers/templates/create.html",locals(),context_instance=RequestContext(request)) # else: # return HttpResponseForbidden(u"O seu sistema não pode criar motoristas para este veículo.") except ObjectDoesNotExist: return HttpResponseNotFound("O veículo solicitado não existe.")