Create screenshots with subtitles (requires pysubparser + ffmpeg)

bulk_screenshot.py

# bulk screenshotter // May 2021 // jeffehobbs@gmail.com

import os, subprocess
from pysubparser import parser

videos_dir = 'Videos/'
screenshot_dir = 'Screenshots/'
season = 'Season 1/'
video_file_extension = '.mp4'
subtitle_file_extension = '.eng.srt'
screenshot_file_extension = '.jpg'
episodes = []

for file in os.listdir(videos_dir + season):
	if file.endswith(video_file_extension):
		episodes.append(file)

for episode in episodes:
	video_file = videos_dir + season + episode
	subtitle_file = videos_dir + season + episode.replace(video_file_extension, subtitle_file_extension)
	subtitles = parser.parse(subtitle_file)
	for subtitle in subtitles:
		print(subtitle)
		timecode = subtitle.start
		screenshot_file = screenshot_dir + episode.replace(' ', '_').replace(video_file_extension,'') + str(timecode).replace(':','-') + screenshot_file_extension
		subprocess.call('ffmpeg -hide_banner -loglevel error -ss ' + str(timecode) + 
			            ' -copyts -i "' + video_file + 
			            '" -vf subtitles="' + subtitle_file + 
			            '" -vframes 1 -q:v 2 "' + screenshot_file + 
			            '" ', shell=True)