使用 Python和 FFmpeg 批量截图视频到各自文件夹中

winfredzhang2024-08-24 18:56

在这篇博客中，我们将创建一个简单的图形用户界面 (GUI) 工具，利用 wxPython 和 FFmpeg 来从视频文件中批量生成截图。这个工具能够让用户选择一个文件夹，遍历其中的所有视频文件，按照视频长度将其分为四等分，然后为每个视频生成四张截图。所有生成的截图将保存在一个以视频名称命名的文件夹中，并在截图完成后自动打开该文件夹。

C:\pythoncode\new\multivideofilescreenshot.py

工具介绍
  • wxPython：用于创建桌面应用程序的图形界面。
  • FFmpeg：一个强大的多媒体处理工具，用于提取视频帧。

所有代码

Python 复制代码 
import wx
import os
import subprocess
import threading
import datetime
import sys

class VideoScreenshotApp(wx.Frame):
    def __init__(self):
        wx.Frame.__init__(self, None, title="视频截图工具", size=(600, 400))

        # 创建面板
        panel = wx.Panel(self)

        # 创建路径选择控件
        self.path_label = wx.StaticText(panel, label="请选择文件夹：")
        self.path_textctrl = wx.TextCtrl(panel, style=wx.TE_READONLY)
        self.path_button = wx.Button(panel, label="选择路径")
        self.path_button.Bind(wx.EVT_BUTTON, self.on_select_path)

        # 创建文件列表控件，使用 CheckListBox 以支持多选
        self.file_list_ctrl = wx.CheckListBox(panel)

        # 创建截图按钮
        self.capture_button = wx.Button(panel, label="截图")
        self.capture_button.Bind(wx.EVT_BUTTON, self.on_capture)

        # 布局
        sizer = wx.BoxSizer(wx.VERTICAL)
        sizer.Add(self.path_label, 0, wx.ALL, 5)
        sizer.Add(self.path_textctrl, 0, wx.EXPAND | wx.ALL, 5)
        sizer.Add(self.path_button, 0, wx.ALL, 5)
        sizer.Add(self.file_list_ctrl, 1, wx.EXPAND | wx.ALL, 5)
        sizer.Add(self.capture_button, 0, wx.ALL | wx.ALIGN_CENTER, 5)
        panel.SetSizer(sizer)

        self.current_path = ""

    def on_select_path(self, event):
        dlg = wx.DirDialog(self, "选择文件夹", style=wx.DD_DEFAULT_STYLE)
        if dlg.ShowModal() == wx.ID_OK:
            self.current_path = dlg.GetPath()
            self.path_textctrl.SetValue(self.current_path)
            self.update_file_list()
        dlg.Destroy()

    def update_file_list(self):
        self.file_list_ctrl.Clear()

        if not self.current_path:
            return

        file_list = self.search_video_files(self.current_path)
        for filename, file_path, duration in file_list:
            self.file_list_ctrl.Append(f"{filename} - {str(datetime.timedelta(seconds=int(duration)))}", file_path)

    def search_video_files(self, directory):
        video_extensions = ['.mp4', '.avi', '.mkv', '.mov', '.wmv', '.flv', '.webm']
        file_list = []

        for root, dirs, files in os.walk(directory):
            for file in files:
                if os.path.splitext(file)[1].lower() in video_extensions:
                    file_path = os.path.join(root, file)
                    duration = self.get_video_duration(file_path)
                    file_list.append((file, file_path, duration))

        return file_list

    def get_video_duration(self, file_path):
        cmd = [
            'ffprobe',
            '-v', 'error',
            '-select_streams', 'v:0',
            '-show_entries', 'stream=duration',
            '-of', 'default=noprint_wrappers=1:nokey=1',
            file_path
        ]
        
        try:
            result = subprocess.run(cmd, stdout=subprocess.PIPE, stderr=subprocess.PIPE, text=True, check=True)
            duration_str = result.stdout.strip()
            
            if not duration_str:
                raise ValueError("ffprobe output is empty")

            return float(duration_str)
        
        except subprocess.CalledProcessError as e:
            wx.LogError(f"ffprobe error: {e.stderr}")
            raise
        except ValueError as e:
            wx.LogError(f"Value error: {e}")
            raise

    def on_capture(self, event):
        selected_indices = self.file_list_ctrl.GetCheckedItems()
        if selected_indices:
            for index in selected_indices:
                file_path = self.file_list_ctrl.GetClientData(index)
                file_name = os.path.basename(file_path)
                duration = self.get_video_duration(file_path)
                thread = threading.Thread(target=self.capture_screenshots, args=(file_path, duration))
                thread.start()
        else:
            wx.MessageBox("请先选择一个或多个视频文件", "提示", wx.OK | wx.ICON_INFORMATION)

    def capture_screenshots(self, file_path, duration):
        output_dir = os.path.join(self.current_path, os.path.splitext(os.path.basename(file_path))[0])
        if not os.path.exists(output_dir):
            os.makedirs(output_dir)

        # 计算每张截图的时间点
        intervals = [duration * i / 4 for i in range(1, 5)]

        # 生成截图
        for i, timestamp in enumerate(intervals):
            cmd = [
                'ffmpeg',
                '-ss', str(datetime.timedelta(seconds=int(timestamp))),
                '-i', file_path,
                '-vframes', '1',
                os.path.join(output_dir, f'screenshot_{i+1}.jpg')
            ]
            subprocess.run(cmd, check=True)

        # 截图完成后，自动打开文件夹
        if sys.platform.startswith('win'):
            subprocess.Popen(['explorer', output_dir])
        elif sys.platform.startswith('darwin'):
            subprocess.Popen(['open', output_dir])
        elif sys.platform.startswith('linux'):
            subprocess.Popen(['xdg-open', output_dir])


