Use yt-dlp and include soundfile

carlthome · Jul 4, 2023 · fa86526 · fa86526
1 parent 000981d
commit fa86526
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 7 deletions.
diff --git a/audioscrape/youtube.py b/audioscrape/youtube.py
@@ -4,7 +4,7 @@
 from urllib.parse import urlencode
 from urllib.request import urlopen
 
-import youtube_dl
+import yt_dlp as yt
 
 logger = logging.getLogger(__name__)
 
@@ -37,7 +37,7 @@ def scrape(query, include, exclude, quiet, verbose, overwrite, limit):
     """Search YouTube and download audio from discovered videos."""
 
     # Search YouTube for videos.
-    query_string = urlencode({"search_query": query})
+    query_string = urlencode({"search_query": f"{query} {include}"})
     url = f"http://youtube.com/results?{query_string}"
 
     # Get video IDs from search results.
@@ -66,12 +66,12 @@ def scrape(query, include, exclude, quiet, verbose, overwrite, limit):
             "postprocessors": [
                 {
                     "key": "FFmpegExtractAudio",
-                    "preferredcodec": "opus",
+                    "preferredcodec": "ogg",
                     "preferredquality": "192",
                 }
             ],
         }
-        ydl = youtube_dl.YoutubeDL(download_options)
+        ydl = yt.YoutubeDL(download_options)
 
         # Fetch metadata.
         video_info = ydl.extract_info(video_url, download=False)

diff --git a/pyproject.toml b/pyproject.toml
@@ -18,11 +18,12 @@ classifiers = [
     "Programming Language :: Python :: 3",
 ]
 dependencies = [
-    "youtube-dl @ git+https://github.com/ytdl-org/youtube-dl@fa7f0effbe4e14fcf70e1dc4496371c9862b64b9",
-    "tqdm",
+    "mediapipe",
     "requests",
     "soundcloud-lib",
-    "mediapipe",
+    "soundfile",
+    "tqdm",
+    "yt-dlp",
 ]
 dynamic = ["version"]