def edit_addition(request, addition_id): if auth.get_user(request).is_authenticated(): additions = get_object_or_404(AgreementAddition, id=addition_id) form = AdditionForm(instance=additions) username = auth.get_user(request).username try: user = User.objects.get(username=username).first_name except: user = '' context = { 'additions': additions, 'username': username, 'user': user, 'form': form, 'addition_id': addition_id, } context.update(csrf(request)) if request.method == 'POST': form = AdditionForm(request.POST, request.FILES, instance=additions) if form.is_valid(): addition = form.save(commit=False) addition.save() return redirect('/agreement/%s' % addition.addition_agreement_number) else: context.update({'error': 'True'}) return render_to_response('agreement/edit_addition.html', context) return render_to_response('agreement/edit_addition.html', context) else: return HttpResponseRedirect('/login/')
def new_addition(request, agreement_id): if auth.get_user(request).is_authenticated(): username = auth.get_user(request).username agreement = Agreement.objects.get(id=agreement_id) form = AdditionForm() try: user = User.objects.get(username=username).first_name except: user = '' context = { 'username': username, 'user': user, 'form': form, 'agreement_id': agreement_id, } context.update(csrf(request)) if request.method == 'POST': form = AdditionForm(request.POST, request.FILES) if form.is_valid(): addition = form.save(commit=False) addition.addition_agreement_number = agreement_id addition.save() agreement.agreement_addition.add(addition) return redirect('/agreement/%s' % agreement_id) else: context.update({'error': 'True'}) return render_to_response('agreement/new_addition.html', context) return render_to_response('agreement/new_addition.html', context) else: return HttpResponseRedirect('/login/')