ninjajiraiya
/
devine
forked from DRMTalks/devine
1
0
Fork 0
Code Pull Requests Activity

Allow --audio/subs/chapters-only to be used simultaneously 

E.g., if you only wanted the subs and chapters, this would now be possible with `--subs-only --chapters-only`.
This commit is contained in:
rlaphoenix 2023-05-26 11:15:38 +01:00
parent 6a9598021d
commit 215730663b
1 changed files with 8 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
devine/commands/dl.py
View File

@ -51,7 +51,7 @@ from devine.core.service import Service
from devine.core.services import Services from devine.core.services import Services
from devine.core.titles import Movie, Song, Title_T from devine.core.titles import Movie, Song, Title_T
from devine.core.titles.episode import Episode from devine.core.titles.episode import Episode
from devine.core.tracks import Audio, Subtitle, Video from devine.core.tracks import Audio, Subtitle, Video, Tracks
from devine.core.utilities import get_binary_path, is_close_match, time_elapsed_since from devine.core.utilities import get_binary_path, is_close_match, time_elapsed_since
from devine.core.utils.click_types import LANGUAGE_RANGE, SEASON_RANGE, ContextData, QUALITY_LIST from devine.core.utils.click_types import LANGUAGE_RANGE, SEASON_RANGE, ContextData, QUALITY_LIST
from devine.core.utils.collections import merge_dict from devine.core.utils.collections import merge_dict
@ -437,22 +437,14 @@ class dl:
sys.exit(1) sys.exit(1)
if audio_only or subs_only or chapters_only: if audio_only or subs_only or chapters_only:
title.tracks.videos.clear() kept_tracks = []
if audio_only: if audio_only:
if not subs_only: kept_tracks.extend(title.tracks.audio)
title.tracks.subtitles.clear() if subs_only:
if not chapters_only: kept_tracks.extend(title.tracks.subtitles)
title.tracks.chapters.clear() if chapters_only:
elif subs_only: kept_tracks.extend(title.tracks.chapters)
if not audio_only: title.tracks = Tracks(kept_tracks)
title.tracks.audio.clear()
if not chapters_only:
title.tracks.chapters.clear()
elif chapters_only:
if not audio_only:
title.tracks.audio.clear()
if not subs_only:
title.tracks.subtitles.clear()
selected_tracks, tracks_progress_callables = title.tracks.tree(add_progress=True) selected_tracks, tracks_progress_callables = title.tracks.tree(add_progress=True)