Update main.py

diff --git a/main.py b/main.py
index 05c1ce41b7d8467399bc82bdb44f3d86ea0307f0..fa0371b6dec18b263af2f40b87cf92e71647cdf8 100644 (file)
--- a/main.py
+++ b/main.py
@@ -1,22 +1,31 @@
 from pytube import YouTube
 
-SAVE_PATH = "D:/Youtube/CodingClub/"
+SAVE_PATH = "D:/Youtube/"
 
 with open('links_file.txt', 'r') as f:
     links = f.read().split('\n')
     if links[-1] == "":
         links = links[:-1]
 
+VIDEO = False
 
-def download(link):
+def download_video(link):
     yt = YouTube(link)
     mp4_files = yt.streams.filter(file_extension="mp4")
     mp4_files = mp4_files.get_highest_resolution()
     mp4_files.download(output_path=SAVE_PATH)
     print("Download is completed successfully")
+def download_audio(link):
+    yt = YouTube(link)
+    audio = yt.streams.filter(only_audio=True).get_audio_only()
+    audio.download(output_path=SAVE_PATH)
+    print("Download is completed successfully")
 
 
 for i in links:
-    download(i)
+    if VIDEO:
+        download_video(i)
+    else:
+        download_audio(i)
 
-print('Videos Downloaded!')
+print('Items Downloaded!')