def add(request):
"""Adds an account"""
if (request.method == 'POST'):
form = AccountForm(request.POST)
if form.is_valid():
acc = form.save(commit=False)
acc.user = request.user
acc.balance_updated = datetime.datetime.today()
acc.starting_balance = acc.balance
acc.save()
return redirect(reverse('moneydj.accounts.views.view', args=[acc.pk]))
else:
form = AccountForm()
return render_to_response('account_add.html', {'form': form}, context_instance=RequestContext(request))
def edit(request, id):
""" Edit an account """
acc = get_object_or_404(Account, pk=id, user=request.user)
orig_balance = acc.balance
if (request.method == 'POST'):
form = AccountForm(data=request.POST, instance=acc)
if form.is_valid():
acc = form.save(commit=False)
acc.user = request.user
if orig_balance is not acc.balance:
acc.set_balance(acc.balance)
acc.save()
return redirect(reverse('moneydj.accounts.views.view', args=[acc.pk]))
else:
form = AccountForm(instance=acc)
return render_to_response('account_edit.html', {'form': form, 'account': acc}, context_instance=RequestContext(request))
