PATCH: Fixed VoDs not generating after stream end

This commit is contained in:
2025-03-04 13:56:05 +00:00
parent 5b5bc161f1
commit b6b7d5643d
3 changed files with 6 additions and 4 deletions

View File

@@ -274,6 +274,7 @@ def publish_stream():
stream_key = data.get("stream_key") stream_key = data.get("stream_key")
stream_title = data.get("title") stream_title = data.get("title")
stream_category = data.get("category_name") stream_category = data.get("category_name")
stream_thumbnail = data.get("thumbnail")
user_id = None user_id = None
username = None username = None
@@ -327,6 +328,7 @@ def update_stream():
stream_category = data.get("category_name") stream_category = data.get("category_name")
stream_thumbnail = data.get("thumbnail") stream_thumbnail = data.get("thumbnail")
with Database() as db: with Database() as db:
user_info = db.fetchone("""SELECT user_id, username, is_live user_info = db.fetchone("""SELECT user_id, username, is_live
FROM users FROM users

View File

@@ -1,7 +1,7 @@
from celery import Celery, shared_task, Task from celery import Celery, shared_task, Task
from datetime import datetime from datetime import datetime
from celery_tasks.preferences import user_preferences from celery_tasks.preferences import user_preferences
from utils.stream_utils import generate_thumbnail, get_streamer_live_status, get_custom_thumbnail_status from utils.stream_utils import generate_thumbnail, get_streamer_live_status, get_custom_thumbnail_status, remove_hls_files
from time import sleep from time import sleep
from os import listdir, remove from os import listdir, remove
import subprocess import subprocess
@@ -50,6 +50,9 @@ def combine_ts_stream(stream_path, vods_path, vod_file_name):
subprocess.run(vod_command) subprocess.run(vod_command)
# Remove HLS files, even if user is not streaming
remove_hls_files(stream_path)
@shared_task @shared_task
def convert_image_to_png(image_path, png_path): def convert_image_to_png(image_path, png_path):
""" """

View File

@@ -80,9 +80,6 @@ def end_user_stream(stream_key, user_id, username):
stream_info = db.fetchone("""SELECT * stream_info = db.fetchone("""SELECT *
FROM streams FROM streams
WHERE user_id = ?""", (user_id,)) WHERE user_id = ?""", (user_id,))
# Remove HLS files, even if user is not streaming
remove_hls_files(path_manager.get_stream_path(username))
# If user is not streaming, just return # If user is not streaming, just return
if not stream_info: if not stream_info: