You cannot select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
dirty-helpers/autopyvideothumbnailer

105 lines
3.0 KiB
Python

#!/usr/bin/env python3
# call pyvideothumbnailer for every video file found
#
# calculates rows based on the duration of the video
import math
import os
import subprocess
import click
from pymediainfo import MediaInfo
VERBOSE = False
def is_video_file(video_file_path, video_extensions):
return video_file_path.lower().endswith(tuple(video_extensions))
def find_video_files(directory, video_extensions):
# Recursively walk through directory
for dirpath, dirs, files in os.walk(directory):
for filename in files:
filename_path = os.path.join(dirpath, filename)
if is_video_file(filename_path, video_extensions):
yield os.path.abspath(filename_path)
def get_video_duration(video_file_path):
# Get media information
media_info = MediaInfo.parse(video_file_path)
# Iterate through tracks in the media file
for track in media_info.tracks:
# If the track type is video
if track.track_type == "Video":
# Get duration and return it
duration_str = track.duration
duration_sec = float(duration_str) / 1000 # Convert to seconds
return duration_sec
# Return None if no video track found
return None
def call_thumbnailer(video_file, width, rows, *, skip_seconds=10):
# Call external thumbnailer program
subprocess.run(
[
"pyvideothumbnailer",
"--width",
str(width),
"--rows",
str(rows),
"--jpeg-quality",
str(90),
"--skip-seconds",
str(skip_seconds),
video_file,
]
)
def process(searchdir):
VIDEO_FILE_EXT = [".mp4", ".mkv", ".avi", ".mov", ".wmv"]
video_files = find_video_files(searchdir, VIDEO_FILE_EXT)
for video_file in video_files:
# Skip if .jpg file already exists
thumbnails_file = video_file + ".jpg"
if os.path.exists(thumbnails_file):
if VERBOSE:
print(f"Skipping: {video_file} (thumbnails already exist)")
continue
try:
duration = get_video_duration(video_file)
if duration is not None:
# Calculate rows
COLS = 4 # pyvideothumbnailer default
EVERY = 5 * 60 # seconds
rows = math.ceil(duration / EVERY / COLS)
skip_seconds = 10
if duration < 60:
skip_seconds = 0
# Call the thumbnailer
WIDTH = 1600
call_thumbnailer(video_file, WIDTH, rows, skip_seconds=skip_seconds)
except (TypeError, UnicodeEncodeError, FileNotFoundError) as e:
print(f"Error for {video_file}: {e}")
@click.command()
@click.argument('searchdirs', nargs=-1)
def main(searchdirs):
"""Run pyvideothumbnailer for every video file found"""
if not searchdirs:
searchdirs=["."]
for searchdir in searchdirs:
process(searchdir)
if __name__ == "__main__":
main()