python

关注公众号 jb51net

关闭
首页 > 脚本专栏 > python > Python视频格式转换

使用Python开发视频格式转换工具

作者：winfredzhang

这篇文章主要介绍了如何使用Python开发一个带图形界面的视频格式转换工具,可以将WMV格式的视频转换为手机可以播放的MP4格式,需要的可以参考下

在这篇文章中，我将介绍如何使用Python开发一个带图形界面的视频格式转换工具。这个工具可以将WMV格式的视频转换为手机可以播放的MP4格式，并且提供了友好的用户界面。我们将使用wxPython来创建GUI界面，使用ffmpeg-python来处理视频转换。

技术栈选择

wxPython：跨平台GUI框架，相比tkinter提供了更现代的界面外观

ffmpeg-python：ffmpeg的Python封装，提供了简洁的API来处理视频转换

Python 3.x：使用最新的Python特性

全部代码

import wx
import os
import ffmpeg
from pathlib import Path

class VideoConverterFrame(wx.Frame):
    def __init__(self):
        super().__init__(parent=None, title='视频格式转换工具')
        self.input_file = None
        self.output_dir = None
        
        # 创建面板
        panel = wx.Panel(self)
        
        # 创建垂直布局
        vbox = wx.BoxSizer(wx.VERTICAL)
        
        # 选择输入文件
        input_box = wx.BoxSizer(wx.HORIZONTAL)
        self.input_text = wx.TextCtrl(panel, style=wx.TE_READONLY)
        input_btn = wx.Button(panel, label='选择WMV文件')
        input_btn.Bind(wx.EVT_BUTTON, self.on_choose_input)
        input_box.Add(self.input_text, proportion=1, flag=wx.EXPAND|wx.RIGHT, border=5)
        input_box.Add(input_btn, proportion=0)
        
        # 选择输出目录
        output_box = wx.BoxSizer(wx.HORIZONTAL)
        self.output_text = wx.TextCtrl(panel, style=wx.TE_READONLY)
        output_btn = wx.Button(panel, label='选择输出文件夹')
        output_btn.Bind(wx.EVT_BUTTON, self.on_choose_output)
        output_box.Add(self.output_text, proportion=1, flag=wx.EXPAND|wx.RIGHT, border=5)
        output_box.Add(output_btn, proportion=0)
        
        # 转换按钮
        convert_btn = wx.Button(panel, label='开始转换')
        convert_btn.Bind(wx.EVT_BUTTON, self.on_convert)
        
        # 进度显示
        self.status_text = wx.TextCtrl(panel, style=wx.TE_MULTILINE|wx.TE_READONLY)
        
        # 添加所有控件到垂直布局
        vbox.Add(wx.StaticText(panel, label="选择要转换的WMV文件:"), flag=wx.ALL, border=5)
        vbox.Add(input_box, flag=wx.EXPAND|wx.LEFT|wx.RIGHT, border=5)
        vbox.Add(wx.StaticText(panel, label="选择输出文件夹:"), flag=wx.ALL, border=5)
        vbox.Add(output_box, flag=wx.EXPAND|wx.LEFT|wx.RIGHT, border=5)
        vbox.Add(convert_btn, flag=wx.ALIGN_CENTER|wx.ALL, border=10)
        vbox.Add(self.status_text, proportion=1, flag=wx.EXPAND|wx.ALL, border=5)
        
        # 设置面板布局
        panel.SetSizer(vbox)
        
        # 设置窗口大小
        self.SetSize((500, 400))
        self.Centre()

    def on_choose_input(self, event):
        """选择输入文件"""
        with wx.FileDialog(self, "选择WMV文件", wildcard="WMV files (*.wmv)|*.wmv",
                         style=wx.FD_OPEN | wx.FD_FILE_MUST_EXIST) as fileDialog:
            if fileDialog.ShowModal() == wx.ID_CANCEL:
                return
            self.input_file = fileDialog.GetPath()
            self.input_text.SetValue(self.input_file)
            
            # 显示视频信息
            try:
                probe = ffmpeg.probe(self.input_file)
                video_info = next(s for s in probe['streams'] if s['codec_type'] == 'video')
                self.status_text.SetValue(f"视频信息:\n")
                self.status_text.AppendText(f"分辨率: {video_info['width']}x{video_info['height']}\n")
                self.status_text.AppendText(f"编码格式: {video_info['codec_name']}\n")
                if 'duration' in probe['format']:
                    duration = float(probe['format']['duration'])
                    self.status_text.AppendText(f"时长: {int(duration//60)}分{int(duration%60)}秒\n")
            except Exception as e:
                self.status_text.AppendText(f"无法读取视频信息: {str(e)}\n")

    def on_choose_output(self, event):
        """选择输出目录"""
        with wx.DirDialog(self, "选择输出文件夹",
                         style=wx.DD_DEFAULT_STYLE | wx.DD_DIR_MUST_EXIST) as dirDialog:
            if dirDialog.ShowModal() == wx.ID_CANCEL:
                return
            self.output_dir = dirDialog.GetPath()
            self.output_text.SetValue(self.output_dir)

    def on_convert(self, event):
        """转换视频"""
        if not self.input_file or not self.output_dir:
            wx.MessageBox('请先选择输入文件和输出文件夹!', '提示', wx.OK | wx.ICON_INFORMATION)
            return
            
        # 生成输出文件路径
        input_filename = Path(self.input_file).stem
        output_file = os.path.join(self.output_dir, f"{input_filename}.mp4")
        
        try:
            self.status_text.AppendText("\n开始转换...\n")
            wx.Yield()
            
            # 配置转换参数
            stream = ffmpeg.input(self.input_file)
            
            # 设置输出参数
            stream = ffmpeg.output(stream, output_file,
                                 vcodec='libx264',    # 视频编码器
                                 acodec='aac',        # 音频编码器
                                 video_bitrate='2M',  # 视频比特率
                                 audio_bitrate='192k',# 音频比特率
                                 # 针对手机优化的其他参数
                                 preset='medium',     # 编码速度预设
                                 movflags='+faststart',# 支持快速播放
                                 )
            
            # 开始转换
            ffmpeg.run(stream, capture_stdout=True, capture_stderr=True)
            
            self.status_text.AppendText("转换完成!\n")
            wx.MessageBox('视频转换成功!', '完成', wx.OK | wx.ICON_INFORMATION)
            
        except ffmpeg.Error as e:
            self.status_text.AppendText(f"\n转换失败: {str(e.stderr.decode())}\n")
            wx.MessageBox('视频转换失败!', '错误', wx.OK | wx.ICON_ERROR)
        except Exception as e:
            self.status_text.AppendText(f"\n发生错误: {str(e)}\n")
            wx.MessageBox(f'转换过程中发生错误: {str(e)}', '错误', wx.OK | wx.ICON_ERROR)

