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

UPDATE: Thumbnail generation for VoDs implemented at /vods/<username>/<vod_id>.png

Browse Source
This commit is contained in:
Dylan De Faoite
2025-03-05 12:19:24 +00:00
parent 633776dfe9
commit cf7e6afe0b
2 changed files with 9 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
web_server/celery_tasks/streaming.py
View File

@@ -4,8 +4,11 @@ from celery_tasks.preferences import user_preferences
from utils.stream_utils import generate_thumbnail, get_streamer_live_status, get_custom_thumbnail_status, remove_hls_files 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
from utils.path_manager import PathManager
import subprocess import subprocess
path_manager = PathManager()
@shared_task @shared_task
def update_thumbnail(user_id, stream_file, thumbnail_file, sleep_time) -> None: def update_thumbnail(user_id, stream_file, thumbnail_file, sleep_time) -> None:
""" """
@@ -21,7 +24,7 @@ def update_thumbnail(user_id, stream_file, thumbnail_file, sleep_time) -> None:
print(f"Stopping thumbnail updates for stream of {user_id}") print(f"Stopping thumbnail updates for stream of {user_id}")
@shared_task @shared_task
def combine_ts_stream(stream_path, vods_path, vod_file_name): def combine_ts_stream(stream_path, vods_path, vod_file_name, thumbnail_file) -> None:
""" """
Combines all ts files into a single vod, and removes the ts files Combines all ts files into a single vod, and removes the ts files
""" """
@@ -53,6 +56,9 @@ def combine_ts_stream(stream_path, vods_path, vod_file_name):
# Remove HLS files, even if user is not streaming # Remove HLS files, even if user is not streaming
remove_hls_files(stream_path) remove_hls_files(stream_path)
# Generate thumbnail for vod
generate_thumbnail(f"{vods_path}/{vod_file_name}.mp4", thumbnail_file)
@shared_task @shared_task
def convert_image_to_png(image_path, png_path): def convert_image_to_png(image_path, png_path):
""" """

3
web_server/utils/stream_utils.py
View File

@@ -113,7 +113,8 @@ def end_user_stream(stream_key, user_id, username):
combine_ts_stream.delay( combine_ts_stream.delay(
path_manager.get_stream_path(username), path_manager.get_stream_path(username),
path_manager.get_vods_path(username), path_manager.get_vods_path(username),
vod_id vod_id,
path_manager.get_vod_thumbnail_file_path(username, vod_id)
) )
print(f"Stream ended for user {username} (ID: {user_id})", flush=True) print(f"Stream ended for user {username} (ID: {user_id})", flush=True)