From a284514663f825115e6d48f2b81a1e3a3ac7d27d Mon Sep 17 00:00:00 2001 From: Robbe Van Herck Date: Wed, 28 Nov 2018 16:14:15 +0100 Subject: [PATCH] Remove achievement when nobody endorses it --- achievements.py | 10 ++++++++-- 1 file changed, 8 insertions(+), 2 deletions(-) diff --git a/achievements.py b/achievements.py index a6be9bf..6356c7b 100644 --- a/achievements.py +++ b/achievements.py @@ -36,5 +36,11 @@ def unendorse_achievement(id): return "Can't unendorse if you never endorsed..." achievement.endorsed_by.remove(current_user) db.session.add(achievement) - db.session.commit() - return redirect(url_for("get_achievement", id=id)) \ No newline at end of file + if len(achievement.endorsed_by) == 0: + db.session.delete(achievement) + db.session.commit() + return redirect(url_for("all_users")) + else: + db.session.commit() + return redirect(url_for("get_achievement", id=id)) +