python

关注公众号 jb51net

关闭
首页 > 脚本专栏 > python > Python视频截图

使用Python创建一个视频管理器并实现视频截图功能

作者:winfredzhang

在这篇博客中,我将向大家展示如何使用 wxPython 创建一个简单的图形用户界面 (GUI) 应用程序,该应用程序可以管理视频文件列表、播放视频,并生成视频截图,我们将逐步实现这些功能,并确保代码易于理解和扩展,感兴趣的小伙伴跟着小编一起来看看吧

项目概述

本项目的目标是创建一个视频文件管理器应用,它能够:

所有代码

import wx
import os
import datetime
import subprocess
import sys
import cv2  # Ensure OpenCV is installed
import threading

class FileListFrame(wx.Frame):
    def __init__(self):
        wx.Frame.__init__(self, None, title="视频文件列表", size=(600, 400))

        self.panel = wx.Panel(self)
        self.current_path = ""

        self.file_list_ctrl = wx.ListCtrl(self.panel, style=wx.LC_REPORT | wx.LC_SINGLE_SEL)
        self.file_list_ctrl.InsertColumn(0, "文件名")
        self.file_list_ctrl.InsertColumn(1, "大小")
        self.file_list_ctrl.InsertColumn(2, "修改时间")
        self.file_list_ctrl.Bind(wx.EVT_LIST_ITEM_SELECTED, self.on_file_selected)
        self.file_list_ctrl.Bind(wx.EVT_LIST_ITEM_ACTIVATED, self.on_file_double_clicked)

        self.path_label = wx.StaticText(self.panel, label="路径:")
        self.path_textctrl = wx.TextCtrl(self.panel, style=wx.TE_READONLY)
        self.path_button = wx.Button(self.panel, label="选择路径")
        self.path_button.Bind(wx.EVT_BUTTON, self.on_select_path)

        self.interval_label = wx.StaticText(self.panel, label="截图间隔(秒):")
        self.interval_textctrl = wx.TextCtrl(self.panel, value="1")
        self.capture_button = wx.Button(self.panel, label="生成截图")
        self.capture_button.Bind(wx.EVT_BUTTON, self.on_capture)

        self.export_button = wx.Button(self.panel, label="导出为文本")
        self.export_button.Bind(wx.EVT_BUTTON, self.on_export)

        self.play_button = wx.Button(self.panel, label="播放")
        self.play_button.Bind(wx.EVT_BUTTON, self.on_play)

        sizer = wx.BoxSizer(wx.VERTICAL)
        sizer.Add(self.path_label, 0, wx.ALL, 5)
        sizer.Add(self.path_textctrl, 0, wx.EXPAND | wx.LEFT | wx.RIGHT, 5)
        sizer.Add(self.path_button, 0, wx.ALL, 5)
        sizer.Add(self.interval_label, 0, wx.ALL, 5)
        sizer.Add(self.interval_textctrl, 0, wx.EXPAND | wx.LEFT | wx.RIGHT, 5)
        sizer.Add(self.capture_button, 0, wx.ALL, 5)
        sizer.Add(self.file_list_ctrl, 1, wx.EXPAND | wx.ALL, 5)
        sizer.Add(self.export_button, 0, wx.ALL, 5)
        sizer.Add(self.play_button, 0, wx.ALL, 5)
        self.panel.SetSizer(sizer)

    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.DeleteAllItems()

        if not self.current_path:
            return

        file_list = self.search_video_files(self.current_path)
        for filename, file_path, file_size, modified_time in file_list:
            modified_time_str = datetime.datetime.fromtimestamp(modified_time).strftime("%Y-%m-%d %H:%M:%S")

            index = self.file_list_ctrl.InsertItem(self.file_list_ctrl.GetItemCount(), filename)
            self.file_list_ctrl.SetItem(index, 1, str(file_size))
            self.file_list_ctrl.SetItem(index, 2, modified_time_str)

    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)
                    file_size = os.path.getsize(file_path)
                    modified_time = os.path.getmtime(file_path)
                    file_list.append((file, file_path, file_size, modified_time))

        return file_list

    def on_file_selected(self, event):
        selected_item = event.GetItem()
        file_name = selected_item.GetText()
        file_path = os.path.join(self.current_path, file_name)

        video = cv2.VideoCapture(file_path)
        fps = video.get(cv2.CAP_PROP_FPS)
        frame_count = int(video.get(cv2.CAP_PROP_FRAME_COUNT))
        duration = frame_count / fps
        video.release()

        duration_str = str(datetime.timedelta(seconds=int(duration)))

        wx.MessageBox(
            f"文件名: {file_name}\n时长: {duration_str}",
            "视频信息", wx.OK | wx.ICON_INFORMATION)

    def on_file_double_clicked(self, event):
        self.on_play(event)

    def on_play(self, event):
        selected_item = self.file_list_ctrl.GetFirstSelected()
        if selected_item != -1:
            file_name = self.file_list_ctrl.GetItemText(selected_item)
            file_path = os.path.join(self.current_path, file_name)
            if sys.platform.startswith('win'):
                subprocess.Popen(['start', '', file_path], shell=True)
            elif sys.platform.startswith('darwin'):
                subprocess.Popen(['open', file_path])
            elif sys.platform.startswith('linux'):
                subprocess.Popen(['xdg-open', file_path])
        else:
            wx.MessageBox("请先选择要播放的文件", "提示", wx.OK | wx.ICON_INFORMATION)


    def on_export(self, event):
        dlg = wx.FileDialog(self, "保存为文本文件", wildcard="Text files (*.txt)|*.txt",
                            style=wx.FD_SAVE | wx.FD_OVERWRITE_PROMPT)

        if dlg.ShowModal() == wx.ID_OK:
            file_path = dlg.GetPath()
            with open(file_path, 'w') as file:
                for index in range(self.file_list_ctrl.GetItemCount()):
                    file.write(self.file_list_ctrl.GetItemText(index) + '\n')

    def on_capture(self, event):
        selected_item = self.file_list_ctrl.GetFirstSelected()
        if selected_item != -1:
            file_name = self.file_list_ctrl.GetItemText(selected_item)
            file_path = os.path.join(self.current_path, file_name)
            try:
                interval = int(self.interval_textctrl.GetValue())
            except ValueError:
                wx.MessageBox("请输入有效的时间间隔(秒)", "错误", wx.OK | wx.ICON_ERROR)
                return

            thread = threading.Thread(target=self.capture_screenshots, args=(file_path, interval))
            thread.start()
        else:
            wx.MessageBox("请先选择要生成截图的文件", "提示", wx.OK | wx.ICON_INFORMATION)

    def capture_screenshots(self, file_path, interval):
        # 生成以视频名称命名的文件夹
        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)

        # 构造 ffmpeg 命令
        cmd = [
            'ffmpeg',
            '-i', file_path,                 # 输入视频文件
            '-vf', f'fps=1/{interval}',      # 每隔 {interval} 秒截取一帧
            os.path.join(output_dir, 'screenshot_%03d.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])

    # def capture_screenshots(self, file_path, interval):
    #     video = cv2.VideoCapture(file_path)
    #     fps = video.get(cv2.CAP_PROP_FPS)
    #     frame_count = int(video.get(cv2.CAP_PROP_FRAME_COUNT))
    #     duration = frame_count / fps

    #     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)

    #     for sec in range(0, int(duration), interval):
            video.set(cv2.CAP_PROP_POS_MSEC, sec * 1000)
            success, image = video.read()
            if success:
                cv2.imwrite(os.path.join(output_dir, f"{sec}.png"), image)

        video.release()
        wx.CallAfter(wx.MessageBox, "截图已生成", "完成", wx.OK | wx.ICON_INFORMATION)
        
        # Automatically open the folder containing screenshots
        if sys.platform.startswith('win'):
            subprocess.Popen(['explorer', output_dir], shell=True)
        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()
    frame = FileListFrame()
    frame.Show()
    app.MainLoop()

项目实现

让我们一步一步实现这个项目。

1. 环境准备

首先,你需要确保系统中安装了以下工具:

pip install wxPython

2. 创建主窗口类

首先,我们创建一个主窗口类 FileListFrame,用于展示视频文件列表并处理用户交互。

import wx
import os
import datetime
import subprocess
import sys
import threading

class FileListFrame(wx.Frame):
    def __init__(self):
        wx.Frame.__init__(self, None, title="视频文件管理器", size=(600, 400))

        self.panel = wx.Panel(self)
        self.current_path = ""

        # 创建文件列表控件
        self.file_list_ctrl = wx.ListCtrl(self.panel, style=wx.LC_REPORT | wx.LC_SINGLE_SEL)
        self.file_list_ctrl.InsertColumn(0, "文件名")
        self.file_list_ctrl.InsertColumn(1, "时长")
        self.file_list_ctrl.InsertColumn(2, "大小")
        self.file_list_ctrl.InsertColumn(3, "修改时间")
        self.file_list_ctrl.Bind(wx.EVT_LIST_ITEM_SELECTED, self.on_file_selected)
        self.file_list_ctrl.Bind(wx.EVT_LIST_ITEM_ACTIVATED, self.on_play)

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

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

        # 创建截图时间间隔输入框
        self.interval_label = wx.StaticText(self.panel, label="截图时间间隔(秒):")
        self.interval_textctrl = wx.TextCtrl(self.panel)

        # 创建布局
        sizer = wx.BoxSizer(wx.VERTICAL)
        sizer.Add(self.path_label, 0, wx.ALL, 5)
        sizer.Add(self.path_textctrl, 0, wx.EXPAND | wx.LEFT | wx.RIGHT, 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.interval_label, 0, wx.ALL, 5)
        sizer.Add(self.interval_textctrl, 0, wx.EXPAND | wx.ALL, 5)
        sizer.Add(self.capture_button, 0, wx.ALL, 5)
        self.panel.SetSizer(sizer)

在 __init__ 方法中,我们初始化了主窗口,并创建了一个 ListCtrl 控件用于显示视频文件列表。还添加了用于选择路径的按钮和输入截图时间间隔的文本框。

3. 列出视频文件

接下来,我们实现选择路径和列出视频文件的功能:

    # 处理选择路径事件
    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.DeleteAllItems()

        if not self.current_path:
            return

        file_list = self.search_video_files(self.current_path)
        for filename, file_path, file_size, modified_time in file_list:
            modified_time_str = datetime.datetime.fromtimestamp(modified_time).strftime("%Y-%m-%d %H:%M:%S")

            index = self.file_list_ctrl.InsertItem(self.file_list_ctrl.GetItemCount(), filename)
            self.file_list_ctrl.SetItem(index, 1, "待获取")
            self.file_list_ctrl.SetItem(index, 2, str(file_size))
            self.file_list_ctrl.SetItem(index, 3, modified_time_str)

    # 搜索视频文件
    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)
                    file_size = os.path.getsize(file_path)
                    modified_time = os.path.getmtime(file_path)
                    file_list.append((file, file_path, file_size, modified_time))

        return file_list

这里我们通过遍历用户选择的路径,查找所有视频文件,并将其添加到列表控件中。视频文件的时长将在用户点击时获取。

4. 显示视频时长

我们使用 ffmpeg 提供的功能来获取视频文件的时长。

    # 处理文件选择事件
    def on_file_selected(self, event):
        selected_item = event.GetItem()
        file_name = selected_item.GetText()
        file_path = os.path.join(self.current_path, file_name)
        duration = self.get_video_duration(file_path)
        self.file_list_ctrl.SetItem(selected_item.GetId(), 1, duration)

    # 获取视频时长
    def get_video_duration(self, file_path):
        cmd = [
            'ffmpeg',
            '-i', file_path,
            '-hide_banner',
            '-loglevel', 'error',
            '-show_entries', 'format=duration',
            '-of', 'default=noprint_wrappers=1:nokey=1'
        ]
        result = subprocess.run(cmd, stdout=subprocess.PIPE, stderr=subprocess.PIPE, text=True)
        duration = float(result.stdout.strip())
        return str(datetime.timedelta(seconds=int(duration)))

在用户选择视频文件时,on_file_selected 事件被触发,应用会调用 ffmpeg 命令获取视频时长并显示在列表中。

5. 播放视频

用户可以通过双击视频文件来播放视频。我们使用默认的媒体播放器来实现播放功能:

    # 处理播放事件
    def on_play(self, event):
        selected_item = self.file_list_ctrl.GetFirstSelected()
        if selected_item != -1:
            file_name = self.file_list_ctrl.GetItemText(selected_item)
            file_path = os.path.join(self.current_path, file_name)
            if sys.platform.startswith('win'):
                subprocess.Popen(['start', '', file_path], shell=True)
            elif sys.platform.startswith('darwin'):
                subprocess.Popen(['open', file_path])
            elif sys.platform.startswith('linux'):
                subprocess.Popen(['xdg-open', file_path])
        else:
            wx.MessageBox("请先选择要播放的文件", "提示", wx.OK | wx.ICON_INFORMATION)

通过调用系统命令,我们让视频文件可以使用系统默认的播放器进行播放。

6. 截取视频截图

用户可以设置时间间隔,并对视频进行截图。截图将保存到以视频文件名命名的文件夹中:

    # 处理截图事件
    def on_capture(self, event):
        selected_item = self.file_list_ctrl.GetFirstSelected()
        if selected_item != -1:
            file_name = self.file_list_ctrl.GetItemText(selected_item)
            file_path = os.path.join(self.current_path, file_name)
            interval = int(self.interval_textctrl.GetValue())
            thread = threading.Thread(target=self.capture_screenshots, args=(file_path, interval))
            thread.start()
        else:
            wx.MessageBox("请先选择要截图的视频文件", "提示", wx.OK | wx.ICON_INFORMATION)

    # 截取视频截图
    def capture_screenshots(self, file_path, interval):
        # 生成以视频名称命名的文件夹
        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)

        # 构造 ffmpeg 命令
        cmd = [
            'ffmpeg',
            '-i', file_path,
            '-vf', f'fps=1/{interval}',
            os.path.join(output_dir, 'screenshot_%03d.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])

7. 运行应用

最后,添加主函数以启动应用:

if __name__ == "__main__":
    app = wx.App(False)
    frame = FileListFrame()
    frame.Show()
    app.MainLoop()

效果如下

总结

在这篇博客中,我们使用 wxPython 和 ffmpeg 创建了一个视频文件管理器。它不仅能管理和播放视频,还能生成视频截图,并将截图存放到特定的文件夹中。你可以根据需要进一步扩展这个应用程序,例如增加视频剪辑、视频合并等功能。希望这篇博客能帮助你更好地理解 wxPython 和 ffmpeg 的使用方法,并激发你在项目中应用这些技术的兴趣。

以上就是使用Python创建一个视频管理器并实现视频截图功能的详细内容,更多关于Python视频截图的资料请关注脚本之家其它相关文章!

您可能感兴趣的文章:
阅读全文