66 lines
2.3 KiB
Python
66 lines
2.3 KiB
Python
import json
|
|
import subprocess
|
|
import click
|
|
from pathlib import Path
|
|
# from rich import print
|
|
|
|
from models.streams import VideoFile, VideoStream, AudioStream, SubtitleStream
|
|
from movie_details import get_movie_details
|
|
from lib.fs import write_shell_script
|
|
|
|
|
|
def parse_streams(stream_data: dict, VideoFile: VideoFile) -> VideoFile:
|
|
for stream in stream_data["streams"]:
|
|
if (
|
|
stream["codec_type"] == "video"
|
|
and stream["disposition"]["attached_pic"] == 0
|
|
):
|
|
VideoFile.video_streams.append(VideoStream(**stream))
|
|
elif stream["codec_type"] == "audio":
|
|
VideoFile.audio_streams.append(AudioStream(**stream))
|
|
elif stream["codec_type"] == "subtitle":
|
|
VideoFile.subtitle_streams.append(SubtitleStream(**stream))
|
|
return VideoFile
|
|
|
|
|
|
@click.command()
|
|
@click.argument("filename", type=click.Path(exists=True))
|
|
@click.option("-i", "--imdb", type=str, help="IMDB ID")
|
|
@click.option("-ns", "--nosubtitles", is_flag=True, help="No subtitles")
|
|
@click.option("-p", "--preset", type=int, default=2, help="Encoding preset (default: 2)")
|
|
@click.option("-c", "--crf", type=int, default=30, help="CRF value (default: 30)")
|
|
@click.option('-o', '--original-media-type', default='DVD', help='Set the ORIGINAL_MEDIA_TYPE metadata')
|
|
def main(filename: Path, imdb: str, nosubtitles: bool, preset: int, crf: int, original_media_type: str):
|
|
"""Run ffprobe on a file and print the output in JSON format."""
|
|
video_details = VideoFile(
|
|
imdb=imdb, video_path=filename, allow_subtitle=not nosubtitles
|
|
)
|
|
video_details = get_movie_details(video_details)
|
|
|
|
cmd = [
|
|
"ffprobe",
|
|
"-v",
|
|
"quiet",
|
|
"-print_format",
|
|
"json",
|
|
"-show_format",
|
|
"-show_streams",
|
|
filename,
|
|
]
|
|
result = subprocess.run(cmd, capture_output=True, check=True)
|
|
data = parse_streams(json.loads(result.stdout), video_details)
|
|
for stream in data.video_streams:
|
|
stream.preset = preset
|
|
stream.crf = crf
|
|
stream.original_media_type = original_media_type
|
|
export_data = " ".join(data.cmd())
|
|
export_path = Path(filename).parent / "encode.sh"
|
|
write_shell_script(export_data, export_path)
|
|
print(Path(filename).parent)
|
|
print(" ".join(data.cmd()))
|
|
# json.dump(data, sys.stdout, indent=4)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|