diff --git a/modules/register.py b/modules/register.py index 7d59def50ac18a352f30a0ee1d970da58ae9805f..34044621974bc701be4d111b5c6ded27d7befcf5 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