Add ENABLE_VIDEO

This commit is contained in:
None 2025-05-14 11:07:30 +02:00
parent c0f3d8619b
commit 11b1a87c15
2 changed files with 13 additions and 3 deletions

View File

@ -33,6 +33,7 @@ from .segments import M3U8_Segments
# Config # Config
ENABLE_VIDEO = config_manager.get_bool('M3U8_DOWNLOAD', 'download_video')
ENABLE_AUDIO = config_manager.get_bool('M3U8_DOWNLOAD', 'download_audio') ENABLE_AUDIO = config_manager.get_bool('M3U8_DOWNLOAD', 'download_audio')
ENABLE_SUBTITLE = config_manager.get_bool('M3U8_DOWNLOAD', 'download_subtitle') ENABLE_SUBTITLE = config_manager.get_bool('M3U8_DOWNLOAD', 'download_subtitle')
DOWNLOAD_SPECIFIC_AUDIO = config_manager.get_list('M3U8_DOWNLOAD', 'specific_list_audio') DOWNLOAD_SPECIFIC_AUDIO = config_manager.get_list('M3U8_DOWNLOAD', 'specific_list_audio')
@ -171,6 +172,9 @@ class M3U8Manager:
logging.error("Resolution not recognized.") logging.error("Resolution not recognized.")
self.video_url, self.video_res = self.parser._video.get_best_uri() self.video_url, self.video_res = self.parser._video.get_best_uri()
if not ENABLE_VIDEO:
self.video_url = None
self.audio_streams = [] self.audio_streams = []
if ENABLE_AUDIO: if ENABLE_AUDIO:
self.audio_streams = [ self.audio_streams = [
@ -248,6 +252,10 @@ class DownloadManager:
def download_video(self, video_url: str): def download_video(self, video_url: str):
"""Downloads video segments from the M3U8 playlist.""" """Downloads video segments from the M3U8 playlist."""
if video_url is None:
console.print("[red]Video URL is None, skipping video download.[/red]")
return False
video_full_url = self.url_fixer.generate_full_url(video_url) video_full_url = self.url_fixer.generate_full_url(video_url)
video_tmp_dir = os.path.join(self.temp_dir, 'video') video_tmp_dir = os.path.join(self.temp_dir, 'video')
@ -262,8 +270,9 @@ class DownloadManager:
def download_audio(self, audio: Dict): def download_audio(self, audio: Dict):
"""Downloads audio segments for a specific language track.""" """Downloads audio segments for a specific language track."""
#if self.stopped: if audio['uri'] is None:
# return True console.print(f"[red]Audio URL is None for language {audio['language']}, skipping audio download.[/red]")
return False
audio_full_url = self.url_fixer.generate_full_url(audio['uri']) audio_full_url = self.url_fixer.generate_full_url(audio['uri'])
audio_tmp_dir = os.path.join(self.temp_dir, 'audio', audio['language']) audio_tmp_dir = os.path.join(self.temp_dir, 'audio', audio['language'])

View File

@ -26,10 +26,11 @@
}, },
"M3U8_DOWNLOAD": { "M3U8_DOWNLOAD": {
"tqdm_delay": 0.01, "tqdm_delay": 0.01,
"download_video": false,
"default_video_workser": 12, "default_video_workser": 12,
"default_audio_workser": 12, "default_audio_workser": 12,
"segment_timeout": 8, "segment_timeout": 8,
"download_audio": true, "download_audio": false,
"merge_audio": true, "merge_audio": true,
"specific_list_audio": [ "specific_list_audio": [
"ita" "ita"