import os
from thumb_gen import Generator

folder = 'C:/input'
for video in os.listdir(folder):
    if video.endswith('.mp4') or video.endswith('.mkv'):
        app = Generator(os.path.join(folder, video), custom_text=False, font_dir="C:/Project/font.ttf", font_size=25, bg_colour='blue', font_colour='red')
        app.run()

This code is very useful, but I want to send more veriables, eg: row size of images, or columns size of images.

Another issue is with custom text. Custom text is under images when it's too long.

Hi there, thanks for this nice script. I've modified it to work with more formats, the only problem I've found where the 4 digit extensions of "webm", it seems the script used self.output_path = self.video_path[:-4] which removed 4 last characters to add a new extension ("." included). The webm extension has 4 letters + 1 ".", meaning that output thumb files where named "file..jpg" if they where made from a .webm source (this is because self.output_path = self.video_path[:-4] would not remove the "." character)

Here are the small modifications I've made in case you want to modify the script:

worker.py:

        if output_path == '':
            if video_path.endswith('webm'):
                self.output_path = self.video_path[:-5]
                self.output_folder = listToString(re.split(pattern = r"[/\\]", string = self.video_path)[:-1], "sys")
            else:
                self.output_path = self.video_path[:-4]
                self.output_folder = listToString(re.split(pattern = r"[/\\]", string = self.video_path)[:-1], "sys")

        else:
            if video_path.endswith('webm'):
                self.filename = re.split(pattern = r"[/\\]", string = self.video_path)[-1]
                self.output_path = os.path.join(output_path, self.filename[:-5])
                self.output_folder = self.output_path
            else:
                self.filename = re.split(pattern = r"[/\\]", string = self.video_path)[-1]
                self.output_path = os.path.join(output_path, self.filename[:-4])
                self.output_folder = self.output_path

main.py:

def check_files(paths_or_files):
    videos = []
    current_directory = os.getcwd()

    for path_or_file in paths_or_files:
        if not os.path.exists(path_or_file):
            if not os.path.exists(os.path.join(current_directory, path_or_file)):
                sys.exit("{}: no such file or directory".format(path_or_file))
            else:
                real_path = os.path.join(current_directory, path_or_file)
        else:
            if os.path.isfile(path_or_file):
                real_path = path_or_file
            else:
                real_path = os.path.join(current_directory, path_or_file)

        if os.path.isfile(real_path):
            if real_path.endswith(('mp4', 'mkv', 'm4v', 'webm', 'avi', 'wmv')):
                videos.append(real_path)
            else:
                sys.exit("{}: file not supported".format(real_path))
        elif os.path.isdir(real_path):
            for file in os.listdir(real_path):
                if file.endswith(('mp4', 'mkv', 'm4v', 'webm', 'avi', 'wmv')):
                    videos.append(os.path.join(real_path, file))
            if videos == []:
                sys.exit("{}: all of files in the directory are not supported".format(real_path))

    return videos