Donwload youtube video with ydl

download.py

@@ -1,4 +1,5 @@
 from pytube import YouTube
+import yt_dlp
 import os
 import argparse
 # import twitchdl.commands as twitch_downloader
@@ -67,55 +68,27 @@ def download_twitch(url, type):
 
 def download_youtube_video(url):
     try:
-        # Configurar el user agent
-        yt = YouTube(url)
-        yt.bypass_age_gate()
-
-        # Configurar headers personalizados
-        yt.stream_monostate.headers.update({
-            'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/120.0.0.0 Safari/537.36',
-            'Accept': 'text/html,application/xhtml+xml,application/xml;q=0.9,*/*;q=0.8',
-            'Accept-Language': 'en-us,en;q=0.5',
-            'Accept-Encoding': 'gzip,deflate',
-            'Accept-Charset': 'ISO-8859-1,utf-8;q=0.7,*;q=0.7',
-            'Keep-Alive': 'timeout=15, max=100',
-            'Connection': 'keep-alive',
-        })
-
-        # Obtener el stream con la mejor resolución en formato mp4
-        video = yt.streams.filter(progressive=True, file_extension='mp4').order_by('resolution').desc().first()
-        if video:
-            # Download video
-            video.download(output_path=VIDEO_FOLDER, filename=f'{DOWNLOAD_VIDEO_NAME}.{DOWNLOAD_VIDEO_FORMAT}')
-        else:
-            print('No se encontró un stream de video compatible')
+        # Config options
+        ydl_options = {
+            "outtmpl": f"{VIDEO_FOLDER}/{DOWNLOAD_VIDEO_NAME}.{DOWNLOAD_VIDEO_FORMAT}",
+            "format": "best",
+        }
+        # Download video
+        with yt_dlp.YoutubeDL(ydl_options) as ydl:
+            ydl.download(url)
     except Exception as e:
         print(f'Error descargando el video: {str(e)}')
 
 def download_youtube_audio(url):
     try:
-        # Configurar el user agent
-        yt = YouTube(url)
-        yt.bypass_age_gate()
-
-        # Configurar headers personalizados
-        yt.stream_monostate.headers.update({
-            'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/120.0.0.0 Safari/537.36',
-            'Accept': 'text/html,application/xhtml+xml,application/xml;q=0.9,*/*;q=0.8',
-            'Accept-Language': 'en-us,en;q=0.5',
-            'Accept-Encoding': 'gzip,deflate',
-            'Accept-Charset': 'ISO-8859-1,utf-8;q=0.7,*;q=0.7',
-            'Keep-Alive': 'timeout=15, max=100',
-            'Connection': 'keep-alive',
-        })
-
-        # Obtener el stream de audio con la mejor calidad
-        audio = yt.streams.filter(only_audio=True).order_by('abr').desc().first()
-        if audio:
-            # Download audio
-            audio.download(output_path=AUDIO_FOLDER, filename=f'{DOWNLOAD_AUDIO_NAME}.{DOWNLOAD_AUDIO_FORMAT}')
-        else:
-            print('No se encontró un stream de audio compatible')
+        # Config options
+        ydl_options = {
+            "outtmpl": f"{AUDIO_FOLDER}/{DOWNLOAD_AUDIO_NAME}.{DOWNLOAD_AUDIO_FORMAT}",
+            "format": "bestaudio",
+        }
+        # Download audio
+        with yt_dlp.YoutubeDL(ydl_options) as ydl:
+            ydl.download(url)
     except Exception as e:
         print(f'Error descargando el audio: {str(e)}')