Skip to content

Commit

Permalink
refactor(dl): Only log errors/warnings from mkvmerge, list after message
Browse files Browse the repository at this point in the history
  • Loading branch information
rlaphoenix committed May 16, 2024
1 parent 2acee30 commit 03f3fec
Show file tree
Hide file tree
Showing 2 changed files with 16 additions and 11 deletions.
18 changes: 11 additions & 7 deletions devine/commands/dl.py
Original file line number Diff line number Diff line change
Expand Up @@ -701,18 +701,22 @@ def result(
):
for task_id, task_tracks in multiplex_tasks:
progress.start_task(task_id) # TODO: Needed?
muxed_path, return_code, output = task_tracks.mux(
muxed_path, return_code, errors = task_tracks.mux(
str(title),
progress=partial(progress.update, task_id=task_id),
delete=False
)
muxed_paths.append(muxed_path)
if return_code == 1:
self.log.warning(output)
self.log.warning("mkvmerge had at least one warning, will continue anyway...")
elif return_code >= 2:
self.log.warning(output)
self.log.error(f"Failed to Mux video to Matroska file ({return_code})")
if return_code >= 2:
self.log.error(f"Failed to Mux video to Matroska file ({return_code}):")
elif return_code == 1 or errors:
self.log.warning("mkvmerge had at least one warning or error, continuing anyway...")
for line in errors:
if line.startswith("#GUI#error"):
self.log.error(line)
else:
self.log.warning(line)
if return_code >= 2:
sys.exit(1)
for video_track in task_tracks.videos:
video_track.delete()
Expand Down
9 changes: 5 additions & 4 deletions devine/core/tracks/tracks.py
Original file line number Diff line number Diff line change
Expand Up @@ -316,7 +316,7 @@ def by_language(tracks: list[TrackT], languages: list[str], per_language: int =
][:per_language or None])
return selected

def mux(self, title: str, delete: bool = True, progress: Optional[partial] = None) -> tuple[Path, int, str]:
def mux(self, title: str, delete: bool = True, progress: Optional[partial] = None) -> tuple[Path, int, list[str]]:
"""
Multiplex all the Tracks into a Matroska Container file.
Expand Down Expand Up @@ -410,17 +410,18 @@ def mux(self, title: str, delete: bool = True, progress: Optional[partial] = Non

# let potential failures go to caller, caller should handle
try:
output = ""
errors = []
p = subprocess.Popen([
*cl,
"--output", str(output_path),
"--gui-mode"
], text=True, stdout=subprocess.PIPE)
for line in iter(p.stdout.readline, ""):
output += line
if line.startswith("#GUI#error") or line.startswith("#GUI#warning"):
errors.append(line)
if "progress" in line:
progress(total=100, completed=int(line.strip()[14:-1]))
return output_path, p.wait(), output
return output_path, p.wait(), errors
finally:
if chapters_path:
# regardless of delete param, we delete as it's a file we made during muxing
Expand Down

0 comments on commit 03f3fec

Please sign in to comment.