Karl/KTVManager_Backend
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

show max connection next to stream url

Browse Source
This commit is contained in:
Karl 2025-07-15 17:54:37 +01:00
parent 28dadf0ca2
commit 2034b4211a
2 changed files with 23 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
ktvmanager/lib/checker.py
View File

@ -103,6 +103,14 @@ def validate_account() -> Tuple[Response, int]:
username, stream, result["data"]["user_info"]["exp_date"]
)
return jsonify({"message": "Account is valid and updated", "data": result}), 200
if result.get("data", {}).get("user_info", {}).get("max_connections") and data.get("max_connections") and result["data"]["user_info"]["max_connections"] != data.get("max_connections"):
from ktvmanager.lib.database import update_max_connections
update_max_connections(
username, stream, result["data"]["user_info"]["max_connections"]
)
return jsonify({"message": "Account is valid and updated", "data": result}), 200
return jsonify({"message": "Account is valid", "data": result}), 200
else:
return jsonify({"message": "Account is invalid"}), 401

16
ktvmanager/lib/database.py
View File

@ -127,7 +127,7 @@ def add_account(user_id: int) -> Response:
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)"
query = "INSERT INTO userAccounts (username, stream, streamURL, expiaryDate, password, userID, maxConnections) VALUES (%s, %s, %s, %s, %s, %s, %s)"
params = (
data["username"],
data["stream"],
@ -135,6 +135,7 @@ def add_account(user_id: int) -> Response:
res["data"]["user_info"]["exp_date"],
encrypted_password,
user_id,
res["data"]["user_info"]["max_connections"],
)
result = _execute_query(query, params)
return jsonify(result)
@ -153,6 +154,19 @@ def update_expiry_date(username: str, stream: str, expiry_date: str) -> None:
_execute_query(query, params)
def update_max_connections(username: str, stream: str, max_connections: int) -> None:
"""Updates the max connections of an account.
Args:
username: The username of the account.
stream: The stream of the account.
max_connections: The new max connections value.
"""
query = "UPDATE userAccounts SET maxConnections = %s WHERE username = %s AND stream = %s"
params = (max_connections, username, stream)
_execute_query(query, params)
def delete_account(user_id: int) -> Response:
"""Deletes an account for a user.