playlist

diff --git a/.gitignore b/.gitignore
index 9f73852c7ea486cb618541433008667328f49d40..e6333ec8b89ccd3d87e39012e2110aa61b332131 100644 (file)
--- a/.gitignore
+++ b/.gitignore
@@ -9,3 +9,4 @@
 .idea/
 examples.txt
 downloaded/
+links.txt

diff --git a/main.py b/main.py
index 8d9e65c685bdb931389669827d93574cb614f9a5..c20b94f7af2b9f7ea50d044236e1b0b7cc09bb8b 100644 (file)
--- a/main.py
+++ b/main.py
@@ -1,4 +1,4 @@
-from pytubefix import YouTube
+from pytubefix import YouTube, Playlist
 import ffmpeg
 
 
@@ -16,6 +16,14 @@ if __name__ == "__main__":
         if links[-1] == "":
             links = links[:-1]
 
+    for link in links:
+        if "playlist" in link:
+            p = Playlist(link)
+            for url in p.video_urls:
+                links.append(url)
+            links.remove(link)
+
+
     # download links one by one
     for link in links:
         target_res = 0