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