if __name__ == "__main__":
    app = wx.App(False)
    frame = VideoScreenshotApp()
    frame.Show()
    app.MainLoop()
代码实现

下面是我们的工具实现代码：

python 复制代码 
import wx
import os
import subprocess
import threading
import datetime
import sys

class VideoScreenshotApp(wx.Frame):
    def __init__(self):
        wx.Frame.__init__(self, None, title="视频截图工具", size=(600, 400))

        # 创建面板
        panel = wx.Panel(self)

        # 创建路径选择控件
        self.path_label = wx.StaticText(panel, label="请选择文件夹：")
        self.path_textctrl = wx.TextCtrl(panel, style=wx.TE_READONLY)
        self.path_button = wx.Button(panel, label="选择路径")
        self.path_button.Bind(wx.EVT_BUTTON, self.on_select_path)

        # 创建文件列表控件，使用 CheckListBox 以支持多选
        self.file_list_ctrl = wx.CheckListBox(panel)

        # 创建截图按钮
        self.capture_button = wx.Button(panel, label="截图")
        self.capture_button.Bind(wx.EVT_BUTTON, self.on_capture)

        # 布局
        sizer = wx.BoxSizer(wx.VERTICAL)
        sizer.Add(self.path_label, 0, wx.ALL, 5)
        sizer.Add(self.path_textctrl, 0, wx.EXPAND | wx.ALL, 5)
        sizer.Add(self.path_button, 0, wx.ALL, 5)
        sizer.Add(self.file_list_ctrl, 1, wx.EXPAND | wx.ALL, 5)
        sizer.Add(self.capture_button, 0, wx.ALL | wx.ALIGN_CENTER, 5)
        panel.SetSizer(sizer)

        self.current_path = ""

    def on_select_path(self, event):
        dlg = wx.DirDialog(self, "选择文件夹", style=wx.DD_DEFAULT_STYLE)
        if dlg.ShowModal() == wx.ID_OK:
            self.current_path = dlg.GetPath()
            self.path_textctrl.SetValue(self.current_path)
            self.update_file_list()
        dlg.Destroy()

    def update_file_list(self):
        self.file_list_ctrl.Clear()

        if not self.current_path:
            return

        file_list = self.search_video_files(self.current_path)
        for filename, file_path, duration in file_list:
            self.file_list_ctrl.Append(f"{filename} - {str(datetime.timedelta(seconds=int(duration)))}", file_path)

    def search_video_files(self, directory):
        video_extensions = ['.mp4', '.avi', '.mkv', '.mov', '.wmv', '.flv', '.webm']
        file_list = []

        for root, dirs, files in os.walk(directory):
            for file in files:
                if os.path.splitext(file)[1].lower() in video_extensions:
                    file_path = os.path.join(root, file)
                    duration = self.get_video_duration(file_path)
                    file_list.append((file, file_path, duration))

        return file_list

    def get_video_duration(self, file_path):
        cmd = [
            'ffprobe',
            '-v', 'error',
            '-select_streams', 'v:0',
            '-show_entries', 'stream=duration',
            '-of', 'default=noprint_wrappers=1:nokey=1',
            file_path
        ]
        
        try:
            result = subprocess.run(cmd, stdout=subprocess.PIPE, stderr=subprocess.PIPE, text=True, check=True)
            duration_str = result.stdout.strip()
            
            if not duration_str:
                raise ValueError("ffprobe output is empty")

            return float(duration_str)
        
        except subprocess.CalledProcessError as e:
            wx.LogError(f"ffprobe error: {e.stderr}")
            raise
        except ValueError as e:
            wx.LogError(f"Value error: {e}")
            raise

    def on_capture(self, event):
        selected_indices = self.file_list_ctrl.GetCheckedItems()
        if selected_indices:
            for index in selected_indices:
                file_path = self.file_list_ctrl.GetClientData(index)
                file_name = os.path.basename(file_path)
                duration = self.get_video_duration(file_path)
                thread = threading.Thread(target=self.capture_screenshots, args=(file_path, duration))
                thread.start()
        else:
            wx.MessageBox("请先选择一个或多个视频文件", "提示", wx.OK | wx.ICON_INFORMATION)

    def capture_screenshots(self, file_path, duration):
        output_dir = os.path.join(self.current_path, os.path.splitext(os.path.basename(file_path))[0])
        if not os.path.exists(output_dir):
            os.makedirs(output_dir)

        # 计算每张截图的时间点
        intervals = [duration * i / 4 for i in range(1, 5)]

        # 生成截图
        for i, timestamp in enumerate(intervals):
            cmd = [
                'ffmpeg',
                '-ss', str(datetime.timedelta(seconds=int(timestamp))),
                '-i', file_path,
                '-vframes', '1',
                os.path.join(output_dir, f'screenshot_{i+1}.jpg')
            ]
            subprocess.run(cmd, check=True)

        # 截图完成后，自动打开文件夹
        if sys.platform.startswith('win'):
            subprocess.Popen(['explorer', output_dir])
        elif sys.platform.startswith('darwin'):
            subprocess.Popen(['open', output_dir])
        elif sys.platform.startswith('linux'):
            subprocess.Popen(['xdg-open', output_dir])


