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

UPDATE: VoD Thumbnails now take a screenshot from exactly halfway through the stream, instead of the beginning

Browse Source
This commit is contained in:
Dylan De Faoite
2025-03-06 18:26:29 +00:00
parent d2ac393f3f
commit 4141fcd428
2 changed files with 37 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
web_server/celery_tasks/streaming.py
View File

@@ -1,7 +1,7 @@
from celery import Celery, shared_task, Task
from datetime import datetime
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, get_video_duration
from time import sleep
from os import listdir, remove
from utils.path_manager import PathManager
@@ -10,7 +10,7 @@ import subprocess
path_manager = PathManager()
@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, second_capture=0) -> None:
"""
Updates the thumbnail of a stream periodically
"""
@@ -19,7 +19,7 @@ def update_thumbnail(user_id, stream_file, thumbnail_file, sleep_time) -> None:
if get_streamer_live_status(user_id)['is_live'] and not get_custom_thumbnail_status(user_id)['custom_thumbnail']:
print("Updating thumbnail...")
generate_thumbnail(stream_file, thumbnail_file)
update_thumbnail.apply_async((user_id, stream_file, thumbnail_file, sleep_time), countdown=sleep_time)
update_thumbnail.apply_async((user_id, stream_file, thumbnail_file, sleep_time, second_capture), countdown=sleep_time)
else:
print(f"Stopping thumbnail updates for stream of {user_id}")
@@ -28,6 +28,8 @@ def combine_ts_stream(stream_path, vods_path, vod_file_name, thumbnail_file) ->
"""
Combines all ts files into a single vod, and removes the ts files
"""
vod_file_path = f"{vods_path}/{vod_file_name}.mp4"
ts_files = [f for f in listdir(stream_path) if f.endswith(".ts")]
ts_files.sort()
@@ -48,7 +50,7 @@ def combine_ts_stream(stream_path, vods_path, vod_file_name, thumbnail_file) ->
f"{stream_path}/list.txt",
"-c",
"copy",
f"{vods_path}/{vod_file_name}.mp4"
vod_file_path
]
subprocess.run(vod_command)
@@ -56,8 +58,12 @@ def combine_ts_stream(stream_path, vods_path, vod_file_name, thumbnail_file) ->
# Remove HLS files, even if user is not streaming
remove_hls_files(stream_path)
# Get video duration and choose middle frame as thumbnail
video_duration = get_video_duration(vod_file_path)
second_capture = video_duration // 2
# Generate thumbnail for vod
generate_thumbnail(f"{vods_path}/{vod_file_name}.mp4", thumbnail_file)
generate_thumbnail(vod_file_path, thumbnail_file, second_capture)
@shared_task
def convert_image_to_png(image_path, png_path):