diff --git a/src/app.py b/src/app.py index ca1e043..355bef8 100644 --- a/src/app.py +++ b/src/app.py @@ -3,7 +3,7 @@ import subprocess import typer from pathlib import Path -from models.streams import VideoFile, VideoStream, AudioStream, SubtitleStream +from models.streams import VideoFile, VideoStream, AudioStream, SubtitleStream, OriginalMediaType from movie_details import get_movie_details from lib.fs import write_shell_script @@ -30,7 +30,7 @@ def main( nosubtitles: bool = typer.Option(False, "-ns", "--nosubtitles", help="Remove subtitles"), preset: int = typer.Option(2, "-p", "--preset", help="Encoding preset"), crf: int = typer.Option(30, "-c", "--crf", help="CRF value"), - original_media_type: str = typer.Option("DVD", "-o", "--original-media-type", help="Set the ORIGINAL_MEDIA_TYPE metadata"), + original_media_type: OriginalMediaType = typer.Option(OriginalMediaType.DVD, "-o", "--original-media-type", case_sensitive=False, help="Source of the original media."), film_grain: int = typer.Option(20, "-fg", "--film-grain", help="Film grain amount (default: 20)") ): video_details = VideoFile( @@ -50,6 +50,7 @@ def main( ] result = subprocess.run(cmd, capture_output=True, check=True) data = parse_streams(json.loads(result.stdout), video_details) + data.original_media_type = original_media_type for stream in data.video_streams: stream.preset = preset stream.crf = crf