adding and deleting account is working

ktvmanager/lib/checker.py

@@ -6,7 +6,8 @@ def build_url(stream_url, username, password):
 
 def check_url(url):
     try:
-        tr = RequestsTor()
+        # tr = RequestsTor()
+        tr = requests
         response = tr.get(url, timeout=5)
         response.raise_for_status()
         if response.json().get("user_info", {}).get("auth"):

ktvmanager/lib/database.py

@@ -2,6 +2,7 @@ import mysql.connector.pooling
 from flask import jsonify, request, current_app
 from ktvmanager.lib.checker import single_account_check
 from ktvmanager.lib.encryption import encrypt_password, decrypt_password
+from ktvmanager.lib.get_urls import get_latest_urls_from_dns
 
 db_pool = None
 
@@ -65,17 +66,18 @@ def single_check():
         return jsonify(result)
     return jsonify({"message": "All checks failed"}), 400
 
-def add_account():
-    data = request.get_json()
+def add_account(user_id):
+    data = request.form
+    res = single_account_check(data, get_latest_urls_from_dns())
     encrypted_password = encrypt_password(data['password'])
     query = "INSERT INTO userAccounts (username, stream, streamURL, expiaryDate, password, userID) VALUES (%s, %s, %s, %s, %s, %s)"
-    params = (data['username'], data['stream'], data['streamURL'], data['expiaryDate'], encrypted_password, data['userID'])
+    params = (data['username'], data['stream'], res['url'], res['data']['user_info']['exp_date'], encrypted_password, user_id)
     result = _execute_query(query, params)
     return jsonify(result)
 
-def delete_account():
-    data = request.get_json()
-    query = "DELETE FROM userAccounts WHERE id = %s"
-    params = (data['id'],)
+def delete_account(user_id):
+    data = request.form
+    query = "DELETE FROM userAccounts WHERE username = %s AND stream = %s AND userId = %s"
+    params = (data['user'],data['stream'],user_id)
     result = _execute_query(query, params)
     return jsonify(result)

routes/api.py

@@ -31,9 +31,11 @@ def single_check_route(username):
 @api_blueprint.route("/addAccount", methods=["POST"])
 @requires_basic_auth
 def add_account_route(username):
-    return add_account()
+    user_id = get_user_id_from_username(username)
+    return add_account(user_id)
 
 @api_blueprint.route("/deleteAccount", methods=["POST"])
 @requires_basic_auth
 def delete_account_route(username):
-    return delete_account()
+    user_id = get_user_id_from_username(username)
+    return delete_account(user_id)