From 9206c2102516f57211dd7ea51fbf8474c04a9c6d Mon Sep 17 00:00:00 2001
From: RunasSudo <runassudo@yingtongli.me>
Date: Sat, 6 Apr 2024 00:39:36 +1100
Subject: [PATCH] Add button to delete balance assertion

---
 drcr/journal/views.py                               | 6 ++++++
 drcr/templates/journal/balance_assertions_edit.html | 5 ++++-
 2 files changed, 10 insertions(+), 1 deletion(-)

diff --git a/drcr/journal/views.py b/drcr/journal/views.py
index 5364997..7891697 100644
--- a/drcr/journal/views.py
+++ b/drcr/journal/views.py
@@ -174,6 +174,12 @@ def balance_assertions_edit():
 	if request.method == 'GET':
 		return render_template('journal/balance_assertions_edit.html', assertion=assertion)
 	
+	if request.form.get('action', None) == 'delete':
+		# Delete balance assertion
+		db.session.delete(assertion)
+		db.session.commit()
+		return redirect(request.form.get('referrer', '') or url_for('balance_assertions'))
+	
 	quantity = round(float(request.form['amount']) * (10**AMOUNT_DPS))
 	if request.form['sign'] == 'cr':
 		quantity = -quantity
diff --git a/drcr/templates/journal/balance_assertions_edit.html b/drcr/templates/journal/balance_assertions_edit.html
index f82535d..5b153de 100644
--- a/drcr/templates/journal/balance_assertions_edit.html
+++ b/drcr/templates/journal/balance_assertions_edit.html
@@ -54,7 +54,10 @@
 				</div>
 			</div>
 			
-			<div class="text-end">
+			<div class="flex justify-end mt-4 space-x-2">
+				{% if assertion and assertion.id %}
+					<button type="submit" name="action" value="delete" class="btn-secondary text-red-600 ring-red-500" onclick="confirm('Are you sure you want to delete this balance assertion? This operation is irreversible.')">Delete</button>
+				{% endif %}
 				<button type="submit" class="btn-primary">Save</button>
 			</div>
 		</form>