if __name__ == "__main__":
    app = wx.App(False)
    frame = VideoScreenshotApp()
    frame.Show()
    app.MainLoop()
代码解释

  1. 创建主窗口：

    • 使用 wx.Frame 创建主窗口，添加路径选择控件、文件列表控件以及截图按钮。

  2. 路径选择：

    • on_select_path 方法允许用户选择一个文件夹，并更新文件列表。

  3. 文件列表更新：

    • update_file_list 方法遍历所选文件夹中的视频文件，获取每个视频的时长，并将信息显示在 CheckListBox 中。

  4. 视频时长获取：

    • get_video_duration 方法使用 ffprobe 命令来获取视频时长。

  5. 截图生成：

    • on_capture 方法处理截图请求，使用多线程来生成截图，以避免阻塞主线程。
    • capture_screenshots 方法使用 ffmpeg 命令生成四张截图，并将截图保存在以视频名称命名的文件夹中。

  6. 自动打开文件夹：

    • 截图完成后，自动在文件浏览器中打开保存截图的文件夹。

效果如下

总结

通过这个工具，你可以轻松地从多个视频文件中生成截图，而无需手动操作。wxPython 提供了一个简单易用的界面，而 FFmpeg 则负责处理视频帧的提取。这个工具不仅对视频编辑工作有帮助，也为批量处理视频文件提供了极大的便利。

相关推荐
用户83562907805113 分钟前
Python 实现 PDF 文件加密与解密方法
后端·python
用户83562907805118 分钟前
使用 Python 冻结与拆分 Excel 窗格教程
后端·python
你好潘先生8 小时前
别再记命令了，用 yeero do 说句人话就能跑脚本，而且不烧 token
服务器·python·命令行
Agent_大师9 小时前
WebSocket 行情重连成功，K线缺口不会自动消失
python
荣码9 小时前
LLM结构化输出：让AI返回JSON而不是废话，我踩了4个坑
java·python
copyer_xyf9 小时前
FastAPI 如何连接 MySQL
后端·python
apocelipes1 天前
常用编程语言和库的正则表达式性能对比
c语言·c++·python·性能优化·golang·开发工具和环境
用户8356290780511 天前
使用 Python 在 PDF 中创建与管理书签
后端·python
MeixianAgent1 天前
Python 回测数据入口怎么验？历史 K 线入库前先做 5 个检查
后端·python
咕白m6251 天前
用 Python 实现一键批量查找与替换 Excel 数据
后端·python
热门推荐
012026 年 AI 编程工具终极横评：Cursor vs Claude Code vs Copilot vs Windsurf02GitHub 镜像站点03【AI】2026 年具身智能模型和世界模型总结042026年6月AI大模型全景报告：GPT-5.6、Claude Opus 4.8、Gemini 3.5，中美AI三足鼎立谁主沉浮？052026 AI 编程工具终极实战指南：Cursor vs Claude Code vs Copilot，开发者该怎么选？06Claude Code、Codex、Cursor三分天下：2026年AI编程Agent生态全景剖析072026 年 AI 大模型 & AI 编程工具实战全总结082026年6月AI行业全景：从百模大战到Agent元年，这30天发生了什么？09AI科技热点日报 | 2026年07月01日10Trae国际版与国内版深度测评：AI原生IDE的双生花