From 9a3ca323de9dbdd4b6fc1b65a82decd65414ae48 Mon Sep 17 00:00:00 2001
From: daphtya <davidpanjaitan1@gmail.com>
Date: Thu, 11 Apr 2019 23:22:17 +0700
Subject: [PATCH] delete user

---
 modules/register.py | 20 ++++++++++++++++----
 1 file changed, 16 insertions(+), 4 deletions(-)

diff --git a/modules/register.py b/modules/register.py
index 7d59def..3404462 100644
--- a/modules/register.py
+++ b/modules/register.py
@@ -28,7 +28,19 @@ def add_new_user():
 
 @register_route.route('/api/unregister', methods=["POST"])
 def delete_user():
-    user_id = request.form.get("user_id")
-    username = request.form.get("username")
-    password = request.form.get("password")
-    check = db.User.objects.with_id("user_id")
\ No newline at end of file
+    try:
+        token = request.headers.get("Authorization")
+        password = request.form.get("password")
+        if token is not None:
+            code, user_id = Token.decode_token(token) 
+            if code > 0:
+                raise Exception(user_id)
+        else:
+            user_id = request.form.get("user_id")
+            if user_id is None:
+                raise Exception("Not found")
+        check = db.User.objects.with_id(user_id)
+        check.delete()
+        return jsonify({"result":True})
+    except:
+        return jsonify({"result":False})
\ No newline at end of file
-- 
GitLab