Add ability to remove a user

1 files changed, 15 insertions, 0 deletions

diff --git a/src/lb_app/app_db/user.py b/src/lb_app/app_db/user.py
index 1af957c..7aeab7c 100644
--- a/src/lb_app/app_db/user.py
+++ b/src/lb_app/app_db/user.py
@@ -25,3 +25,18 @@ class User(object):
             mongo.db.users.insert({"username": username})
             return None
         return error
+
+    def remove(self, username):
+        db = MongoDB(self.app)
+        mongo = db.connection()
+        error = None
+
+        if not username:
+            error = 'Username is required.'
+        elif len(list(mongo.db.users.find({"username": username}))) == 0:
+            error = "User {} not exist. So it's good.".format(username)
+
+        if error is None:
+            mongo.db.users.remove({"username": username})
+            return None
+        return error