Update bleep_that_sht/yt_download.py

bleep_that_sht/yt_download.py

@@ -1,5 +1,6 @@
-from pytube import YouTube
-from pytube.query import StreamQuery
+# from pytube import YouTube
+
+import yt_dlp
 import re
 
 
@@ -12,31 +13,20 @@ def is_valid_youtube_url(url: str) -> bool:
     return re.match(pattern, url) is not None
 
 
-def download_video(
-    url: str,
-    savepath: str,
-    my_proxies: dict = {}
-) -> None:
+def download_video(url: str, savepath: str, my_proxies: dict = {}) -> None:
     try:
         print("Downloading video from youtube...")
         if is_valid_youtube_url(url):
-            yt = YouTube(url, proxies=my_proxies)
-            audio_video_streams = (
-                yt.streams.filter(
-                    file_extension="mp4",
-                    only_audio=False,
-                    only_video=False,
-                    progressive=True,
-                    type="video",
-                )
-                .order_by("resolution")
-                .asc()
-            )
-            audio_video_itags = [v.itag for v in audio_video_streams]
-            first_choice_itag = audio_video_itags[0]     
-            yt.streams.get_by_itag(first_choice_itag).download(filename=savepath)
+            ydl_opts = {
+                'format': 'bestvideo[height<=720]+bestaudio/best', 
+                'merge_output_format': 'mp4',
+                'outtmpl': savepath,
+            }
+            with yt_dlp.YoutubeDL(ydl_opts) as ydl:
+                ydl.download([url])
+
             print("...done!")
         else:
             raise ValueError(f"invalid input url: {url}")
     except Exception as e:
-        raise ValueError(f"yt_download failed with exception {e}")
+        raise ValueError(f"yt_download failed with exception {e}")