python中使用yt-dlp模块实现带进程条下载音视频

news2024/12/25 9:03:48

当代的互联网时代，视频内容的流行无疑是其中的重要组成部分。作为全球最大的视频分享平台，每天吸引着数以亿计的用户观看各种各样的视频内容。有时候，我们可能希望将某些喜欢的视频保存到本地进行观看，或者将它们用于其他用途。在本文中，我们将介绍如何使用 Python 和 wxPython 库编写一个简单的视频下载程序。

首先，我们需要安装两个必要的库：wxPython 和 yt-dlp。你可以使用以下命令来安装它们：

pip install wxPython yt-dlp

安装完成后，我们就可以开始编写程序了。下面是完整的代码示例：

import wx
import yt_dlp
import threading
import urllib.parse

class YouTubeDownloaderFrame(wx.Frame):
    def __init__(self):
        super().__init__(None, title="YouTube Downloader", size=(400, 300))
        
        self.panel = wx.Panel(self)
        
        self.url_label = wx.StaticText(self.panel, label="Video URL:")
        self.url_text = wx.TextCtrl(self.panel)
        
        self.folder_label = wx.StaticText(self.panel, label="Save Folder:")
        self.folder_text = wx.TextCtrl(self.panel)
        self.folder_btn = wx.Button(self.panel, label="Browse...")
        self.folder_btn.Bind(wx.EVT_BUTTON, self.on_browse)
        
        self.start_btn = wx.Button(self.panel, label="Start Download")
        self.start_btn.Bind(wx.EVT_BUTTON, self.on_start)
        
        self.progress_label = wx.StaticText(self.panel, label="Progress:")
        self.progress_bar = wx.Gauge(self.panel, range=100)
        
        self.sizer = wx.BoxSizer(wx.VERTICAL)
        self.sizer.Add(self.url_label, 0, wx.ALL, 5)
        self.sizer.Add(self.url_text, 0, wx.EXPAND|wx.ALL, 5)
        self.sizer.Add(self.folder_label, 0, wx.ALL, 5)
        self.sizer.Add(self.folder_text, 0, wx.EXPAND|wx.ALL, 5)
        self.sizer.Add(self.folder_btn, 0, wx.ALL, 5)
        self.sizer.Add(self.start_btn, 0, wx.ALL, 5)
        self.sizer.Add(self.progress_label, 0, wx.ALL, 5)
        self.sizer.Add(self.progress_bar, 0, wx.EXPAND|wx.ALL, 5)
        
        self.panel.SetSizer(self.sizer)
        self.Show()
    
    def on_browse(self, event):
        dlg = wx.DirDialog(self, "Choose a directory", style=wx.DD_DEFAULT_STYLE)
        if dlg.ShowModal() == wx.ID_OK:
            self.folder_text.SetValue(dlg.GetPath())
        dlg.Destroy()
    
    def on_start(self, event):
        url = self.url_text.GetValue()
        save_folder = self.folder_text.GetValue()

        download_thread = threading.Thread(target=self.download_video, args=(url, save_folder))
        download_thread.start()
    
    def download_video(self, url, save_folder):
        try:
            options = {
                'outtmpl': save_folder + '/%(title)s.%(ext)s',
                'progress_hooks': [self.update_progress],
            }
            with yt_dlp.YoutubeDL(options) as ydl:
                video_info = ydl.extract_info(url, download=False)
                if 'uploader' in video_info and 'youtube' in urllib.parse.urlparse(url).netloc:
                    options['progress_hooks'] = [self.update_progress]
                    ydl.download([url])
                else:
                    ydl.download([url])
            wx.CallAfter(wx.MessageBox, 'Download completed!', 'Info', wx.OK | wx.ICON_INFORMATION)
        
        except Exception as e:
            wx.CallAfter(wx.MessageBox, f'Download error occurred: {str(e)}', 'Error', wx.OK | wx.ICON_ERROR)
    
    def update_progress(self, progress):
        if 'total_bytes' in progress and 'downloaded_bytes' in progress:
            percentage = int(progress['downloaded_bytes'] * 100 / progress['total_bytes'])
            wx.CallAfter(self.progress_bar.SetValue, percentage)

app = wx.App()
frame = YouTubeDownloaderFrame()
app.MainLoop()

在这个程序中，我们使用了 wxPython 库来创建一个简单的图形用户界面（GUI）。用户可以在界面上输入视频的地址，并选择保存视频的文件夹。当用户点击"开始下载"按钮后，程序将使用 yt-dlp 库开始下载视频。同时，我们还添加了一个进度条来显示下载进度。

程序的主要逻辑位于 YouTubeDownloaderFrame 类中。在类的初始化方法中，我们创建了界面上的各个控件，如文本标签、文本输入框、按钮和进度条。通过绑定按钮的事件处理函数，我们可以在用户点击按钮时执行相应的操作。

下载视频的过程是在一个单独的线程中进行的，以避免阻塞用户界面。在 download_video 方法中，我们使用 yt-dlp 库来下载视频。我们配置了一些下载选项，如输出模板和进度钩子函数。下载过程中，进度钩子函数将被调用，并更新进度条的值。

下载完成后，我们使用 wx.CallAfter 方法来确保在主线程中显示下载完成的消息框，以避免线程间的竞争条件。