使用Python開發(fā)視頻格式轉換工具

更新時間：2024年12月19日 09:41:07 作者：winfredzhang

這篇文章主要介紹了如何使用Python開發(fā)一個帶圖形界面的視頻格式轉換工具,可以將WMV格式的視頻轉換為手機可以播放的MP4格式,需要的可以參考下

在這篇文章中，我將介紹如何使用Python開發(fā)一個帶圖形界面的視頻格式轉換工具。這個工具可以將WMV格式的視頻轉換為手機可以播放的MP4格式，并且提供了友好的用戶界面。我們將使用wxPython來創(chuàng)建GUI界面，使用ffmpeg-python來處理視頻轉換。

技術棧選擇

wxPython：跨平臺GUI框架，相比tkinter提供了更現(xiàn)代的界面外觀

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
        
        # 創(chuàng)建面板
        panel = wx.Panel(self)
        
        # 創(chuàng)建垂直布局
        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()
            
            # 配置轉換參數(shù)
            stream = ffmpeg.input(self.input_file)
            
            # 設置輸出參數(shù)
            stream = ffmpeg.output(stream, output_file,
                                 vcodec='libx264',    # 視頻編碼器
                                 acodec='aac',        # 音頻編碼器
                                 video_bitrate='2M',  # 視頻比特率
                                 audio_bitrate='192k',# 音頻比特率
                                 # 針對手機優(yōu)化的其他參數(shù)
                                 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發(fā)生錯誤: {str(e)}\n")
            wx.MessageBox(f'轉換過程中發(fā)生錯誤: {str(e)}', '錯誤', wx.OK | wx.ICON_ERROR)

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

開發(fā)環(huán)境準備

首先需要安裝必要的Python模塊：

pip install wxPython ffmpeg-python

項目結構

整個項目主要包含一個主類VideoConverterFrame，繼承自wx.Frame。這個類實現(xiàn)了：

圖形界面布局

文件選擇功能

視頻信息讀取

格式轉換功能

詳細實現(xiàn)

1. 創(chuàng)建主窗口

首先，我們創(chuàng)建主窗口類并設置基本布局：

class VideoConverterFrame(wx.Frame):
    def __init__(self):
        super().__init__(parent=None, title='視頻格式轉換工具')
        self.input_file = None
        self.output_dir = None
        
        # 創(chuàng)建面板
        panel = wx.Panel(self)
        
        # 創(chuàng)建垂直布局
        vbox = wx.BoxSizer(wx.VERTICAL)
        
        # ... 添加控件
        
        # 設置窗口大小
        self.SetSize((500, 400))
        self.Centre()

2. 添加文件選擇功能

實現(xiàn)文件選擇對話框，并在選擇文件后顯示視頻信息：

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:
        # 配置轉換參數(shù)
        stream = ffmpeg.input(self.input_file)
        
        # 設置輸出參數(shù)
        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")