yt-dlp base class

tubearchivist/home/src/download/yt_dlp_base.py

@@ -0,0 +1,33 @@
+"""
+functionality:
+- base class to make all calls to yt-dlp
+- handle yt-dlp errors
+"""
+
+from time import sleep
+
+import yt_dlp
+
+
+class YtWrap:
+    """wrap calls to yt"""
+
+    def __init__(self, obs):
+        self.obs = obs
+
+    def download(self, url):
+        """make download request"""
+
+        with yt_dlp.YoutubeDL(self.obs) as ydl:
+            try:
+                ydl.download([url])
+            except yt_dlp.utils.DownloadError:
+                print(f"{url}: failed to download, retry...")
+                sleep(3)
+                ydl.download([url])
+
+    def extract(self, url):
+        """make extract request"""
+        response = yt_dlp.YoutubeDL(self.obs).extract_info(url)
+
+        return response