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

update stream url if its changed

Browse Source
This commit is contained in:
Karl 2025-07-15 19:21:55 +01:00
parent 900d3c1cfe
commit 796acd1b4e
3 changed files with 46 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
.vscode/settings.json vendored
View File

@ -1,6 +1,6 @@
{ {
"[python]": { "[python]": {
"editor.defaultFormatter": "ms-python.black-formatter" "editor.defaultFormatter": "mikoz.black-py"
}, },
"python.formatting.provider": "none" "python.formatting.provider": "none"
} }

39
ktvmanager/lib/checker.py
View File

@ -88,6 +88,7 @@ def validate_account() -> Tuple[Response, int]:
password = data.get("password") password = data.get("password")
expiry_date = data.get("expiry_date") expiry_date = data.get("expiry_date")
stream = data.get("stream") stream = data.get("stream")
stream_url = data.get("streamURL")
if not all([username, password]): if not all([username, password]):
return jsonify({"message": "Missing required fields"}), 400 return jsonify({"message": "Missing required fields"}), 400
@ -97,20 +98,46 @@ def validate_account() -> Tuple[Response, int]:
result = single_account_check(account_data, stream_urls) result = single_account_check(account_data, stream_urls)
if result: if result:
if expiry_date and stream and int(result["data"]["user_info"]["exp_date"]) != expiry_date: if result["url"] != stream_url:
from ktvmanager.lib.database import update_stream_url
update_stream_url(result["url"], stream_url)
return (
jsonify({"message": "Account is valid and updated", "data": result}),
200,
)
if (
expiry_date
and stream
and int(result["data"]["user_info"]["exp_date"]) != expiry_date
):
from ktvmanager.lib.database import update_expiry_date from ktvmanager.lib.database import update_expiry_date
update_expiry_date( update_expiry_date(
username, stream, result["data"]["user_info"]["exp_date"] username, stream, result["data"]["user_info"]["exp_date"]
) )
return jsonify({"message": "Account is valid and updated", "data": result}), 200 return (
jsonify({"message": "Account is valid and updated", "data": result}),
200,
)
if int(result.get("data", {}).get("user_info", {}).get("max_connections")) and data.get("max_connections") and int(result["data"]["user_info"]["max_connections"]) != data.get("max_connections"): if (
int(result.get("data", {}).get("user_info", {}).get("max_connections"))
and data.get("max_connections")
and int(result["data"]["user_info"]["max_connections"])
!= data.get("max_connections")
):
from ktvmanager.lib.database import update_max_connections from ktvmanager.lib.database import update_max_connections
update_max_connections( update_max_connections(
username, stream, int(result["data"]["user_info"]["max_connections"]) username, stream, int(result["data"]["user_info"]["max_connections"])
) )
return jsonify({"message": "Account is valid and updated", "data": result}), 200 return (
jsonify({"message": "Account is valid and updated", "data": result}),
200,
)
return jsonify({"message": "Account is valid", "data": result}), 200 return jsonify({"message": "Account is valid", "data": result}), 200
else: else:
return jsonify({"message": "Account is invalid"}), 401 return jsonify({"message": "Account is invalid"}), 401

12
ktvmanager/lib/database.py
View File

@ -167,6 +167,18 @@ def update_max_connections(username: str, stream: str, max_connections: int) ->
_execute_query(query, params) _execute_query(query, params)
def update_stream_url(new_stream: str, old_stream: str) -> None:
"""Updates the stream URL of an account.
Args:
new_stream: The stream of the account.
old_stream: The new stream URL.
"""
query = "UPDATE userAccounts SET streamURL = %s WHERE streamURL = %s"
params = (new_stream, old_stream)
_execute_query(query, params)
def delete_account(user_id: int) -> Response: def delete_account(user_id: int) -> Response:
"""Deletes an account for a user. """Deletes an account for a user.