diff --git a/treasureapp/templates/accounts/form.html b/treasureapp/templates/accounts/form.html index a389b69..83afd5b 100644 --- a/treasureapp/templates/accounts/form.html +++ b/treasureapp/templates/accounts/form.html @@ -1,10 +1,15 @@ {% extends "base.html" %} {% load bootstrap %} -{% block title %}Treasure Chest | Add account{% endblock %} + +{% block title %}{{Treasure Chest | Add/Update account"}{% endblock %} {% block content %} -

Add new account

+{% if mode == 'create' %} +

Add new account

+{% elif mode == 'update' %} +

Update existing account

+{% endif %}
{% csrf_token %} @@ -12,6 +17,7 @@

Add new account

+ Cancel
diff --git a/treasureapp/templates/registration/registration_form.html b/treasureapp/templates/registration/registration_form.html index d3a2748..af9b22d 100644 --- a/treasureapp/templates/registration/registration_form.html +++ b/treasureapp/templates/registration/registration_form.html @@ -11,6 +11,7 @@
+ Cancel
diff --git a/treasureapp/templates/transactions/form.html b/treasureapp/templates/transactions/form.html index b4e6236..3824117 100644 --- a/treasureapp/templates/transactions/form.html +++ b/treasureapp/templates/transactions/form.html @@ -12,6 +12,7 @@

Transaction out of {{ account.name }}

+ Cancel
diff --git a/treasureapp/view/account.py b/treasureapp/view/account.py index 9efbe70..d8cb8bb 100644 --- a/treasureapp/view/account.py +++ b/treasureapp/view/account.py @@ -81,7 +81,7 @@ def account_create(request, *args, **kargs): # Update the CSRF token kargs.update(csrf(request)) - context = RequestContext(request, dict(section="accounts", + context = RequestContext(request, dict(section="accounts", mode = "create", form=account_form, **kargs)) return render_to_response("accounts/form.html", context) @@ -113,5 +113,5 @@ def account_update(request, account_id, *args, **kargs): # Pass back the form we have, after updating CSRF kargs.update(csrf(request)) context = RequestContext(request, dict(section="accounts", - form=account_form, **kargs)) + form=account_form, mode="update", account=account, **kargs)) return render_to_response("accounts/form.html", context) diff --git a/treasureapp/view/transaction.py b/treasureapp/view/transaction.py index d6faaf5..8100c97 100644 --- a/treasureapp/view/transaction.py +++ b/treasureapp/view/transaction.py @@ -60,6 +60,7 @@ def transaction_create(request, account_id, *args, **kargs): context = RequestContext(request, dict(section="accounts", account=from_acct, form=transaction_form, + mode='create', **kargs)) return render_to_response("transactions/form.html", context) @@ -93,5 +94,7 @@ def transaction_update(request, transaction_id, *args, **kargs): context = RequestContext(request, dict(section="accounts", account=transaction.from_acct, form=transaction_form, + transaction=transaction, + mode='update', **kargs)) return render_to_response("transactions/form.html", context)