dylan/gander
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

PATCH: Added check to see if stream ever published in /end_stream, otherwise errors occured

Browse Source
This commit is contained in:
Dylan De Faoite
2025-02-26 15:31:13 +00:00
parent e69f799caa
commit 3d05dc8571
1 changed files with 7 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
web_server/blueprints/streams.py
View File

@@ -173,7 +173,7 @@ def init_stream():
""" """
stream_key = request.form.get("name") stream_key = request.form.get("name")
print(f"Stream initialization requested in nginx with key: {stream_key}") print(f"Stream initialization requested in nginx with key: {stream_key}", flush=True)
with Database() as db: with Database() as db:
# Check if valid stream key and user is allowed to stream # Check if valid stream key and user is allowed to stream
@@ -314,7 +314,12 @@ def end_stream():
if not user_info: if not user_info:
print("Unauthorized - No user found from stream key", flush=True) print("Unauthorized - No user found from stream key", flush=True)
return "Unauthorized", 403 return "Unauthorized", 403
# If stream never published, return
if not stream_info:
print("Stream never began", flush=True)
return "Stream ended", 200
# Remove stream from database # Remove stream from database
db.execute("""DELETE FROM streams db.execute("""DELETE FROM streams
WHERE user_id = ?""", (user_info["user_id"],)) WHERE user_id = ?""", (user_info["user_id"],))