From 0e5bab13af39d2fc7073860aa05279bcf9f19e82 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Fr=C3=A9d=C3=A9ric=20Sureau?= <fred@milka.(none)>
Date: Tue, 9 Aug 2011 19:34:46 +0200
Subject: Users are deleted when no bill is assigned to them. Fixes #5

---
 budget/web.py | 11 ++++++++---
 1 file changed, 8 insertions(+), 3 deletions(-)

(limited to 'budget/web.py')

diff --git a/budget/web.py b/budget/web.py
index a838b09..6b3849f 100644
--- a/budget/web.py
+++ b/budget/web.py
@@ -134,9 +134,14 @@ def add_member(project):
 def remove_member(project, member_id):
     person = Person.query.get_or_404(member_id)
     if person.project == project:
-        person.activated = False
-        db.session.commit()
-        flash("%s has been removed" % person.name)
+        if not person.is_used():
+            db.session.delete(person)
+            db.session.commit()
+            flash("User '%s' has been removed" % person.name)
+        else:
+            person.activated = False
+            db.session.commit()
+            flash("User '%s' has been desactivated" % person.name)
     return redirect(url_for("list_bills", project_id=project.id))
 
 @app.route("/<string:project_id>/add", methods=["GET", "POST"])
-- 
cgit v1.1