From a3ef647a2d94b1f82d5d6e15ce7624401e51aa63 Mon Sep 17 00:00:00 2001
From: RunasSudo <runassudo@yingtongli.me>
Date: Sun, 28 May 2023 13:02:12 +1000
Subject: [PATCH] After editing transaction, redirect back to previous page

---
 drcr/journal/views.py                                | 4 ++--
 drcr/templates/journal/journal_edit_transaction.html | 2 ++
 2 files changed, 4 insertions(+), 2 deletions(-)

diff --git a/drcr/journal/views.py b/drcr/journal/views.py
index b9dbbaf..6f47a42 100644
--- a/drcr/journal/views.py
+++ b/drcr/journal/views.py
@@ -85,7 +85,7 @@ def journal_edit_transaction():
 				db.session.delete(reconciliation)
 		
 		db.session.commit()
-		return redirect(url_for('journal'))
+		return redirect(request.form.get('referrer', '') or url_for('journal'))
 	
 	# Edit transaction
 	transaction.dt = datetime.strptime(request.form['dt'], '%Y-%m-%d')
@@ -121,7 +121,7 @@ def journal_edit_transaction():
 	
 	db.session.commit()
 	
-	return redirect(url_for('journal'))
+	return redirect(request.form.get('referrer', '') or url_for('journal'))
 
 @app.route('/balance-assertions')
 def balance_assertions():
diff --git a/drcr/templates/journal/journal_edit_transaction.html b/drcr/templates/journal/journal_edit_transaction.html
index 13f8774..60025c0 100644
--- a/drcr/templates/journal/journal_edit_transaction.html
+++ b/drcr/templates/journal/journal_edit_transaction.html
@@ -125,6 +125,8 @@
 			{% endif %}
 			<button type="submit" class="btn btn-primary ms-auto">Save</button>
 		</div>
+		
+		<input type="hidden" name="referrer" value="{{ request.referrer or '' }}">
 	</form>
 {% endblock %}