if __name__ == '__main__':
    app = wx.App()
    frame = VideoConverterFrame()
    frame.Show()
    app.MainLoop()

开发环境准备

首先需要安装必要的Python模块：

pip install wxPython ffmpeg-python

项目结构

整个项目主要包含一个主类VideoConverterFrame，继承自wx.Frame。这个类实现了：

图形界面布局

文件选择功能

视频信息读取

格式转换功能

详细实现

1. 创建主窗口

首先，我们创建主窗口类并设置基本布局：

class VideoConverterFrame(wx.Frame):
    def __init__(self):
        super().__init__(parent=None, title='视频格式转换工具')
        self.input_file = None
        self.output_dir = None
        
        # 创建面板
        panel = wx.Panel(self)
        
        # 创建垂直布局
        vbox = wx.BoxSizer(wx.VERTICAL)
        
        # ... 添加控件
        
        # 设置窗口大小
        self.SetSize((500, 400))
        self.Centre()

2. 添加文件选择功能

实现文件选择对话框，并在选择文件后显示视频信息：

def on_choose_input(self, event):
    """选择输入文件"""
    with wx.FileDialog(self, "选择WMV文件", wildcard="WMV files (*.wmv)|*.wmv",
                     style=wx.FD_OPEN | wx.FD_FILE_MUST_EXIST) as fileDialog:
        if fileDialog.ShowModal() == wx.ID_CANCEL:
            return
        self.input_file = fileDialog.GetPath()
        self.input_text.SetValue(self.input_file)
        
        # 显示视频信息
        try:
            probe = ffmpeg.probe(self.input_file)
            video_info = next(s for s in probe['streams'] if s['codec_type'] == 'video')
            # ... 显示视频信息
        except Exception as e:
            self.status_text.AppendText(f"无法读取视频信息: {str(e)}\n")

3. 视频转换核心功能

使用ffmpeg-python进行视频转换的核心代码：

def on_convert(self, event):
    """转换视频"""
    if not self.input_file or not self.output_dir:
        wx.MessageBox('请先选择输入文件和输出文件夹!', '提示', wx.OK | wx.ICON_INFORMATION)
        return
        
    # 生成输出文件路径
    input_filename = Path(self.input_file).stem
    output_file = os.path.join(self.output_dir, f"{input_filename}.mp4")
    
    try:
        # 配置转换参数
        stream = ffmpeg.input(self.input_file)
        
        # 设置输出参数
        stream = ffmpeg.output(stream, output_file,
                             vcodec='libx264',    # 视频编码器
                             acodec='aac',        # 音频编码器
                             video_bitrate='2M',  # 视频比特率
                             audio_bitrate='192k',# 音频比特率
                             preset='medium',     # 编码速度预设
                             movflags='+faststart'# 支持快速播放
                             )
        
        # 开始转换
        ffmpeg.run(stream, capture_stdout=True, capture_stderr=True)
        
    except ffmpeg.Error as e:
        # 错误处理
        self.status_text.AppendText(f"\n转换失败: {str(e.stderr.decode())}\n")

技术要点解析

1. 针对手机优化的转换参数

视频编码：使用H.264(libx264)编码器，这是目前手机设备支持最广泛的视频编码格式

音频编码：使用AAC编码器，提供良好的音质和压缩比

比特率设置：

快速启动：使用movflags='+faststart’参数，让视频可以在下载的同时开始播放

2. 错误处理

程序实现了完整的错误处理机制：

3. 进度反馈

通过状态文本框实时显示：

使用说明

运行程序后，点击"选择WMV文件"按钮选择要转换的视频

点击"选择输出文件夹"按钮选择保存位置

点击"开始转换"按钮开始转换

等待转换完成，查看状态提示

运行结果

以上就是使用Python开发视频格式转换工具的详细内容，更多关于Python视频格式转换的资料请关注脚本之家其它相关文章！

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