add a delete feature

2 files changed, 10 insertions, 1 deletions

diff --git a/budget/budget.py b/budget/budget.py
index e22b053..6a579f1 100644
--- a/budget/budget.py
+++ b/budget/budget.py
@@ -125,6 +125,14 @@ def reset_bills():
     return redirect(url_for('list_bills'))
 
 
+@app.route("/delete/<int:bill_id>")
+def delete_bill(bill_id):
+    Bill.query.filter(Bill.id == bill_id).delete() 
+    BillOwer.query.filter(BillOwer.bill_id == bill_id).delete()
+    db.session.commit()
+    flash("the bill was deleted")
+
+    return redirect(url_for('list_bills'))
 
 if __name__ == '__main__':
     app.run(host="0.0.0.0", debug=True)
diff --git a/budget/templates/list_bills.html b/budget/templates/list_bills.html
index fba328e..09caa48 100644
--- a/budget/templates/list_bills.html
+++ b/budget/templates/list_bills.html
@@ -2,7 +2,7 @@
 {% block content %}
 {% if bills.count() > 0 %}
 <table>
-<thead><tr><th>When ?</th><th>Who paid?</th><th>for what ?</th><th>Owers</th><th>How much ?</th></tr></thead>
+    <thead><tr><th>When ?</th><th>Who paid?</th><th>for what ?</th><th>Owers</th><th>How much ?</th><th>Actions</th></tr></thead>
 <tbody>
 {% for bill in bills %}
 <tr class="{{ loop.cycle("odd", "even") }}">
@@ -11,6 +11,7 @@
 	<td>{{ bill.what }}</td>
 	<td>{% for ower in bill.owers %}{{ ower.name }} {% endfor %}</td>
 	<td>{{ bill.amount }} ({{ bill.pay_each() }} each)</td>
+        <td><a href="{{ url_for("delete_bill", bill_id=bill.id) }}">delete</a></td>
 </tr>
 {% endfor %}
 </tbody>